From 0060655fe5914281190cf6fb4012f6e9562dd4e1 Mon Sep 17 00:00:00 2001 From: Alex Rubinsteyn Date: Wed, 13 Mar 2024 15:19:16 -0400 Subject: [PATCH] fixed arg to numpy allclose --- tests/test_netmhc_stabpan.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_netmhc_stabpan.py b/tests/test_netmhc_stabpan.py index 577dfed..ceae57e 100644 --- a/tests/test_netmhc_stabpan.py +++ b/tests/test_netmhc_stabpan.py @@ -10,7 +10,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -import numpy as np +from numpy.testing import assert_allclose from mhctools import NetMHCstabpan @@ -57,6 +57,6 @@ def test_netmhc_stabpan_accuracy(): for i, (expected, actual) in enumerate(zip(web_server_predictions, stability_predictions)): # Check to make sure that the stability predictions are within 0.01 of the webserver values. # This could be the result of different versions of dependencies or the nature of the ANN itself. - np.testing.assert_allclose(expected, actual, atol=0.01, error_msg="Peptide %d: expected %f but got %f" % (i, expected, actual)) + assert_allclose(expected, actual, atol=0.01, err_msg="Peptide %d: expected %f but got %f" % (i, expected, actual)) \ No newline at end of file