Skip to content
This repository was archived by the owner on Apr 1, 2026. It is now read-only.

Commit e0e0ed7

Browse files
committed
fix import
1 parent ae0536a commit e0e0ed7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

bigframes/bigquery/_operations/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,11 +18,11 @@
1818

1919
import bigframes
2020
from bigframes import dataframe
21-
import bigframes.ml.base
21+
from bigframes.ml import base as ml_base
2222

2323

2424
def get_model_name_and_session(
25-
model: Union[bigframes.ml.base.BaseEstimator, str, pd.Series],
25+
model: Union[ml_base.BaseEstimator, str, pd.Series],
2626
# Other dataframe arguments to extract session from
2727
*dataframes: Optional[Union[pd.DataFrame, dataframe.DataFrame, str]],
2828
) -> tuple[str, Optional[bigframes.session.Session]]:

0 commit comments

Comments
 (0)