chore(ci): fix the unit test of transformers (bentoml#3003)

the check_task now returns three values
sauyon · Sep 16, 2022 · cbed3b0 · cbed3b0
1 parent 7732ad9
commit cbed3b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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")