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

Fix shape check

Browse Source
This commit is contained in:
Angelos Katharopoulos 2025-05-13 13:21:34 -07:00
parent 134ed4a58a
commit 40d2fc1263
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/tests/test_ops.py
View File

@ -1259,7 +1259,7 @@ class TestOps(mlx_tests.MLXTestCase):
b = mx.put_along_axis(a, a, a, axis=None) b = mx.put_along_axis(a, a, a, axis=None)
mx.eval(b) mx.eval(b)
self.assertEqual(b.size, 0) self.assertEqual(b.size, 0)
self.assertEqual(b.shape, tuple()) self.assertEqual(b.shape, a.shape)
def test_split(self): def test_split(self):
a = mx.array([1, 2, 3]) a = mx.array([1, 2, 3])