"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "d4cda29af14d0dce1c285cb98ad92b06dc46a085"
Unverified Commit 06bab003 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Remove some pipeline skip cases (#22865)



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 648bd5a8
...@@ -303,7 +303,7 @@ class FNetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase): ...@@ -303,7 +303,7 @@ class FNetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
def is_pipeline_test_to_skip( def is_pipeline_test_to_skip(
self, pipeline_test_casse_name, config_class, model_architecture, tokenizer_name, processor_name self, pipeline_test_casse_name, config_class, model_architecture, tokenizer_name, processor_name
): ):
if pipeline_test_casse_name == "QAPipelineTests": if pipeline_test_casse_name == "QAPipelineTests" and not tokenizer_name.endswith("Fast"):
return True return True
return False return False
......
...@@ -257,8 +257,7 @@ class MBartModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi ...@@ -257,8 +257,7 @@ class MBartModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi
def is_pipeline_test_to_skip( def is_pipeline_test_to_skip(
self, pipeline_test_casse_name, config_class, model_architecture, tokenizer_name, processor_name self, pipeline_test_casse_name, config_class, model_architecture, tokenizer_name, processor_name
): ):
if pipeline_test_casse_name != "FeatureExtractionPipelineTests": if pipeline_test_casse_name == "QAPipelineTests" and not tokenizer_name.endswith("Fast"):
# IndexError: index out of range in self
return True return True
return False return False
......
...@@ -546,8 +546,10 @@ class XLNetModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi ...@@ -546,8 +546,10 @@ class XLNetModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi
def is_pipeline_test_to_skip( def is_pipeline_test_to_skip(
self, pipeline_test_casse_name, config_class, model_architecture, tokenizer_name, processor_name self, pipeline_test_casse_name, config_class, model_architecture, tokenizer_name, processor_name
): ):
# IndexError: index out of range in self if pipeline_test_casse_name == "QAPipelineTests" and not tokenizer_name.endswith("Fast"):
return True return True
return False
# XLNet has 2 QA models -> need to manually set the correct labels for one of them here # XLNet has 2 QA models -> need to manually set the correct labels for one of them here
def _prepare_for_class(self, inputs_dict, model_class, return_labels=False): def _prepare_for_class(self, inputs_dict, model_class, return_labels=False):
......
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