From 0b4db2421c912dfd1373dbea1840419a082671e6 Mon Sep 17 00:00:00 2001 From: Garrett Wu Date: Mon, 23 Dec 2024 19:28:21 +0000 Subject: [PATCH] fix --- .../bigframes_vendored/sklearn/model_selection/_validation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/third_party/bigframes_vendored/sklearn/model_selection/_validation.py b/third_party/bigframes_vendored/sklearn/model_selection/_validation.py index bc4bf90c8a..b93c47ea04 100644 --- a/third_party/bigframes_vendored/sklearn/model_selection/_validation.py +++ b/third_party/bigframes_vendored/sklearn/model_selection/_validation.py @@ -23,8 +23,8 @@ def cross_validate(estimator, X, y=None, *, cv=None): >>> X = bpd.DataFrame({"feat0": [1, 3, 5], "feat1": [2, 4, 6]}) >>> y = bpd.DataFrame({"label": [1, 2, 3]}) >>> model = LinearRegression() - >>> scores = cross_validate(model, X, y, cv=KFold(n_splits=3, random_state=42)) - >>> for score in scores["test_score"]: + >>> scores = cross_validate(model, X, y, cv=3) # doctest: +SKIP + >>> for score in scores["test_score"]: # doctest: +SKIP ... print(score["mean_squared_error"][0]) ... 5.218167286047954e-19