Removed unused custom classifier options in tests
This commit is contained in:
parent
de619e2702
commit
14e18a03d6
|
@ -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.
|
||||||
|
|
Loading…
Reference in New Issue
Block a user