Correct @is_flaky test decoration (#31480)

* Correct @is_flaky decorator
2024-06-24 08:09:21 +01:00
parent 4b822560a1
commit 3c2d4d60d7
2 changed files with 5 additions and 5 deletions
--- a/tests/models/gemma/test_modeling_gemma.py
+++ b/tests/models/gemma/test_modeling_gemma.py
@@ -493,7 +493,7 @@ class GemmaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi
    @require_flash_attn
    @require_torch_gpu
    @pytest.mark.flash_attn_test
-    @is_flaky
+    @is_flaky()
    @slow
    def test_flash_attn_2_equivalence(self):
        for model_class in self.all_model_classes: