@@ -158,12 +158,6 @@ def test_gemini_text_generator_predict_default_params_success(
158
158
@pytest .mark .parametrize (
159
159
"model_name" ,
160
160
(
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" ,
167
161
"gemini-2.0-flash-exp" ,
168
162
"gemini-2.0-flash-001" ,
169
163
"gemini-2.0-flash-lite-001" ,
@@ -212,13 +206,9 @@ def test_gemini_text_generator_multi_cols_predict_success(
212
206
@pytest .mark .parametrize (
213
207
"model_name" ,
214
208
(
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" ,
221
209
"gemini-2.0-flash-exp" ,
210
+ "gemini-2.0-flash-001" ,
211
+ "gemini-2.0-flash-lite-001" ,
222
212
),
223
213
)
224
214
@pytest .mark .flaky (retries = 2 )
@@ -738,8 +728,6 @@ def test_text_embedding_generator_retry_no_progress(session, bq_connection):
738
728
@pytest .mark .parametrize (
739
729
"model_name" ,
740
730
(
741
- "gemini-1.5-pro-002" ,
742
- "gemini-1.5-flash-002" ,
743
731
"gemini-2.0-flash-001" ,
744
732
"gemini-2.0-flash-lite-001" ,
745
733
),
@@ -768,8 +756,6 @@ def test_llm_gemini_score(llm_fine_tune_df_default_index, model_name):
768
756
@pytest .mark .parametrize (
769
757
"model_name" ,
770
758
(
771
- "gemini-1.5-pro-002" ,
772
- "gemini-1.5-flash-002" ,
773
759
"gemini-2.0-flash-001" ,
774
760
"gemini-2.0-flash-lite-001" ,
775
761
),
@@ -797,11 +783,7 @@ def test_llm_gemini_pro_score_params(llm_fine_tune_df_default_index, model_name)
797
783
798
784
@pytest .mark .parametrize (
799
785
"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" ,),
805
787
)
806
788
def test_gemini_preview_model_warnings (model_name ):
807
789
with pytest .warns (exceptions .PreviewWarning ):
0 commit comments