Skip to content

Commit eb0ef75

Browse files
authored
test: enable Gemini 1.5 flash default test (#859)
1 parent 10da997 commit eb0ef75

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

tests/system/small/ml/test_llm.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -328,12 +328,7 @@ def test_create_load_gemini_text_generator_model(
328328

329329
@pytest.mark.parametrize(
330330
"model_name",
331-
(
332-
"gemini-pro",
333-
"gemini-1.5-pro-preview-0514",
334-
# TODO(garrrettwu): enable when cl/637028077 is in prod.
335-
# "gemini-1.5-flash-preview-0514"
336-
),
331+
("gemini-pro", "gemini-1.5-pro-preview-0514", "gemini-1.5-flash-preview-0514"),
337332
)
338333
@pytest.mark.flaky(retries=2)
339334
def test_gemini_text_generator_predict_default_params_success(

0 commit comments

Comments
 (0)