zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-19 00:04:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix unsqueeze with None (#899)

Browse Source 
* Fix unsqueeze with None

* Clean unnecessary files
This commit is contained in:
Abdussamet Türker
2024-03-26 23:59:44 +03:00
committed by GitHub
parent 570f2bf29e
commit 5611e1a95e
2 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/tests/test_array.py
View File

@@ -763,6 +763,10 @@ class TestArray(mlx_tests.MLXTestCase):
a_sliced_npy = np.asarray(a_sliced_mlx)
self.assertTrue(np.array_equal(a_sliced_npy, a_npy[None]))
a_sliced_mlx = a_mlx[:, None]
a_sliced_npy = np.asarray(a_sliced_mlx)
self.assertTrue(np.array_equal(a_sliced_npy, a_npy[:, None]))
# Multi dim indexing, all ints
self.assertEqual(a_mlx[0, 0].item(), 0)
self.assertEqual(a_mlx[0, 0].ndim, 0)