From 4c8428a076adcb8b5fac576f6daf11b54b59f779 Mon Sep 17 00:00:00 2001 From: bojiang Date: Thu, 15 Sep 2022 17:51:51 +0800 Subject: [PATCH] chore(ci): fix the unit test of transformers --- tests/integration/frameworks/test_transformers_unit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/integration/frameworks/test_transformers_unit.py b/tests/integration/frameworks/test_transformers_unit.py index 4a24c3470a..22286e2650 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")