Skip to content

Commit 6608735

Browse files
authored
test: remove deprecated gemini 1.5 tests (#1776)
1 parent ec81dd2 commit 6608735

File tree

1 file changed

+3
-21
lines changed

1 file changed

+3
-21
lines changed

tests/system/small/ml/test_llm.py

Lines changed: 3 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -158,12 +158,6 @@ def test_gemini_text_generator_predict_default_params_success(
158158
@pytest.mark.parametrize(
159159
"model_name",
160160
(
161-
"gemini-1.5-pro-preview-0514",
162-
"gemini-1.5-flash-preview-0514",
163-
"gemini-1.5-pro-001",
164-
"gemini-1.5-pro-002",
165-
"gemini-1.5-flash-001",
166-
"gemini-1.5-flash-002",
167161
"gemini-2.0-flash-exp",
168162
"gemini-2.0-flash-001",
169163
"gemini-2.0-flash-lite-001",
@@ -212,13 +206,9 @@ def test_gemini_text_generator_multi_cols_predict_success(
212206
@pytest.mark.parametrize(
213207
"model_name",
214208
(
215-
"gemini-1.5-pro-preview-0514",
216-
"gemini-1.5-flash-preview-0514",
217-
"gemini-1.5-pro-001",
218-
"gemini-1.5-pro-002",
219-
"gemini-1.5-flash-001",
220-
"gemini-1.5-flash-002",
221209
"gemini-2.0-flash-exp",
210+
"gemini-2.0-flash-001",
211+
"gemini-2.0-flash-lite-001",
222212
),
223213
)
224214
@pytest.mark.flaky(retries=2)
@@ -738,8 +728,6 @@ def test_text_embedding_generator_retry_no_progress(session, bq_connection):
738728
@pytest.mark.parametrize(
739729
"model_name",
740730
(
741-
"gemini-1.5-pro-002",
742-
"gemini-1.5-flash-002",
743731
"gemini-2.0-flash-001",
744732
"gemini-2.0-flash-lite-001",
745733
),
@@ -768,8 +756,6 @@ def test_llm_gemini_score(llm_fine_tune_df_default_index, model_name):
768756
@pytest.mark.parametrize(
769757
"model_name",
770758
(
771-
"gemini-1.5-pro-002",
772-
"gemini-1.5-flash-002",
773759
"gemini-2.0-flash-001",
774760
"gemini-2.0-flash-lite-001",
775761
),
@@ -797,11 +783,7 @@ def test_llm_gemini_pro_score_params(llm_fine_tune_df_default_index, model_name)
797783

798784
@pytest.mark.parametrize(
799785
"model_name",
800-
(
801-
"gemini-1.5-pro-preview-0514",
802-
"gemini-1.5-flash-preview-0514",
803-
"gemini-2.0-flash-exp",
804-
),
786+
("gemini-2.0-flash-exp",),
805787
)
806788
def test_gemini_preview_model_warnings(model_name):
807789
with pytest.warns(exceptions.PreviewWarning):

0 commit comments

Comments
 (0)