From 7f37aeff02a927e1a940284c6a3fb168ad0af4de Mon Sep 17 00:00:00 2001 From: stefanradev93 Date: Sun, 1 Dec 2024 23:00:20 -0500 Subject: [PATCH] Increase tolerance to atol=1e-5 --- tests/utils/ops.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/utils/ops.py b/tests/utils/ops.py index f46320a06..b12e90478 100644 --- a/tests/utils/ops.py +++ b/tests/utils/ops.py @@ -1,11 +1,11 @@ import keras -def isclose(x1, x2, rtol=1e-5, atol=1e-8): +def isclose(x1, x2, rtol=1e-5, atol=1e-5): return keras.ops.abs(x1 - x2) <= atol + rtol * keras.ops.abs(x2) -def allclose(x1, x2, rtol=1e-5, atol=1e-8): +def allclose(x1, x2, rtol=1e-5, atol=1e-5): return keras.ops.all(isclose(x1, x2, rtol, atol))