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

feat: Update pre-commit-config.yaml (#667)

Browse Source
This commit is contained in:
Nripesh Niketan
2024-02-11 18:08:20 +04:00
committed by GitHub
parent 06072601ce
commit 0dbc4c7547
4 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/tests/test_ops.py
View File

@@ -1333,9 +1333,7 @@ class TestOps(mlx_tests.MLXTestCase):
for d in dims:
anp = np.random.randint(-20, 20, (size**d,)).reshape([size] * d)
for n_bsx in range(d):
bnp = np.random.randint(-20, 20, (size**n_bsx,)).reshape(
[size] * n_bsx
)
bnp = np.random.randint(-20, 20, (size**n_bsx,)).reshape([size] * n_bsx)
for _ in range(trial_mul * d):
amlx = mx.array(anp)
bmlx = mx.array(bnp)