makr test_llm_gemini_score as flask

googleapis · Dec 26, 2024 · d25b2d5 · d25b2d5
1 parent e5c5854
commit d25b2d5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/system/small/ml/test_llm.py b/tests/system/small/ml/test_llm.py
@@ -417,6 +417,7 @@ def test_llm_palm_score_params(llm_fine_tune_df_default_index):
     )
 
 
+@pytest.mark.flaky(retries=2)
 @pytest.mark.parametrize(
     "model_name",
     (