SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Add require_torch to 2 pipeline tests (#20585)

Browse Source 
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2022-12-05 16:06:39 +01:00
committed by GitHub
parent e7e6d1818a
commit cc8aec6740
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/pipelines/test_pipelines_table_question_answering.py
View File

@@ -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):
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",