File tree Expand file tree Collapse file tree 2 files changed +11
-10
lines changed
Expand file tree Collapse file tree 2 files changed +11
-10
lines changed Original file line number Diff line number Diff line change 1313# limitations under the License.
1414
1515import pandas as pd
16+ import pytest
1617
17- from bigframes .ml import model_selection
18+ from bigframes .ml import core , model_selection
1819import bigframes .ml .linear_model
1920from bigframes .testing import utils
2021
2122
23+ @pytest .fixture (scope = "function" )
24+ def penguins_linear_model_w_global_explain (
25+ penguins_bqml_linear_model : core .BqmlModel ,
26+ ) -> bigframes .ml .linear_model .LinearRegression :
27+ bf_model = bigframes .ml .linear_model .LinearRegression (enable_global_explain = True )
28+ bf_model ._bqml_model = penguins_bqml_linear_model
29+ return bf_model
30+
31+
2232def test_linear_regression_configure_fit_score (penguins_df_default_index , dataset_id ):
2333 model = bigframes .ml .linear_model .LinearRegression ()
2434
Original file line number Diff line number Diff line change @@ -83,15 +83,6 @@ def ephemera_penguins_linear_model(
8383 return bf_model
8484
8585
86- @pytest .fixture (scope = "function" )
87- def penguins_linear_model_w_global_explain (
88- penguins_bqml_linear_model : core .BqmlModel ,
89- ) -> linear_model .LinearRegression :
90- bf_model = linear_model .LinearRegression (enable_global_explain = True )
91- bf_model ._bqml_model = penguins_bqml_linear_model
92- return bf_model
93-
94-
9586@pytest .fixture (scope = "session" )
9687def penguins_logistic_model (
9788 session , penguins_logistic_model_name
You can’t perform that action at this time.
0 commit comments