@@ -113,7 +113,7 @@ def test_create_load_multimodal_embedding_generator_model(
113
113
"gemini-1.5-pro-preview-0514" ,
114
114
"gemini-1.5-flash-preview-0514" ,
115
115
"gemini-1.5-pro-001" ,
116
- "gemini-1.5-pro-002" ,
116
+ # "gemini-1.5-pro-002",
117
117
"gemini-1.5-flash-001" ,
118
118
"gemini-1.5-flash-002" ,
119
119
"gemini-2.0-flash-exp" ,
@@ -148,7 +148,7 @@ def test_create_load_gemini_text_generator_model(
148
148
"gemini-1.5-pro-preview-0514" ,
149
149
"gemini-1.5-flash-preview-0514" ,
150
150
"gemini-1.5-pro-001" ,
151
- "gemini-1.5-pro-002" ,
151
+ # "gemini-1.5-pro-002",
152
152
"gemini-1.5-flash-001" ,
153
153
"gemini-1.5-flash-002" ,
154
154
"gemini-2.0-flash-exp" ,
@@ -175,7 +175,7 @@ def test_gemini_text_generator_predict_default_params_success(
175
175
"gemini-1.5-pro-preview-0514" ,
176
176
"gemini-1.5-flash-preview-0514" ,
177
177
"gemini-1.5-pro-001" ,
178
- "gemini-1.5-pro-002" ,
178
+ # "gemini-1.5-pro-002",
179
179
"gemini-1.5-flash-001" ,
180
180
"gemini-1.5-flash-002" ,
181
181
"gemini-2.0-flash-exp" ,
@@ -204,7 +204,7 @@ def test_gemini_text_generator_predict_with_params_success(
204
204
"gemini-1.5-pro-preview-0514" ,
205
205
"gemini-1.5-flash-preview-0514" ,
206
206
"gemini-1.5-pro-001" ,
207
- "gemini-1.5-pro-002" ,
207
+ # "gemini-1.5-pro-002",
208
208
"gemini-1.5-flash-001" ,
209
209
"gemini-1.5-flash-002" ,
210
210
"gemini-2.0-flash-exp" ,
@@ -235,7 +235,7 @@ def test_gemini_text_generator_multi_cols_predict_success(
235
235
"gemini-1.5-pro-preview-0514" ,
236
236
"gemini-1.5-flash-preview-0514" ,
237
237
"gemini-1.5-pro-001" ,
238
- "gemini-1.5-pro-002" ,
238
+ # "gemini-1.5-pro-002",
239
239
"gemini-1.5-flash-001" ,
240
240
"gemini-1.5-flash-002" ,
241
241
"gemini-2.0-flash-exp" ,
@@ -768,7 +768,7 @@ def test_text_embedding_generator_retry_no_progress(session, bq_connection):
768
768
@pytest .mark .parametrize (
769
769
"model_name" ,
770
770
(
771
- "gemini-1.5-pro-002" ,
771
+ # "gemini-1.5-pro-002",
772
772
"gemini-1.5-flash-002" ,
773
773
"gemini-2.0-flash-001" ,
774
774
"gemini-2.0-flash-lite-001" ,
@@ -798,7 +798,7 @@ def test_llm_gemini_score(llm_fine_tune_df_default_index, model_name):
798
798
@pytest .mark .parametrize (
799
799
"model_name" ,
800
800
(
801
- "gemini-1.5-pro-002" ,
801
+ # "gemini-1.5-pro-002",
802
802
"gemini-1.5-flash-002" ,
803
803
"gemini-2.0-flash-001" ,
804
804
"gemini-2.0-flash-lite-001" ,
0 commit comments