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

Remove tuple unpacking syntax to comply with earlier python versions. Add cuda skip to nanpropagation tests, fix cuda implementation in a separate PR.

Browse Source
This commit is contained in:
Joona Havukainen
2025-07-08 20:52:09 -07:00
parent aca7fac9ef
commit 9a742090ae
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/tests/test_reduce.py
View File

@@ -170,7 +170,7 @@ class TestReduce(mlx_tests.MLXTestCase):
x = (mx.random.normal((4, 4)) * 10).astype(getattr(mx, dtype))
indices = mx.random.randint(0, 4, shape=(6,)).reshape(3, 2)
for idx in indices:
x[*idx] = mx.nan
x[idx[0], idx[1]] = mx.nan
x_np = np.array(x)
for op in ["max"]: