Fix CI

hertschuh · Jan 23, 2024 · f97e3c7 · f97e3c7
1 parent 9ff93f3
commit f97e3c7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras/ops/math_test.py b/keras/ops/math_test.py
@@ -1099,7 +1099,7 @@ def test_solve_call(self):
         b = np.array([[9, 8], [5, 4]], dtype=np.float32)
         output = solve_op.call(a, b)
         expected_output = np.linalg.solve(a, b)
-        np.testing.assert_allclose(output, expected_output, atol=1e-6)
+        self.assertAllClose(output, expected_output, atol=1e-6)
 
 
 class FFT2Test(testing.TestCase):