Rerun test_tokenizer on fail

Lightning-AI · Apr 16, 2024 · d6d9b76 · d6d9b76
1 parent ca5816f
commit d6d9b76
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_tokenizer.py b/tests/test_tokenizer.py
@@ -11,6 +11,7 @@
 from litgpt.tokenizer import Tokenizer
 
 
+@pytest.mark.flaky(reruns=5)
 @pytest.mark.parametrize("config", config_module.configs, ids=[c["hf_config"]["name"] for c in config_module.configs])
 def test_tokenizer_against_hf(config):
     access_token = os.getenv("HF_TOKEN")