Add require_torch to 2 pipeline tests (#20585)

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2022-12-05 16:06:39 +01:00
parent e7e6d1818a
commit cc8aec6740
1 changed files with 2 additions and 0 deletions
--- a/tests/pipelines/test_pipelines_table_question_answering.py
+++ b/tests/pipelines/test_pipelines_table_question_answering.py
@@ -481,6 +481,7 @@ class TQAPipelineTests(unittest.TestCase, metaclass=PipelineTestCaseMeta):
            )

    @slow
+    @require_torch
    def test_integration_wtq_pt(self):
        table_querier = pipeline("table-question-answering")

@@ -574,6 +575,7 @@ class TQAPipelineTests(unittest.TestCase, metaclass=PipelineTestCaseMeta):
        self.assertListEqual(results, expected_results)

    @slow
+    @require_torch
    def test_integration_sqa_pt(self):
        table_querier = pipeline(
            "table-question-answering",