Skip to content

GitLab

Unverified Commit cc8aec67 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Add `require_torch` to 2 pipeline tests (#20585) 


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent e7e6d181
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
tests/pipelines/test_pipelines_table_question_answering.py
View file @ cc8aec67
...@@ -481,6 +481,7 @@ class TQAPipelineTests(unittest.TestCase, metaclass=PipelineTestCaseMeta): ...@@ -481,6 +481,7 @@ class TQAPipelineTests(unittest.TestCase, metaclass=PipelineTestCaseMeta):
) )
@slow @slow
@require_torch
def test_integration_wtq_pt(self): def test_integration_wtq_pt(self):
table_querier = pipeline("table-question-answering") table_querier = pipeline("table-question-answering")
...@@ -574,6 +575,7 @@ class TQAPipelineTests(unittest.TestCase, metaclass=PipelineTestCaseMeta): ...@@ -574,6 +575,7 @@ class TQAPipelineTests(unittest.TestCase, metaclass=PipelineTestCaseMeta):
self.assertListEqual(results, expected_results) self.assertListEqual(results, expected_results)
@slow @slow
@require_torch
def test_integration_sqa_pt(self): def test_integration_sqa_pt(self):
table_querier = pipeline( table_querier = pipeline(
"table-question-answering", "table-question-answering",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment