diff --git a/mediapipe/tasks/ios/text/text_classifier/utils/sources/MPPTextClassifierOptions+Helpers.mm b/mediapipe/tasks/ios/text/text_classifier/utils/sources/MPPTextClassifierOptions+Helpers.mm index 728000b44..79e4a38cc 100644 --- a/mediapipe/tasks/ios/text/text_classifier/utils/sources/MPPTextClassifierOptions+Helpers.mm +++ b/mediapipe/tasks/ios/text/text_classifier/utils/sources/MPPTextClassifierOptions+Helpers.mm @@ -27,10 +27,10 @@ using TextClassifierGraphOptionsProto = @implementation MPPTextClassifierOptions (Helpers) - (void)copyToProto:(CalculatorOptionsProto *)optionsProto { - TextClassifierGraphOptionsProto *graph_options = + TextClassifierGraphOptionsProto *graphOptions = optionsProto->MutableExtension(TextClassifierGraphOptionsProto::ext); - [self.baseOptions copyToProto:graph_options->mutable_base_options()]; - [self.classifierOptions copyToProto:graph_options->mutable_classifier_options()]; + [self.baseOptions copyToProto:graphOptions->mutable_base_options()]; + [self.classifierOptions copyToProto:graphOptions->mutable_classifier_options()]; } @end