From a913b52478ae63e4427215e16f58081686f8e8f2 Mon Sep 17 00:00:00 2001 From: Arthur Chen <36494787+ArthurChen189@users.noreply.github.com> Date: Thu, 11 May 2023 20:00:11 -0400 Subject: [PATCH] Fix M1 incompatibility issue for ONNX integration (#2114) --- .../java/io/anserini/search/SimpleImpactSearcherTest.java | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/test/java/io/anserini/search/SimpleImpactSearcherTest.java b/src/test/java/io/anserini/search/SimpleImpactSearcherTest.java index 451ab1aaee..86379d3f7b 100644 --- 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 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); } }