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

Fixing typos

Browse Source
This commit is contained in:
Luca Arnaboldi
2024-02-20 17:33:29 +01:00
parent 955c985768
commit 8dcdd1bb8c
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
python/tests/test_array.py
View File

@@ -667,9 +667,9 @@ class TestArray(mlx_tests.MLXTestCase):
mx.uint16, mx.uint16,
mx.uint32, mx.uint32,
mx.uint64, mx.uint64,
# mx.float16, mx.float16,
mx.float32, mx.float32,
# mx.bfloat16, mx.bfloat16,
mx.complex64, mx.complex64,
] ]
import pickle import pickle
@@ -691,9 +691,9 @@ class TestArray(mlx_tests.MLXTestCase):
mx.uint16, mx.uint16,
mx.uint32, mx.uint32,
mx.uint64, mx.uint64,
# mx.float16, mx.float16,
mx.float32, mx.float32,
# mx.bfloat16, mx.bfloat16,
mx.complex64, mx.complex64,
] ]
@@ -706,7 +706,6 @@ class TestArray(mlx_tests.MLXTestCase):
self.assertEqualArray(x, y) self.assertEqualArray(x, y)
y -= 1 y -= 1
print(x, y)
self.assertEqualArray(x - 1, y) self.assertEqualArray(x - 1, y)
def test_indexing(self): def test_indexing(self):