Add skip logic for attentions test - Levit (#17633)

2022-06-10 12:46:30 +02:00
parent cdaed367b0
commit fd1e67033e
1 changed files with 4 additions and 0 deletions
--- a/tests/models/levit/test_modeling_levit.py
+++ b/tests/models/levit/test_modeling_levit.py
@@ -205,6 +205,10 @@ class LevitModelTest(ModelTesterMixin, unittest.TestCase):
    def test_model_common_attributes(self):
        pass

+    @unittest.skip(reason="Levit does not output attentions")
+    def test_attention_outputs(self):
+        pass
+
    def test_forward_signature(self):
        config, _ = self.model_tester.prepare_config_and_inputs_for_common()