Skip to content

Commit e26ec20

Browse files
authored
chore: disable BQML regression affected tests (#661)
1 parent 16866d2 commit e26ec20

File tree

3 files changed

+9
-0
lines changed

3 files changed

+9
-0
lines changed

tests/system/large/ml/test_core.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,13 @@
1313
# limitations under the License.
1414

1515
import pandas
16+
import pytest
1617

1718
from bigframes.ml import globals
1819

1920

21+
# TODO(garrettwu): Re-enable or not check exact numbers.
22+
@pytest.mark.skip(reason="bqml regression")
2023
def test_bqml_e2e(session, dataset_id, penguins_df_default_index, new_penguins_df):
2124
df = penguins_df_default_index.dropna()
2225
X_train = df[

tests/system/large/ml/test_ensemble.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,8 @@
2020
import bigframes.ml.ensemble
2121

2222

23+
# TODO(garrettwu): Re-enable or not check exact numbers.
24+
@pytest.mark.skip(reason="bqml regression")
2325
@pytest.mark.flaky(retries=2)
2426
def test_xgbregressor_default_params(penguins_df_default_index, dataset_id):
2527
model = bigframes.ml.ensemble.XGBRegressor()

tests/system/large/ml/test_pipeline.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -222,6 +222,8 @@ def test_pipeline_logistic_regression_fit_score_predict(
222222
)
223223

224224

225+
# TODO(garrettwu): Re-enable or not check exact numbers.
226+
@pytest.mark.skip(reason="bqml regression")
225227
@pytest.mark.flaky(retries=2)
226228
def test_pipeline_xgbregressor_fit_score_predict(session, penguins_df_default_index):
227229
"""Test a supervised model with a minimal preprocessing step"""
@@ -297,6 +299,8 @@ def test_pipeline_xgbregressor_fit_score_predict(session, penguins_df_default_in
297299
)
298300

299301

302+
# TODO(garrettwu): Re-enable or not check exact numbers.
303+
@pytest.mark.skip(reason="bqml regression")
300304
@pytest.mark.flaky(retries=2)
301305
def test_pipeline_random_forest_classifier_fit_score_predict(
302306
session, penguins_df_default_index

0 commit comments

Comments
 (0)