fix: Fixed raising TypeError instead of ValueError for invalid type (#32111)
* Raised TypeError instead of ValueError for invalid types. * Updated formatting using ruff. * Retrieved few changes. * Retrieved few changes. * Updated tests accordingly.
This commit is contained in:
@@ -60,7 +60,7 @@ def output_type(output):
|
||||
elif isinstance(output, (torch.Tensor, AgentAudio)):
|
||||
return "audio"
|
||||
else:
|
||||
raise ValueError(f"Invalid output: {output}")
|
||||
raise TypeError(f"Invalid output: {output}")
|
||||
|
||||
|
||||
@is_agent_test
|
||||
|
||||
@@ -188,7 +188,7 @@ class LukeTokenizerTest(TokenizerTesterMixin, unittest.TestCase):
|
||||
with self.assertRaises(ValueError):
|
||||
tokenizer(sentence, entities=tuple(entities), entity_spans=spans)
|
||||
|
||||
with self.assertRaises(ValueError):
|
||||
with self.assertRaises(TypeError):
|
||||
tokenizer(sentence, entities=entities, entity_spans=tuple(spans))
|
||||
|
||||
with self.assertRaises(ValueError):
|
||||
|
||||
@@ -151,7 +151,7 @@ class MLukeTokenizerTest(TokenizerTesterMixin, unittest.TestCase):
|
||||
with self.assertRaises(ValueError):
|
||||
tokenizer(sentence, entities=tuple(entities), entity_spans=spans)
|
||||
|
||||
with self.assertRaises(ValueError):
|
||||
with self.assertRaises(TypeError):
|
||||
tokenizer(sentence, entities=entities, entity_spans=tuple(spans))
|
||||
|
||||
with self.assertRaises(ValueError):
|
||||
|
||||
@@ -171,7 +171,7 @@ class FeatureExtractionPipelineTests(unittest.TestCase):
|
||||
elif isinstance(input_, float):
|
||||
return 0
|
||||
else:
|
||||
raise ValueError("We expect lists of floats, nothing else")
|
||||
raise TypeError("We expect lists of floats, nothing else")
|
||||
return shape
|
||||
|
||||
def get_test_pipeline(self, model, tokenizer, processor, torch_dtype="float32"):
|
||||
|
||||
@@ -145,7 +145,7 @@ class PipelineTesterMixin:
|
||||
if not isinstance(model_architectures, tuple):
|
||||
model_architectures = (model_architectures,)
|
||||
if not isinstance(model_architectures, tuple):
|
||||
raise ValueError(f"`model_architectures` must be a tuple. Got {type(model_architectures)} instead.")
|
||||
raise TypeError(f"`model_architectures` must be a tuple. Got {type(model_architectures)} instead.")
|
||||
|
||||
for model_architecture in model_architectures:
|
||||
model_arch_name = model_architecture.__name__
|
||||
|
||||
Reference in New Issue
Block a user