Forgotten pdb

ml-explore · Aug 23, 2024 · f06d8c2 · f06d8c2
1 parent f716bab
commit f06d8c2
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/python/tests/test_reduce.py b/python/tests/test_reduce.py
@@ -43,10 +43,6 @@ def test_expand_sums(self):
                                 z_npy = np.sum(y_npy, axis=a) / 1000
                                 z_mlx = mx.sum(y_mlx, axis=a) / 1000
                                 mx.eval(z_mlx)
-                                if not np.allclose(z_npy, np.array(z_mlx), atol=1e-4):
-                                    import pdb
-
-                                    pdb.set_trace()
                                 self.assertTrue(
                                     np.allclose(z_npy, np.array(z_mlx), atol=1e-4)
                                 )