From fd1e67033e9ef79a9a2211f9a97c2ebbf0332b57 Mon Sep 17 00:00:00 2001 From: amyeroberts Date: Fri, 10 Jun 2022 12:46:30 +0200 Subject: [PATCH] Add skip logic for attentions test - Levit (#17633) --- tests/models/levit/test_modeling_levit.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/models/levit/test_modeling_levit.py b/tests/models/levit/test_modeling_levit.py index d5854220d1..725b279fd0 100644 --- 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()