Skip to content

Commit 82a0c5f

Browse files
test: Fix penguins_linear_model_w_global_explain fixture
1 parent 2526448 commit 82a0c5f

File tree

2 files changed

+11
-10
lines changed

2 files changed

+11
-10
lines changed

tests/system/large/ml/test_linear_model.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,12 +13,22 @@
1313
# limitations under the License.
1414

1515
import pandas as pd
16+
import pytest
1617

17-
from bigframes.ml import model_selection
18+
from bigframes.ml import core, model_selection
1819
import bigframes.ml.linear_model
1920
from 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+
2232
def test_linear_regression_configure_fit_score(penguins_df_default_index, dataset_id):
2333
model = bigframes.ml.linear_model.LinearRegression()
2434

tests/system/small/ml/conftest.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff 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")
9687
def penguins_logistic_model(
9788
session, penguins_logistic_model_name

0 commit comments

Comments
 (0)