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",