small fix bench

NNPDF · Jul 24, 2024 · bbfc210 · bbfc210
1 parent 807fc02
commit bbfc210
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/bench_regression.py b/benchmarks/bench_regression.py
@@ -133,4 +133,4 @@ def benchmark_regression(tmp_path, dataset):
         raise FileNotFoundError("Regression did not exist and has been regenerated")
 
     regression_data = np.load(regression_path)
-    np.testing.assert_allclose(regression_data, result)
+    np.testing.assert_allclose(regression_data, result, rtol=4e-6)