Skip to content

Commit e1daff1

Browse files
author
Juan Pablo González
committed
fix formatting issue
1 parent c10bfc2 commit e1daff1

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

Diff for: ml_garden/core/steps/encode.py

+5-6
Original file line numberDiff line numberDiff line change
@@ -374,19 +374,18 @@ def _restore_target_encoded_dtypes(self, encoded_data, encoder):
374374
"""Convert the columns handled by any TargetEncoder in the given encoder to float32."""
375375
for name, transformer, cols in encoder.transformers:
376376
# Direct TargetEncoder
377-
if type(transformer).__name__ == 'TargetEncoder':
377+
if type(transformer).__name__ == "TargetEncoder":
378378
for col in cols:
379-
encoded_data[col] = encoded_data[col].astype('float32')
379+
encoded_data[col] = encoded_data[col].astype("float32")
380380

381381
# Nested transformers (if inside pipelines or additional ColumnTransformers)
382-
elif hasattr(transformer, 'transformers'):
382+
elif hasattr(transformer, "transformers"):
383383
for nested_name, nested_transformer, nested_cols in transformer.transformers:
384-
if type(nested_transformer).__name__ == 'TargetEncoder':
384+
if type(nested_transformer).__name__ == "TargetEncoder":
385385
for col in nested_cols:
386-
encoded_data[col] = encoded_data[col].astype('float32')
386+
encoded_data[col] = encoded_data[col].astype("float32")
387387
return encoded_data
388388

389-
390389
def _convert_float64_to_float32(self, encoded_data: pd.DataFrame) -> pd.DataFrame:
391390
"""Convert float64 columns to float32."""
392391
float64_columns = encoded_data.select_dtypes(include=["float64"]).columns

0 commit comments

Comments
 (0)