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 build w/ flatten (#195)

Browse Source
This commit is contained in:
Awni Hannun
2023-12-17 11:58:45 -08:00
committed by GitHub
parent 52e1589a52
commit 90d04072b7
3 changed files with 30 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/tests/test_array.py
View File

@@ -906,7 +906,7 @@ class TestArray(mlx_tests.MLXTestCase):
# Check slice assign with negative indices works
a = mx.zeros((5, 5), mx.int32)
a[2:-2, 2:-2] = 4
self.assertEquals(a[2, 2].item(), 4)
self.assertEqual(a[2, 2].item(), 4)
def test_slice_negative_step(self):
a_np = np.arange(20)