Skip to content

Commit e691ffa

Browse files
committed
chore: renamed class to avoid name conflict in sphinx
1 parent ae37bd0 commit e691ffa

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

packages/bigframes/bigframes/ml/model_selection.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
from itertools import chain
2222
import time
2323
import typing
24+
from itertools import chain as _chain
2425
from typing import Generator, List, Optional, Union
2526

2627
import bigframes_vendored.sklearn.model_selection._split as vendored_model_selection_split
@@ -72,7 +73,7 @@ def train_test_split(
7273
test_rows = total_rows - train_rows
7374

7475
return list(
75-
chain.from_iterable(
76+
_chain.from_iterable(
7677
[
7778
[bf_array.head(train_rows), bf_array.tail(test_rows)]
7879
for bf_array in bf_arrays

0 commit comments

Comments
 (0)