@@ -26,6 +26,9 @@ def test_create_text_generator_model(palm2_text_generator_model):
26
26
assert palm2_text_generator_model ._bqml_model is not None
27
27
28
28
29
+ @pytest .mark .skip (
30
+ reason = "Temporarily disable to validate the hypothesis that LLM capacity is causing the presubmit tests to take long to run."
31
+ )
29
32
@pytest .mark .flaky (retries = 2 , delay = 120 )
30
33
def test_create_text_generator_model_default_session (bq_connection , llm_text_pandas_df ):
31
34
import bigframes .pandas as bpd
@@ -48,6 +51,9 @@ def test_create_text_generator_model_default_session(bq_connection, llm_text_pan
48
51
assert all (series .str .len () > 20 )
49
52
50
53
54
+ @pytest .mark .skip (
55
+ reason = "Temporarily disable to validate the hypothesis that LLM capacity is causing the presubmit tests to take long to run."
56
+ )
51
57
@pytest .mark .flaky (retries = 2 , delay = 120 )
52
58
def test_create_text_generator_model_default_connection (llm_text_pandas_df ):
53
59
from bigframes import _config
@@ -74,6 +80,9 @@ def test_create_text_generator_model_default_connection(llm_text_pandas_df):
74
80
75
81
76
82
# Marked as flaky only because BQML LLM is in preview, the service only has limited capacity, not stable enough.
83
+ @pytest .mark .skip (
84
+ reason = "Temporarily disable to validate the hypothesis that LLM capacity is causing the presubmit tests to take long to run."
85
+ )
77
86
@pytest .mark .flaky (retries = 2 , delay = 120 )
78
87
def test_text_generator_predict_default_params_success (
79
88
palm2_text_generator_model , llm_text_df
@@ -85,6 +94,9 @@ def test_text_generator_predict_default_params_success(
85
94
assert all (series .str .len () > 20 )
86
95
87
96
97
+ @pytest .mark .skip (
98
+ reason = "Temporarily disable to validate the hypothesis that LLM capacity is causing the presubmit tests to take long to run."
99
+ )
88
100
@pytest .mark .flaky (retries = 2 , delay = 120 )
89
101
def test_text_generator_predict_series_default_params_success (
90
102
palm2_text_generator_model , llm_text_df
@@ -96,6 +108,9 @@ def test_text_generator_predict_series_default_params_success(
96
108
assert all (series .str .len () > 20 )
97
109
98
110
111
+ @pytest .mark .skip (
112
+ reason = "Temporarily disable to validate the hypothesis that LLM capacity is causing the presubmit tests to take long to run."
113
+ )
99
114
@pytest .mark .flaky (retries = 2 , delay = 120 )
100
115
def test_text_generator_predict_arbitrary_col_label_success (
101
116
palm2_text_generator_model , llm_text_df
@@ -108,6 +123,9 @@ def test_text_generator_predict_arbitrary_col_label_success(
108
123
assert all (series .str .len () > 20 )
109
124
110
125
126
+ @pytest .mark .skip (
127
+ reason = "Temporarily disable to validate the hypothesis that LLM capacity is causing the presubmit tests to take long to run."
128
+ )
111
129
@pytest .mark .flaky (retries = 2 , delay = 120 )
112
130
def test_text_generator_predict_with_params_success (
113
131
palm2_text_generator_model , llm_text_df
@@ -139,6 +157,9 @@ def test_create_text_embedding_generator_model_defaults(bq_connection):
139
157
assert model ._bqml_model is not None
140
158
141
159
160
+ @pytest .mark .skip (
161
+ reason = "Temporarily disable to validate the hypothesis that LLM capacity is causing the presubmit tests to take long to run."
162
+ )
142
163
@pytest .mark .flaky (retries = 2 , delay = 120 )
143
164
def test_embedding_generator_predict_success (
144
165
palm2_embedding_generator_model , llm_text_df
@@ -152,6 +173,9 @@ def test_embedding_generator_predict_success(
152
173
assert value .size == 768
153
174
154
175
176
+ @pytest .mark .skip (
177
+ reason = "Temporarily disable to validate the hypothesis that LLM capacity is causing the presubmit tests to take long to run."
178
+ )
155
179
@pytest .mark .flaky (retries = 2 , delay = 120 )
156
180
def test_embedding_generator_predict_series_success (
157
181
palm2_embedding_generator_model , llm_text_df
0 commit comments