From 8342ed3e5b82ef44e550fb4d5fe63b2b6246e0f3 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Wed, 8 Mar 2023 21:30:05 +0100 Subject: [PATCH 1/2] slow me --- tests/models/bridgetower/test_modeling_bridgetower.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/models/bridgetower/test_modeling_bridgetower.py b/tests/models/bridgetower/test_modeling_bridgetower.py index 9e70c4cdcd44..b59f0e88438e 100644 --- a/tests/models/bridgetower/test_modeling_bridgetower.py +++ b/tests/models/bridgetower/test_modeling_bridgetower.py @@ -202,6 +202,7 @@ def prepare_config_and_inputs_for_common(self): return config, inputs_dict +@slow @require_torch @unittest.skipIf(not is_torch_greater_or_equal_than_1_10, "BridgeTower is only available in torch v1.10+") class BridgeTowerModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase): From adcef01f531476d599107b116b90ad958a30d1a2 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Wed, 8 Mar 2023 21:31:22 +0100 Subject: [PATCH 2/2] slow me --- tests/models/bridgetower/test_modeling_bridgetower.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/models/bridgetower/test_modeling_bridgetower.py b/tests/models/bridgetower/test_modeling_bridgetower.py index b59f0e88438e..20396c8bf7bf 100644 --- a/tests/models/bridgetower/test_modeling_bridgetower.py +++ b/tests/models/bridgetower/test_modeling_bridgetower.py @@ -495,6 +495,7 @@ def test_constrastive_learning(self): self.assertEqual(outputs.logits.shape, expected_shape) +@slow @require_torch @unittest.skipIf(not is_torch_greater_or_equal_than_1_10, "BridgeTower is only available in torch v1.10+") class BridgeTowerModelTrainingTest(unittest.TestCase):