Skip to content

Commit c51d2b1

Browse files
authored
test: remove gemini-1x model tests due to model deprecation (#1764)
1 parent 768a757 commit c51d2b1

File tree

3 files changed

+0
-28
lines changed

3 files changed

+0
-28
lines changed

tests/system/load/test_llm.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,8 +41,6 @@ def llm_remote_text_df(session, llm_remote_text_pandas_df):
4141
@pytest.mark.parametrize(
4242
"model_name",
4343
(
44-
"gemini-1.5-pro-002",
45-
"gemini-1.5-flash-002",
4644
"gemini-2.0-flash-001",
4745
"gemini-2.0-flash-lite-001",
4846
),

tests/system/small/ml/test_llm.py

Lines changed: 0 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -108,12 +108,6 @@ def test_create_load_multimodal_embedding_generator_model(
108108
@pytest.mark.parametrize(
109109
"model_name",
110110
(
111-
"gemini-1.5-pro-preview-0514",
112-
"gemini-1.5-flash-preview-0514",
113-
"gemini-1.5-pro-001",
114-
"gemini-1.5-pro-002",
115-
"gemini-1.5-flash-001",
116-
"gemini-1.5-flash-002",
117111
"gemini-2.0-flash-exp",
118112
"gemini-2.0-flash-001",
119113
"gemini-2.0-flash-lite-001",
@@ -143,12 +137,6 @@ def test_create_load_gemini_text_generator_model(
143137
@pytest.mark.parametrize(
144138
"model_name",
145139
(
146-
"gemini-1.5-pro-preview-0514",
147-
"gemini-1.5-flash-preview-0514",
148-
"gemini-1.5-pro-001",
149-
"gemini-1.5-pro-002",
150-
"gemini-1.5-flash-001",
151-
"gemini-1.5-flash-002",
152140
"gemini-2.0-flash-exp",
153141
"gemini-2.0-flash-001",
154142
"gemini-2.0-flash-lite-001",
@@ -199,12 +187,6 @@ def test_gemini_text_generator_predict_with_params_success(
199187
@pytest.mark.parametrize(
200188
"model_name",
201189
(
202-
"gemini-1.5-pro-preview-0514",
203-
"gemini-1.5-flash-preview-0514",
204-
"gemini-1.5-pro-001",
205-
"gemini-1.5-pro-002",
206-
"gemini-1.5-flash-001",
207-
"gemini-1.5-flash-002",
208190
"gemini-2.0-flash-exp",
209191
"gemini-2.0-flash-001",
210192
"gemini-2.0-flash-lite-001",

tests/system/small/ml/test_multimodal_llm.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -41,10 +41,6 @@ def test_multimodal_embedding_generator_predict_default_params_success(
4141
@pytest.mark.parametrize(
4242
"model_name",
4343
(
44-
"gemini-1.5-pro-001",
45-
"gemini-1.5-pro-002",
46-
"gemini-1.5-flash-001",
47-
"gemini-1.5-flash-002",
4844
"gemini-2.0-flash-exp",
4945
"gemini-2.0-flash-001",
5046
"gemini-2.0-flash-lite-001",
@@ -71,10 +67,6 @@ def test_gemini_text_generator_multimodal_input(
7167
@pytest.mark.parametrize(
7268
"model_name",
7369
(
74-
"gemini-1.5-pro-001",
75-
"gemini-1.5-pro-002",
76-
"gemini-1.5-flash-001",
77-
"gemini-1.5-flash-002",
7870
"gemini-2.0-flash-exp",
7971
"gemini-2.0-flash-001",
8072
),

0 commit comments

Comments
 (0)