From 3774a08ae43e0df04a020fae518674591d3ea330 Mon Sep 17 00:00:00 2001 From: ArthurZucker Date: Mon, 23 Oct 2023 06:27:37 -0400 Subject: [PATCH 1/3] skip two tests --- .../blenderbot_small/test_modeling_tf_blenderbot_small.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tests/models/blenderbot_small/test_modeling_tf_blenderbot_small.py b/tests/models/blenderbot_small/test_modeling_tf_blenderbot_small.py index 021f171789e4..d167efd60f11 100644 --- a/tests/models/blenderbot_small/test_modeling_tf_blenderbot_small.py +++ b/tests/models/blenderbot_small/test_modeling_tf_blenderbot_small.py @@ -209,6 +209,14 @@ def test_decoder_model_past_large_inputs(self): config_and_inputs = self.model_tester.prepare_config_and_inputs_for_common() 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_tf From 280573c97698733e59245b937e8198e60d75bbb5 Mon Sep 17 00:00:00 2001 From: ArthurZucker Date: Mon, 23 Oct 2023 06:28:28 -0400 Subject: [PATCH 2/3] skip torch as well --- .../models/blenderbot_small/test_modeling_blenderbot_small.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/models/blenderbot_small/test_modeling_blenderbot_small.py b/tests/models/blenderbot_small/test_modeling_blenderbot_small.py index 249a8a799a83..2397b6fee972 100644 --- a/tests/models/blenderbot_small/test_modeling_blenderbot_small.py +++ b/tests/models/blenderbot_small/test_modeling_blenderbot_small.py @@ -244,7 +244,9 @@ def is_pipeline_test_to_skip( ): if pipeline_test_casse_name == "TextGenerationPipelineTests": return True - + # TODO @Rocketnight1 to fix + if pipeline_test_casse_name == "ConversationalPipelineTests": + return True return False def setUp(self): From e2cd71fad50a8395e863f525df7b6683fce3dc1c Mon Sep 17 00:00:00 2001 From: ArthurZucker Date: Mon, 23 Oct 2023 06:34:15 -0400 Subject: [PATCH 3/3] fixup --- .../models/blenderbot_small/test_modeling_tf_blenderbot_small.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/models/blenderbot_small/test_modeling_tf_blenderbot_small.py b/tests/models/blenderbot_small/test_modeling_tf_blenderbot_small.py index d167efd60f11..6fde705082d1 100644 --- a/tests/models/blenderbot_small/test_modeling_tf_blenderbot_small.py +++ b/tests/models/blenderbot_small/test_modeling_tf_blenderbot_small.py @@ -218,6 +218,7 @@ def is_pipeline_test_to_skip( return True return False + @require_tokenizers @require_tf class TFBlenderbot90MIntegrationTests(unittest.TestCase):