fix assert array equal

Jax922 · Oct 6, 2023 · 4c2a36d · 4c2a36d
1 parent 6e273b7
commit 4c2a36d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/rainbow/util/test_tools.py b/python/rainbow/util/test_tools.py
@@ -6,7 +6,7 @@
 """
 
 
-def is_array_equal(arr1, arr2, dec=8):
+def is_array_equal(arr1, arr2):
     """
     2022-04-06 Kenny TODO: Write proper documentation.
 
@@ -15,7 +15,7 @@ def is_array_equal(arr1, arr2, dec=8):
     :param dec:
     :return:
     """
-    return None == np.testing.assert_array_almost_equal(arr1, arr2, dec)
+    return None == np.testing.assert_allclose(arr1, arr2, rtol=1e-5, atol=1e-8)
 
 
 def is_array_not_equal(arr1, arr2):