From 0d33381fade9e8b887e91dd5e20afda9158d535e Mon Sep 17 00:00:00 2001 From: Joao Gante Date: Thu, 9 Feb 2023 14:46:15 +0000 Subject: [PATCH] =?UTF-8?q?Tag=20tests=20as=20slow=20=E2=8C=9B=20(#21537)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit begone slow tests --- tests/models/bridgetower/test_modeling_bridgetower.py | 5 +++++ tests/models/mobilebert/test_modeling_tf_mobilebert.py | 5 +++++ tests/models/mobilevit/test_modeling_tf_mobilevit.py | 1 + tests/models/rag/test_modeling_tf_rag.py | 1 + tests/test_modeling_common.py | 1 + 5 files changed, 13 insertions(+) diff --git a/tests/models/bridgetower/test_modeling_bridgetower.py b/tests/models/bridgetower/test_modeling_bridgetower.py index 4be58f450e..7d35959968 100644 --- a/tests/models/bridgetower/test_modeling_bridgetower.py +++ b/tests/models/bridgetower/test_modeling_bridgetower.py @@ -227,6 +227,11 @@ class BridgeTowerModelTest(ModelTesterMixin, unittest.TestCase): model = BridgeTowerModel.from_pretrained(model_name) self.assertIsNotNone(model) + @slow + def test_save_load_fast_init_from_base(self): + # Override as it is a slow test on this model + super().test_save_load_fast_init_from_base() + # Override as extracting meaningful tensor from output is different for BridgeTower def test_save_load(self): config, input_dict = self.model_tester.prepare_config_and_inputs_for_common() diff --git a/tests/models/mobilebert/test_modeling_tf_mobilebert.py b/tests/models/mobilebert/test_modeling_tf_mobilebert.py index 40586d2001..2ee38a6ee6 100644 --- a/tests/models/mobilebert/test_modeling_tf_mobilebert.py +++ b/tests/models/mobilebert/test_modeling_tf_mobilebert.py @@ -317,6 +317,11 @@ class TFMobileBertModelTest(TFModelTesterMixin, unittest.TestCase): name = model.get_bias() assert name is None + @slow + def test_keras_fit(self): + # Override as it is a slow test on this model + super().test_keras_fit() + @tooslow def test_saved_model_creation(self): pass diff --git a/tests/models/mobilevit/test_modeling_tf_mobilevit.py b/tests/models/mobilevit/test_modeling_tf_mobilevit.py index d46ee895ed..eea07f9413 100644 --- a/tests/models/mobilevit/test_modeling_tf_mobilevit.py +++ b/tests/models/mobilevit/test_modeling_tf_mobilevit.py @@ -268,6 +268,7 @@ class MobileViTModelTest(TFModelTesterMixin, unittest.TestCase): not is_tf_available() or len(tf.config.list_physical_devices("GPU")) == 0, reason="TF does not support backprop for grouped convolutions on CPU.", ) + @slow def test_keras_fit(self): config, _ = self.model_tester.prepare_config_and_inputs_for_common() diff --git a/tests/models/rag/test_modeling_tf_rag.py b/tests/models/rag/test_modeling_tf_rag.py index 541d7cd132..4a0e4176b4 100644 --- a/tests/models/rag/test_modeling_tf_rag.py +++ b/tests/models/rag/test_modeling_tf_rag.py @@ -260,6 +260,7 @@ class TFRagTestMixin: num_beams=2, num_return_sequences=2, decoder_start_token_id=config.generator.eos_token_id, + max_new_tokens=5, ) self.assertIsNotNone(outputs) diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py index b78e777551..d7d82e694a 100755 --- a/tests/test_modeling_common.py +++ b/tests/test_modeling_common.py @@ -3305,6 +3305,7 @@ class ModelUtilsTest(TestCasePlus): _ = ModelWithHead.from_pretrained(tmp_dir) @require_torch_gpu + @slow def test_pretrained_low_mem_new_config(self): # Checking for 1 model(the same one which was described in the issue) . model_ids = ["gpt2"]