diff --git a/mediapipe/tasks/ios/test/vision/image_classifier/MPPImageClassifierTests.m b/mediapipe/tasks/ios/test/vision/image_classifier/MPPImageClassifierTests.m index a2fd68482..d873d409a 100644 --- a/mediapipe/tasks/ios/test/vision/image_classifier/MPPImageClassifierTests.m +++ b/mediapipe/tasks/ios/test/vision/image_classifier/MPPImageClassifierTests.m @@ -652,7 +652,7 @@ static NSString *const kLiveStreamTestsDictExpectationKey = @"expectation"; }]; AssertEqualErrors(error, expectedError); - [self waitForExpectations:@[ expectation ] timeout:1e-2f]; + [self waitForExpectations:@[ expectation ] timeout:0.5f]; } - (void)testClassifyWithLiveStreamModeSucceeds { @@ -699,7 +699,7 @@ static NSString *const kLiveStreamTestsDictExpectationKey = @"expectation"; XCTAssertTrue([imageClassifier classifyAsyncImage:image timestampInMilliseconds:i error:nil]); } - [self waitForExpectations:@[ expectation ] timeout:1e-2f]; + [self waitForExpectations:@[ expectation ] timeout:0.5f]; } - (void)imageClassifier:(MPPImageClassifier *)imageClassifier