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

bump pre commit and fix format (#373)

Browse Source
This commit is contained in:
Awni Hannun
2024-01-04 16:28:52 -08:00
committed by GitHub
parent c82a8cc526
commit b9e415d19c
6 changed files with 2 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/tests/test_reduce.py
View File

@@ -65,7 +65,6 @@ class TestReduce(mlx_tests.MLXTestCase):
for op in ("sum", "prod", "min", "max"):
with self.subTest(op=op):
np_op = getattr(np, op)
mlx_op = getattr(mx, op)
@@ -96,7 +95,6 @@ class TestReduce(mlx_tests.MLXTestCase):
]
for dtype in dtypes:
with self.subTest(dtype=dtype):
data = np.random.rand(10, 12, 13).astype(getattr(np, dtype))
x = mx.array(data)
for op in ["argmin", "argmax"]: