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

[Tests] flaky test_constrained_beam_search_generate_dict_output (#37276)

Browse Source
This commit is contained in:
Joao Gante
2025-04-04 13:38:42 +01:00
committed by GitHub
parent 4ba0989eab
commit acbcb5d07d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/generation/test_utils.py
View File

@@ -889,8 +889,7 @@ class GenerationTesterMixin:
num_beams=beam_kwargs["num_beams"], num_beams=beam_kwargs["num_beams"],
) )
# TODO: @gante check why it is flaky @is_flaky() # Some models have position-specific tokens, this test may try to force them in an invalid position
@is_flaky()
@pytest.mark.generate @pytest.mark.generate
def test_constrained_beam_search_generate(self): def test_constrained_beam_search_generate(self):
for model_class in self.all_generative_model_classes: for model_class in self.all_generative_model_classes:
@@ -947,6 +946,7 @@ class GenerationTesterMixin:
for generation_output in output_generate: for generation_output in output_generate:
self._check_sequence_inside_sequence(force_tokens, generation_output) self._check_sequence_inside_sequence(force_tokens, generation_output)
@is_flaky() # Some models have position-specific tokens, this test may try to force them in an invalid position
@pytest.mark.generate @pytest.mark.generate
def test_constrained_beam_search_generate_dict_output(self): def test_constrained_beam_search_generate_dict_output(self):
for model_class in self.all_generative_model_classes: for model_class in self.all_generative_model_classes: