Add require_torch to 2 pipeline tests (#20585)

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
huggingface · Dec 5, 2022 · cc8aec6 · cc8aec6
1 parent e7e6d18
commit cc8aec6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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",