Skip to content

Commit e9bfb56

Browse files
committed
Set attribute transformed_names_ when unpickling
Resolves #127.
1 parent b490559 commit e9bfb56

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

sklearn_pandas/dataframe_mapper.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -150,6 +150,7 @@ def __setstate__(self, state):
150150
self.input_df = state.get('input_df', False)
151151
self.built_features = state.get('built_features', self.features)
152152
self.built_default = state.get('built_default', self.default)
153+
self.transformed_names_ = state.get('transformed_names_', [])
153154

154155
def _get_col_subset(self, X, cols, input_df=False):
155156
"""

0 commit comments

Comments
 (0)