From 1cbac6867b34d1e41b17112638136e501814015d Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Wed, 8 Mar 2023 21:48:29 +0100 Subject: [PATCH] Mark all `BridgeTower` tests slow for now (#22039) * slow me --------- Co-authored-by: ydshieh --- tests/models/bridgetower/test_modeling_bridgetower.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/models/bridgetower/test_modeling_bridgetower.py b/tests/models/bridgetower/test_modeling_bridgetower.py index 9e70c4cdcd..20396c8bf7 100644 --- a/tests/models/bridgetower/test_modeling_bridgetower.py +++ b/tests/models/bridgetower/test_modeling_bridgetower.py @@ -202,6 +202,7 @@ class BridgeTowerModelTester: 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): @@ -494,6 +495,7 @@ class BridgeTowerModelIntegrationTest(unittest.TestCase): 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):