From cc8aec67409fd495d165f46162847548e218b9b6 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Mon, 5 Dec 2022 16:06:39 +0100 Subject: [PATCH] Add `require_torch` to 2 pipeline tests (#20585) Co-authored-by: ydshieh --- tests/pipelines/test_pipelines_table_question_answering.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/pipelines/test_pipelines_table_question_answering.py b/tests/pipelines/test_pipelines_table_question_answering.py index 9e2e7e5317531..3527f3ef22648 100644 --- a/tests/pipelines/test_pipelines_table_question_answering.py +++ b/tests/pipelines/test_pipelines_table_question_answering.py @@ -481,6 +481,7 @@ def test_slow_tokenizer_sqa_tf(self): ) @slow + @require_torch def test_integration_wtq_pt(self): table_querier = pipeline("table-question-answering") @@ -574,6 +575,7 @@ def test_integration_wtq_tf(self): self.assertListEqual(results, expected_results) @slow + @require_torch def test_integration_sqa_pt(self): table_querier = pipeline( "table-question-answering",