SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Correct @is_flaky test decoration (#31480)

Browse Source 
* Correct @is_flaky decorator
This commit is contained in:
Pavel Iakubovskii
2024-06-24 08:09:21 +01:00
committed by GitHub
parent 4b822560a1
commit 3c2d4d60d7
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/gemma/test_modeling_gemma.py
View File

@@ -493,7 +493,7 @@ class GemmaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi
@require_flash_attn @require_flash_attn
@require_torch_gpu @require_torch_gpu
@pytest.mark.flash_attn_test @pytest.mark.flash_attn_test
@is_flaky @is_flaky()
@slow @slow
def test_flash_attn_2_equivalence(self): def test_flash_attn_2_equivalence(self):
for model_class in self.all_model_classes: for model_class in self.all_model_classes:

8
tests/test_modeling_common.py
View File

@@ -3407,7 +3407,7 @@ class ModelTesterMixin:
@require_torch_gpu @require_torch_gpu
@mark.flash_attn_test @mark.flash_attn_test
@slow @slow
@is_flaky @is_flaky()
def test_flash_attn_2_inference_equivalence(self): def test_flash_attn_2_inference_equivalence(self):
for model_class in self.all_model_classes: for model_class in self.all_model_classes:
if not model_class._supports_flash_attn_2: if not model_class._supports_flash_attn_2:
@@ -3501,7 +3501,7 @@ class ModelTesterMixin:
@require_torch_gpu @require_torch_gpu
@mark.flash_attn_test @mark.flash_attn_test
@slow @slow
@is_flaky @is_flaky()
def test_flash_attn_2_inference_equivalence_right_padding(self): def test_flash_attn_2_inference_equivalence_right_padding(self):
for model_class in self.all_model_classes: for model_class in self.all_model_classes:
if not model_class._supports_flash_attn_2: if not model_class._supports_flash_attn_2:
@@ -3591,7 +3591,7 @@ class ModelTesterMixin:
@require_torch_gpu @require_torch_gpu
@mark.flash_attn_test @mark.flash_attn_test
@slow @slow
@is_flaky @is_flaky()
def test_flash_attn_2_generate_left_padding(self): def test_flash_attn_2_generate_left_padding(self):
for model_class in self.all_generative_model_classes: for model_class in self.all_generative_model_classes:
if not model_class._supports_flash_attn_2: if not model_class._supports_flash_attn_2:
@@ -3635,7 +3635,7 @@ class ModelTesterMixin:
@require_flash_attn @require_flash_attn
@require_torch_gpu @require_torch_gpu
@mark.flash_attn_test @mark.flash_attn_test
@is_flaky @is_flaky()
@slow @slow
def test_flash_attn_2_generate_padding_right(self): def test_flash_attn_2_generate_padding_right(self):
for model_class in self.all_generative_model_classes: for model_class in self.all_generative_model_classes: