Skip to content

Commit 54300fb

Browse files
committed
fix.iterator setting fixed to pass the CI tests
1 parent adefd43 commit 54300fb

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

pandas/core/reshape/encoding.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
Hashable,
66
Iterable,
77
)
8-
import itertools
98
from typing import TYPE_CHECKING
109

1110
import numpy as np
@@ -185,7 +184,7 @@ def check_len(item, name: str) -> None:
185184
check_len(prefix_sep, "prefix_sep")
186185

187186
if isinstance(prefix, str):
188-
prefix = itertools.cycle([prefix])
187+
prefix = [prefix] * len(data_to_encode.columns)
189188
if isinstance(prefix, dict):
190189
prefix = [prefix[col] for col in data_to_encode.columns]
191190

@@ -194,7 +193,7 @@ def check_len(item, name: str) -> None:
194193

195194
# validate separators
196195
if isinstance(prefix_sep, str):
197-
prefix_sep = itertools.cycle([prefix_sep])
196+
prefix_sep = [prefix_sep] * len(data_to_encode.columns)
198197
elif isinstance(prefix_sep, dict):
199198
prefix_sep = [prefix_sep[col] for col in data_to_encode.columns]
200199

@@ -212,7 +211,7 @@ def check_len(item, name: str) -> None:
212211
with_dummies = [data.select_dtypes(exclude=dtypes_to_encode)]
213212

214213
for col, pre, sep in zip(
215-
data_to_encode.items(), prefix, prefix_sep, strict=False
214+
data_to_encode.items(), prefix, prefix_sep, strict=True
216215
):
217216
# col is (column_name, column), use just column data here
218217
dummy = _get_dummies_1d(

0 commit comments

Comments
 (0)