Commit 407a850f authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

[test][embeddings] relax np.all_close tolerance

parent 59ae733a
Pipeline #45875 passed with stage
in 33 minutes and 35 seconds
......@@ -28,7 +28,7 @@ def test_idiap_inceptionv2_msceleb():
transformer_sample = wrap(["sample"], transformer)
output = [ for s in transformer_sample.transform([sample])][0]
assert np.allclose(output, reference)
np.testing.assert_allclose(output, reference.flatten(), rtol=1e-5, atol=1e-4)
assert output.size == 128, output.shape
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment