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

fd1e6703 · amyeroberts · GitHub · cdaed367 · fd1e6703
Unverified Commit fd1e6703 authored Jun 10, 2022 by amyeroberts Committed by GitHub Jun 10, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

tests/models/levit/test_modeling_levit.py tests/models/levit/test_modeling_levit.py +4 -0

No files found.
--- 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()