zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-11-02 17:28:12 +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

1
python/tests/test_bf16.py
View File

@@ -122,7 +122,6 @@ class TestBF16(mlx_tests.MLXTestCase):
for op in ("min", "max"):
with self.subTest(op=op):
for axes in (0, 1, 2, (0, 1), (0, 2), (1, 2), (0, 1, 2)):
with self.subTest(axes=axes):
np_args = (x.astype(np.float32),)