Fix unsqueeze with None (#899)

* 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

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)