From 14e18a03d6c0bc589554fb395c583e754106379d Mon Sep 17 00:00:00 2001 From: kinaryml Date: Thu, 3 Nov 2022 02:54:57 -0700 Subject: [PATCH] Removed unused custom classifier options in tests --- mediapipe/tasks/python/test/text/text_classifier_test.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/mediapipe/tasks/python/test/text/text_classifier_test.py b/mediapipe/tasks/python/test/text/text_classifier_test.py index c97299270..bb4528c75 100644 --- a/mediapipe/tasks/python/test/text/text_classifier_test.py +++ b/mediapipe/tasks/python/test/text/text_classifier_test.py @@ -196,9 +196,7 @@ class ImageClassifierTest(parameterized.TestCase): # Should never happen raise ValueError('model_file_type is invalid.') - custom_classifier_options = _ClassifierOptions() - options = _TextClassifierOptions( - base_options=base_options, classifier_options=custom_classifier_options) + options = _TextClassifierOptions(base_options=base_options) classifier = _TextClassifier.create_from_options(options) # Performs text classification on the input. @@ -230,9 +228,7 @@ class ImageClassifierTest(parameterized.TestCase): # Should never happen raise ValueError('model_file_type is invalid.') - custom_classifier_options = _ClassifierOptions() - options = _TextClassifierOptions( - base_options=base_options, classifier_options=custom_classifier_options) + options = _TextClassifierOptions(base_options=base_options) with _TextClassifier.create_from_options(options) as classifier: # Performs text classification on the input.