Fix M1 incompatibility issue for ONNX integration (#2114)

castorini · May 12, 2023 · a913b52 · a913b52
1 parent 0009387
commit a913b52
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/test/java/io/anserini/search/SimpleImpactSearcherTest.java b/src/test/java/io/anserini/search/SimpleImpactSearcherTest.java
@@ -211,8 +211,8 @@ public void testOnnxEncoder() throws Exception{
     searcher.set_onnx_query_encoder("SpladePlusPlusEnsembleDistil");
 
     Map<String, Float> encoded_query = searcher.encode_with_onnx("here is a test");
-    assertEquals(encoded_query.get("here"), EXPECTED_ENCODED_QUERY.get("here"));
-    assertEquals(encoded_query.get("a"), EXPECTED_ENCODED_QUERY.get("a"));
-    assertEquals(encoded_query.get("test"), EXPECTED_ENCODED_QUERY.get("test"));
+    assertEquals(encoded_query.get("here"), EXPECTED_ENCODED_QUERY.get("here"), 2e-4);
+    assertEquals(encoded_query.get("a"), EXPECTED_ENCODED_QUERY.get("a"), 2e-4);
+    assertEquals(encoded_query.get("test"), EXPECTED_ENCODED_QUERY.get("test"), 2e-4);
   }
 }