Skip to content
Snippets Groups Projects
Commit 407a850f authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

[test][embeddings] relax np.all_close tolerance

parent 59ae733a
No related branches found
No related tags found
1 merge request!82Cleanup
Pipeline #45875 passed
......@@ -28,7 +28,7 @@ def test_idiap_inceptionv2_msceleb():
transformer_sample = wrap(["sample"], transformer)
output = [s.data 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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment