diff --git a/tests/integration/frameworks/test_transformers_unit.py b/tests/integration/frameworks/test_transformers_unit.py index 4a24c3470a3..22286e26504 100644 --- a/tests/integration/frameworks/test_transformers_unit.py +++ b/tests/integration/frameworks/test_transformers_unit.py @@ -38,7 +38,7 @@ def test_convert_to_auto_class(): alias = "sentiment-analysis" -original_task, _ = check_task(alias) # type: ignore (unfinished transformers type) +original_task, *_ = check_task(alias) # type: ignore (unfinished transformers type) sentiment = pipeline(alias, model="hf-internal-testing/tiny-random-distilbert")