Internal change

PiperOrigin-RevId: 524012087
This commit is contained in:
Fergus Henderson 2023-04-13 08:47:19 -07:00 committed by Copybara-Service
parent d095829c7c
commit c392ed2c09

View File

@ -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<MPPCategory *> *expectedCategories = @[ [[MPPCategory alloc] initWithIndex:934
score:0.972656f
score:0.968750f
categoryName:@"cheeseburger"
displayName:nil] ];