Remove separate numpy import

NNPDF · Mar 7, 2024 · 9cfa7c2 · 9cfa7c2
1 parent 90311c1
commit 9cfa7c2
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/n3fit/src/n3fit/tests/test_fit.py b/n3fit/src/n3fit/tests/test_fit.py
@@ -18,7 +18,6 @@
 import subprocess as sp
 
 import h5py
-import numpy as np
 from numpy.testing import assert_allclose, assert_equal
 import pytest
 
@@ -240,6 +239,4 @@ def compare_weights(option_1, option_2, file_1, file_2):
         else:
             weight_name = file_1[key].name
             err_msg = f"Difference between runs `n3fit {option_1}` and `n3fit {option_2}` in weights {weight_name}"
-            np.testing.assert_allclose(
-                file_1[key][:], file_2[key][:], rtol=1e-5, atol=1e-5, err_msg=err_msg
-            )
+            assert_allclose(file_1[key][:], file_2[key][:], rtol=1e-5, atol=1e-5, err_msg=err_msg)