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

fix flaky test (#2371)

Browse Source
This commit is contained in:
Awni Hannun
2025-07-14 17:16:18 -07:00
committed by GitHub
parent e7d2ebadd2
commit 49114f28ab
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/tests/test_ops.py
View File

@@ -1478,7 +1478,7 @@ class TestOps(mlx_tests.MLXTestCase):
r_mlx = mlxop(y)
mx.eval(r_mlx)
self.assertTrue(np.allclose(r_np, r_mlx, atol=atol))
self.assertTrue(np.allclose(r_np, r_mlx, atol=atol, equal_nan=True))
x = np.random.rand(9, 12, 18)
xi = np.random.rand(9, 12, 18)