@@ -49,7 +49,7 @@ def test_create_text_generator_32k_model(
49
49
assert reloaded_model .connection_name == bq_connection
50
50
51
51
52
- @pytest .mark .flaky (retries = 2 , delay = 120 )
52
+ @pytest .mark .flaky (retries = 2 )
53
53
def test_create_text_generator_model_default_session (
54
54
bq_connection , llm_text_pandas_df , bigquery_client
55
55
):
@@ -76,7 +76,7 @@ def test_create_text_generator_model_default_session(
76
76
assert all (series .str .len () > 20 )
77
77
78
78
79
- @pytest .mark .flaky (retries = 2 , delay = 120 )
79
+ @pytest .mark .flaky (retries = 2 )
80
80
def test_create_text_generator_32k_model_default_session (
81
81
bq_connection , llm_text_pandas_df , bigquery_client
82
82
):
@@ -103,7 +103,7 @@ def test_create_text_generator_32k_model_default_session(
103
103
assert all (series .str .len () > 20 )
104
104
105
105
106
- @pytest .mark .flaky (retries = 2 , delay = 120 )
106
+ @pytest .mark .flaky (retries = 2 )
107
107
def test_create_text_generator_model_default_connection (
108
108
llm_text_pandas_df , bigquery_client
109
109
):
@@ -131,7 +131,7 @@ def test_create_text_generator_model_default_connection(
131
131
132
132
133
133
# Marked as flaky only because BQML LLM is in preview, the service only has limited capacity, not stable enough.
134
- @pytest .mark .flaky (retries = 2 , delay = 120 )
134
+ @pytest .mark .flaky (retries = 2 )
135
135
def test_text_generator_predict_default_params_success (
136
136
palm2_text_generator_model , llm_text_df
137
137
):
@@ -142,7 +142,7 @@ def test_text_generator_predict_default_params_success(
142
142
assert all (series .str .len () > 20 )
143
143
144
144
145
- @pytest .mark .flaky (retries = 2 , delay = 120 )
145
+ @pytest .mark .flaky (retries = 2 )
146
146
def test_text_generator_predict_series_default_params_success (
147
147
palm2_text_generator_model , llm_text_df
148
148
):
@@ -153,7 +153,7 @@ def test_text_generator_predict_series_default_params_success(
153
153
assert all (series .str .len () > 20 )
154
154
155
155
156
- @pytest .mark .flaky (retries = 2 , delay = 120 )
156
+ @pytest .mark .flaky (retries = 2 )
157
157
def test_text_generator_predict_arbitrary_col_label_success (
158
158
palm2_text_generator_model , llm_text_df
159
159
):
@@ -165,7 +165,7 @@ def test_text_generator_predict_arbitrary_col_label_success(
165
165
assert all (series .str .len () > 20 )
166
166
167
167
168
- @pytest .mark .flaky (retries = 2 , delay = 120 )
168
+ @pytest .mark .flaky (retries = 2 )
169
169
def test_text_generator_predict_with_params_success (
170
170
palm2_text_generator_model , llm_text_df
171
171
):
@@ -255,7 +255,7 @@ def test_create_text_embedding_generator_multilingual_model_defaults(bq_connecti
255
255
assert model ._bqml_model is not None
256
256
257
257
258
- @pytest .mark .flaky (retries = 2 , delay = 120 )
258
+ @pytest .mark .flaky (retries = 2 )
259
259
def test_embedding_generator_predict_success (
260
260
palm2_embedding_generator_model , llm_text_df
261
261
):
@@ -267,7 +267,7 @@ def test_embedding_generator_predict_success(
267
267
assert len (value ) == 768
268
268
269
269
270
- @pytest .mark .flaky (retries = 2 , delay = 120 )
270
+ @pytest .mark .flaky (retries = 2 )
271
271
def test_embedding_generator_multilingual_predict_success (
272
272
palm2_embedding_generator_multilingual_model , llm_text_df
273
273
):
@@ -279,7 +279,7 @@ def test_embedding_generator_multilingual_predict_success(
279
279
assert len (value ) == 768
280
280
281
281
282
- @pytest .mark .flaky (retries = 2 , delay = 120 )
282
+ @pytest .mark .flaky (retries = 2 )
283
283
def test_embedding_generator_predict_series_success (
284
284
palm2_embedding_generator_model , llm_text_df
285
285
):
@@ -306,7 +306,7 @@ def test_create_gemini_text_generator_model(
306
306
assert reloaded_model .connection_name == bq_connection
307
307
308
308
309
- @pytest .mark .flaky (retries = 2 , delay = 120 )
309
+ @pytest .mark .flaky (retries = 2 )
310
310
def test_gemini_text_generator_predict_default_params_success (
311
311
gemini_text_generator_model , llm_text_df
312
312
):
@@ -317,7 +317,7 @@ def test_gemini_text_generator_predict_default_params_success(
317
317
assert all (series .str .len () > 20 )
318
318
319
319
320
- @pytest .mark .flaky (retries = 2 , delay = 120 )
320
+ @pytest .mark .flaky (retries = 2 )
321
321
def test_gemini_text_generator_predict_with_params_success (
322
322
gemini_text_generator_model , llm_text_df
323
323
):
0 commit comments