@@ -26,9 +26,6 @@ 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
- )
32
29
@pytest .mark .flaky (retries = 2 , delay = 120 )
33
30
def test_create_text_generator_model_default_session (bq_connection , llm_text_pandas_df ):
34
31
import bigframes .pandas as bpd
@@ -51,9 +48,6 @@ def test_create_text_generator_model_default_session(bq_connection, llm_text_pan
51
48
assert all (series .str .len () > 20 )
52
49
53
50
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
- )
57
51
@pytest .mark .flaky (retries = 2 , delay = 120 )
58
52
def test_create_text_generator_model_default_connection (llm_text_pandas_df ):
59
53
from bigframes import _config
@@ -80,9 +74,6 @@ def test_create_text_generator_model_default_connection(llm_text_pandas_df):
80
74
81
75
82
76
# 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
- )
86
77
@pytest .mark .flaky (retries = 2 , delay = 120 )
87
78
def test_text_generator_predict_default_params_success (
88
79
palm2_text_generator_model , llm_text_df
@@ -94,9 +85,6 @@ def test_text_generator_predict_default_params_success(
94
85
assert all (series .str .len () > 20 )
95
86
96
87
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
- )
100
88
@pytest .mark .flaky (retries = 2 , delay = 120 )
101
89
def test_text_generator_predict_series_default_params_success (
102
90
palm2_text_generator_model , llm_text_df
@@ -108,9 +96,6 @@ def test_text_generator_predict_series_default_params_success(
108
96
assert all (series .str .len () > 20 )
109
97
110
98
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
- )
114
99
@pytest .mark .flaky (retries = 2 , delay = 120 )
115
100
def test_text_generator_predict_arbitrary_col_label_success (
116
101
palm2_text_generator_model , llm_text_df
@@ -123,9 +108,6 @@ def test_text_generator_predict_arbitrary_col_label_success(
123
108
assert all (series .str .len () > 20 )
124
109
125
110
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
- )
129
111
@pytest .mark .flaky (retries = 2 , delay = 120 )
130
112
def test_text_generator_predict_with_params_success (
131
113
palm2_text_generator_model , llm_text_df
@@ -157,9 +139,6 @@ def test_create_text_embedding_generator_model_defaults(bq_connection):
157
139
assert model ._bqml_model is not None
158
140
159
141
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
- )
163
142
@pytest .mark .flaky (retries = 2 , delay = 120 )
164
143
def test_embedding_generator_predict_success (
165
144
palm2_embedding_generator_model , llm_text_df
@@ -173,9 +152,6 @@ def test_embedding_generator_predict_success(
173
152
assert value .size == 768
174
153
175
154
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
- )
179
155
@pytest .mark .flaky (retries = 2 , delay = 120 )
180
156
def test_embedding_generator_predict_series_success (
181
157
palm2_embedding_generator_model , llm_text_df
0 commit comments