diff --git a/datacleaner/datacleaner.py b/datacleaner/datacleaner.py index 0fcb87b..3e89ba2 100644 --- a/datacleaner/datacleaner.py +++ b/datacleaner/datacleaner.py @@ -94,7 +94,7 @@ def autoclean(input_dataframe, drop_nans=False, copy=False, encoder=None, column_encoder = LabelEncoder().fit(input_dataframe[column].values) input_dataframe[column] = column_encoder.transform(input_dataframe[column].values) - + input_dataframe.columns = [c.strip() for c in input_dataframe.columns] return input_dataframe def autoclean_cv(training_dataframe, testing_dataframe, drop_nans=False, copy=False, @@ -174,6 +174,8 @@ def autoclean_cv(training_dataframe, testing_dataframe, drop_nans=False, copy=Fa training_dataframe[column] = column_encoder.transform(training_dataframe[column].values) testing_dataframe[column] = column_encoder.transform(testing_dataframe[column].values) + + return training_dataframe, testing_dataframe