@@ -26,6 +26,9 @@ def test_create_text_generator_model(palm2_text_generator_model):
2626 assert palm2_text_generator_model ._bqml_model is not None
2727
2828
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+ )
2932@pytest .mark .flaky (retries = 2 , delay = 120 )
3033def test_create_text_generator_model_default_session (bq_connection , llm_text_pandas_df ):
3134 import bigframes .pandas as bpd
@@ -48,6 +51,9 @@ def test_create_text_generator_model_default_session(bq_connection, llm_text_pan
4851 assert all (series .str .len () > 20 )
4952
5053
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+ )
5157@pytest .mark .flaky (retries = 2 , delay = 120 )
5258def test_create_text_generator_model_default_connection (llm_text_pandas_df ):
5359 from bigframes import _config
@@ -74,6 +80,9 @@ def test_create_text_generator_model_default_connection(llm_text_pandas_df):
7480
7581
7682# 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+ )
7786@pytest .mark .flaky (retries = 2 , delay = 120 )
7887def test_text_generator_predict_default_params_success (
7988 palm2_text_generator_model , llm_text_df
@@ -85,6 +94,9 @@ def test_text_generator_predict_default_params_success(
8594 assert all (series .str .len () > 20 )
8695
8796
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+ )
88100@pytest .mark .flaky (retries = 2 , delay = 120 )
89101def test_text_generator_predict_series_default_params_success (
90102 palm2_text_generator_model , llm_text_df
@@ -96,6 +108,9 @@ def test_text_generator_predict_series_default_params_success(
96108 assert all (series .str .len () > 20 )
97109
98110
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+ )
99114@pytest .mark .flaky (retries = 2 , delay = 120 )
100115def test_text_generator_predict_arbitrary_col_label_success (
101116 palm2_text_generator_model , llm_text_df
@@ -108,6 +123,9 @@ def test_text_generator_predict_arbitrary_col_label_success(
108123 assert all (series .str .len () > 20 )
109124
110125
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+ )
111129@pytest .mark .flaky (retries = 2 , delay = 120 )
112130def test_text_generator_predict_with_params_success (
113131 palm2_text_generator_model , llm_text_df
@@ -139,6 +157,9 @@ def test_create_text_embedding_generator_model_defaults(bq_connection):
139157 assert model ._bqml_model is not None
140158
141159
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+ )
142163@pytest .mark .flaky (retries = 2 , delay = 120 )
143164def test_embedding_generator_predict_success (
144165 palm2_embedding_generator_model , llm_text_df
@@ -152,6 +173,9 @@ def test_embedding_generator_predict_success(
152173 assert value .size == 768
153174
154175
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+ )
155179@pytest .mark .flaky (retries = 2 , delay = 120 )
156180def test_embedding_generator_predict_series_success (
157181 palm2_embedding_generator_model , llm_text_df
0 commit comments