Skip to content

Commit 648c512

Browse files
committed
Update if condition
1 parent a627a86 commit 648c512

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

sklbench/datasets/common.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -234,7 +234,7 @@ def preprocess_x(
234234
scaler = MinMaxScaler(feature_range=(0, 1))
235235
else:
236236
logger.warning(f'Unknown "{normalize}" normalization type.')
237-
if scaler is not None and return_type == pd.DataFrame:
237+
if scaler is not None:
238238
return pd.DataFrame(scaler.fit_transform(x), columns=x.columns, index=x.index)
239239
if return_type == np.ndarray:
240240
return x.values

0 commit comments

Comments
 (0)