From 03c3e91b0d6d5d76011ed0306d6b272a2f404ad7 Mon Sep 17 00:00:00 2001 From: Adam Pocock Date: Wed, 9 Aug 2023 14:43:05 -0400 Subject: [PATCH] [java] Relaxing CoreML test (#16777) ### Description Reduces precision on the CoreML provider test as it returns slightly different answers than the other tested providers. Checked on a 2020 13" M1 MBP. ### Motivation and Context Fixes Java CoreML test failure after #16763. --- java/src/test/java/ai/onnxruntime/InferenceTest.java | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/java/src/test/java/ai/onnxruntime/InferenceTest.java b/java/src/test/java/ai/onnxruntime/InferenceTest.java index 5ca7bbc651..08d2a5698d 100644 --- a/java/src/test/java/ai/onnxruntime/InferenceTest.java +++ b/java/src/test/java/ai/onnxruntime/InferenceTest.java @@ -656,7 +656,12 @@ public class InferenceTest { OnnxValue resultTensor = result.get(0); float[] resultArray = TestHelpers.flattenFloat(resultTensor.getValue()); assertEquals(expectedOutput.length, resultArray.length); - assertArrayEquals(expectedOutput, resultArray, 1e-6f); + if (provider == OrtProvider.CORE_ML) { + // CoreML gives slightly different answers on a 2020 13" M1 MBP + assertArrayEquals(expectedOutput, resultArray, 1e-2f); + } else { + assertArrayEquals(expectedOutput, resultArray, 1e-6f); + } } catch (OrtException e) { throw new IllegalStateException("Failed to execute a scoring operation", e); }