From c392ed2c09a7c8756cc87c14e4e8d31062ce4911 Mon Sep 17 00:00:00 2001 From: Fergus Henderson Date: Thu, 13 Apr 2023 08:47:19 -0700 Subject: [PATCH] Internal change PiperOrigin-RevId: 524012087 --- .../test/vision/image_classifier/MPPImageClassifierTests.m | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mediapipe/tasks/ios/test/vision/image_classifier/MPPImageClassifierTests.m b/mediapipe/tasks/ios/test/vision/image_classifier/MPPImageClassifierTests.m index f7b26837e..613239944 100644 --- a/mediapipe/tasks/ios/test/vision/image_classifier/MPPImageClassifierTests.m +++ b/mediapipe/tasks/ios/test/vision/image_classifier/MPPImageClassifierTests.m @@ -40,7 +40,7 @@ static NSString *const kExpectedErrorDomain = @"com.google.mediapipe.tasks"; XCTAssertEqual(categories.count, expectedCategories.count); \ for (int i = 0; i < categories.count; i++) { \ XCTAssertEqual(categories[i].index, expectedCategories[i].index, @"index i = %d", i); \ - XCTAssertEqualWithAccuracy(categories[i].score, expectedCategories[i].score, 1e-3, \ + XCTAssertEqualWithAccuracy(categories[i].score, expectedCategories[i].score, 1e-2, \ @"index i = %d", i); \ XCTAssertEqualObjects(categories[i].categoryName, expectedCategories[i].categoryName, \ @"index i = %d", i); \ @@ -263,7 +263,7 @@ static NSString *const kExpectedErrorDomain = @"com.google.mediapipe.tasks"; MPPImageClassifier *imageClassifier = [self imageClassifierWithOptionsSucceeds:options]; NSArray *expectedCategories = @[ [[MPPCategory alloc] initWithIndex:934 - score:0.972656f + score:0.968750f categoryName:@"cheeseburger" displayName:nil] ];