zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-11-03 09:58:17 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Add CircleCI configuration (#4)

Browse Source 
* Add CircleCI configuration
This commit is contained in:
Angelos Katharopoulos
2023-12-04 16:04:11 -08:00
committed by GitHub
parent db487e6b1a
commit 7546fdb100
4 changed files with 251 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/tests/test_ops.py
View File

@@ -1084,7 +1084,7 @@ class TestOps(mlx_tests.MLXTestCase):
a[-1] = 0.0
a = mx.softmax(mx.array(a))
self.assertFalse(np.any(np.isnan(a)))
self.assertTrue((a[:-1] == 0).all())
self.assertTrue((a[:-1] < 1e-9).all())
self.assertEqual(a[-1], 1)
def test_concatenate(self):