From 08d60635a7da369ddcbe6149af6ab527125d4ab3 Mon Sep 17 00:00:00 2001 From: Anthony Shoumikhin Date: Mon, 18 Nov 2024 19:21:57 -0800 Subject: [PATCH] Lower confidence threshold in tests to avoid flakiness. Summary: . Differential Revision: D66146583 --- .../Sources/MobileNet/Test/MobileNetClassifierTest.swift | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/demo-apps/apple_ios/ExecuTorchDemo/ExecuTorchDemo/Sources/MobileNet/Test/MobileNetClassifierTest.swift b/examples/demo-apps/apple_ios/ExecuTorchDemo/ExecuTorchDemo/Sources/MobileNet/Test/MobileNetClassifierTest.swift index f6886ab37c2..7936699bb9b 100644 --- a/examples/demo-apps/apple_ios/ExecuTorchDemo/ExecuTorchDemo/Sources/MobileNet/Test/MobileNetClassifierTest.swift +++ b/examples/demo-apps/apple_ios/ExecuTorchDemo/ExecuTorchDemo/Sources/MobileNet/Test/MobileNetClassifierTest.swift @@ -48,9 +48,9 @@ final class MobileNetClassifierTest: XCTestCase { modelFilePath: modelFilePath, labelsFilePath: labelsFilePath) for expectedClassification in [ - Classification(label: "Arctic fox", confidence: 0.92), - Classification(label: "Samoyed", confidence: 0.74), - Classification(label: "hot pot", confidence: 0.82), + Classification(label: "Arctic fox", confidence: 0.9), + Classification(label: "Samoyed", confidence: 0.7), + Classification(label: "hot pot", confidence: 0.8), ] { guard let imagePath = Bundle(for: type(of: self))