"vscode:/vscode.git/clone" did not exist on "0d0d77693f79c7f7d39bba6921cc9741f00de988"
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
......@@ -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",
......
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