From 06bab003380bce7e7b64bafc4fe89601d418ee90 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Wed, 19 Apr 2023 20:27:19 +0200 Subject: [PATCH] Remove some pipeline skip cases (#22865) fix Co-authored-by: ydshieh --- tests/models/fnet/test_modeling_fnet.py | 2 +- tests/models/mbart/test_modeling_mbart.py | 3 +-- tests/models/xlnet/test_modeling_xlnet.py | 6 ++++-- 3 files changed, 6 insertions(+), 5 deletions(-) diff --git a/tests/models/fnet/test_modeling_fnet.py b/tests/models/fnet/test_modeling_fnet.py index 836da26021..e7e592d5b6 100644 --- a/tests/models/fnet/test_modeling_fnet.py +++ b/tests/models/fnet/test_modeling_fnet.py @@ -303,7 +303,7 @@ class FNetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase): def is_pipeline_test_to_skip( 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 False diff --git a/tests/models/mbart/test_modeling_mbart.py b/tests/models/mbart/test_modeling_mbart.py index b28d539b78..d7094ca6e4 100644 --- a/tests/models/mbart/test_modeling_mbart.py +++ b/tests/models/mbart/test_modeling_mbart.py @@ -257,8 +257,7 @@ class MBartModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi def is_pipeline_test_to_skip( self, pipeline_test_casse_name, config_class, model_architecture, tokenizer_name, processor_name ): - if pipeline_test_casse_name != "FeatureExtractionPipelineTests": - # IndexError: index out of range in self + if pipeline_test_casse_name == "QAPipelineTests" and not tokenizer_name.endswith("Fast"): return True return False diff --git a/tests/models/xlnet/test_modeling_xlnet.py b/tests/models/xlnet/test_modeling_xlnet.py index 98c935cdec..2b3f4752ee 100644 --- a/tests/models/xlnet/test_modeling_xlnet.py +++ b/tests/models/xlnet/test_modeling_xlnet.py @@ -546,8 +546,10 @@ class XLNetModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi def is_pipeline_test_to_skip( self, pipeline_test_casse_name, config_class, model_architecture, tokenizer_name, processor_name ): - # IndexError: index out of range in self - return True + if pipeline_test_casse_name == "QAPipelineTests" and not tokenizer_name.endswith("Fast"): + return True + + return False # 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):