Remove separate numpy import

NNPDF · Mar 8, 2024 · be8fa20 · be8fa20
1 parent 3c6d024
commit be8fa20
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)