cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Removed unused custom classifier options in tests

Browse Source
This commit is contained in:
kinaryml 2022-11-03 02:54:57 -07:00
parent de619e2702
commit 14e18a03d6
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mediapipe/tasks/python/test/text/text_classifier_test.py
View File

@ -196,9 +196,7 @@ class ImageClassifierTest(parameterized.TestCase):
# Should never happen # Should never happen
raise ValueError('model_file_type is invalid.') raise ValueError('model_file_type is invalid.')
custom_classifier_options = _ClassifierOptions() options = _TextClassifierOptions(base_options=base_options)
options = _TextClassifierOptions(
base_options=base_options, classifier_options=custom_classifier_options)
classifier = _TextClassifier.create_from_options(options) classifier = _TextClassifier.create_from_options(options)
# Performs text classification on the input. # Performs text classification on the input.
@ -230,9 +228,7 @@ class ImageClassifierTest(parameterized.TestCase):
# Should never happen # Should never happen
raise ValueError('model_file_type is invalid.') raise ValueError('model_file_type is invalid.')
custom_classifier_options = _ClassifierOptions() options = _TextClassifierOptions(base_options=base_options)
options = _TextClassifierOptions(
base_options=base_options, classifier_options=custom_classifier_options)
with _TextClassifier.create_from_options(options) as classifier: with _TextClassifier.create_from_options(options) as classifier:
# Performs text classification on the input. # Performs text classification on the input.