fix for theano

openvax · Feb 19, 2018 · 885a97c · 885a97c
1 parent 823f8c7
commit 885a97c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_class1_affinity_predictor.py b/test/test_class1_affinity_predictor.py
@@ -233,7 +233,7 @@ def test_predict_implementations_equivalent():
                 peptides=peptides + ["SSSN"],
                 throw=False,
                 centrality_measure=centrality_measure).prediction.values
-            testing.assert_equal(pred1, pred2)
+            testing.assert_almost_equal(pred1, pred2, decimal=2)
 
             pred1 = DOWNLOADED_PREDICTOR.predict(
                 allele=allele,
@@ -243,4 +243,4 @@ def test_predict_implementations_equivalent():
                 allele=allele,
                 peptides=peptides,
                 centrality_measure=centrality_measure).prediction.values
-            testing.assert_equal(pred1, pred2)
+            testing.assert_almost_equal(pred1, pred2, decimal=2)