From d6db8c7268084af7c2e874c8db254fd1eac4415f Mon Sep 17 00:00:00 2001 From: Prianka Liz Kariat Date: Thu, 4 May 2023 16:17:37 +0530 Subject: [PATCH] Updated MPPImageClassifierTests.m to reflect new delegate name --- .../vision/image_classifier/MPPImageClassifierTests.m | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/mediapipe/tasks/ios/test/vision/image_classifier/MPPImageClassifierTests.m b/mediapipe/tasks/ios/test/vision/image_classifier/MPPImageClassifierTests.m index 1f06a3e41..b09a58674 100644 --- a/mediapipe/tasks/ios/test/vision/image_classifier/MPPImageClassifierTests.m +++ b/mediapipe/tasks/ios/test/vision/image_classifier/MPPImageClassifierTests.m @@ -56,7 +56,7 @@ static NSString *const kLiveStreamTestsDictExpectationKey = @"expectation"; XCTAssertEqual(imageClassifierResult.classificationResult.classifications.count, 1); \ XCTAssertEqual(imageClassifierResult.classificationResult.classifications[0].headIndex, 0); -@interface MPPImageClassifierTests : XCTestCase { +@interface MPPImageClassifierTests : XCTestCase { NSDictionary *liveStreamSucceedsTestDict; NSDictionary *outOfOrderTimestampTestDict; } @@ -447,7 +447,7 @@ static NSString *const kLiveStreamTestsDictExpectationKey = @"expectation"; MPPImageClassifierOptions *options = [self imageClassifierOptionsWithModelName:kFloatModelName]; options.runningMode = runningModesToTest[i]; - options.imageClassifierDelegate = self; + options.imageClassifierLiveStreamDelegate = self; [self assertCreateImageClassifierWithOptions:options @@ -558,7 +558,7 @@ static NSString *const kLiveStreamTestsDictExpectationKey = @"expectation"; MPPImageClassifierOptions *options = [self imageClassifierOptionsWithModelName:kFloatModelName]; options.runningMode = MPPRunningModeLiveStream; - options.imageClassifierDelegate = self; + options.imageClassifierLiveStreamDelegate = self; MPPImageClassifier *imageClassifier = [self imageClassifierWithOptionsSucceeds:options]; @@ -622,7 +622,7 @@ static NSString *const kLiveStreamTestsDictExpectationKey = @"expectation"; options.maxResults = 3; options.runningMode = MPPRunningModeLiveStream; - options.imageClassifierDelegate = self; + options.imageClassifierLiveStreamDelegate = self; XCTestExpectation *expectation = [[XCTestExpectation alloc] initWithDescription:@"classifyWithOutOfOrderTimestampsAndLiveStream"]; @@ -662,7 +662,7 @@ static NSString *const kLiveStreamTestsDictExpectationKey = @"expectation"; options.maxResults = maxResults; options.runningMode = MPPRunningModeLiveStream; - options.imageClassifierDelegate = self; + options.imageClassifierLiveStreamDelegate = self; NSInteger iterationCount = 100;