zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-18 23:21:16 +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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats 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) r_mlx = mlxop(y)
mx.eval(r_mlx) 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) x = np.random.rand(9, 12, 18)
xi = np.random.rand(9, 12, 18) xi = np.random.rand(9, 12, 18)