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

skip two tests (#27013)

Browse Source 
* skip two tests

* skip torch as well

* fixup
This commit is contained in:
Arthur
2023-10-23 12:52:05 +02:00
committed by GitHub
parent 45425660d0
commit ef978d0a7b
2 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/models/blenderbot_small/test_modeling_blenderbot_small.py
View File

@@ -244,7 +244,9 @@ class BlenderbotSmallModelTest(ModelTesterMixin, GenerationTesterMixin, Pipeline
): ):
if pipeline_test_casse_name == "TextGenerationPipelineTests": if pipeline_test_casse_name == "TextGenerationPipelineTests":
return True return True
# TODO @Rocketnight1 to fix
if pipeline_test_casse_name == "ConversationalPipelineTests":
return True
return False return False
def setUp(self): def setUp(self):

9
tests/models/blenderbot_small/test_modeling_tf_blenderbot_small.py
View File

@@ -209,6 +209,15 @@ class TFBlenderbotSmallModelTest(TFModelTesterMixin, PipelineTesterMixin, unitte
config_and_inputs = self.model_tester.prepare_config_and_inputs_for_common() config_and_inputs = self.model_tester.prepare_config_and_inputs_for_common()
self.model_tester.check_decoder_model_past_large_inputs(*config_and_inputs) self.model_tester.check_decoder_model_past_large_inputs(*config_and_inputs)
# TODO: Fix the failed tests when this model gets more usage
def is_pipeline_test_to_skip(
self, pipeline_test_casse_name, config_class, model_architecture, tokenizer_name, processor_name
):
# TODO @Rocketnight1 to fix
if pipeline_test_casse_name == "ConversationalPipelineTests":
return True
return False
@require_tokenizers @require_tokenizers
@require_tf @require_tf