From cbed3b08e8acca0f876743ff696afa1e3baf6858 Mon Sep 17 00:00:00 2001
From: bojiang <5886138+bojiang@users.noreply.github.com>
Date: Thu, 15 Sep 2022 18:01:06 +0800
Subject: [PATCH] chore(ci): fix the unit test of transformers (#3003)

the check_task now returns three values
---
 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 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")