[Bug]: DatetimeInfoExtractor prevents pipeline from being saved using joblib or pickle #257
Closed
1 task done
Labels
bug
Something isn't working
What happened?
The DatetimeInfoExtractor stores the mappings_provided attribute as a .keys() type. If you try and save a pipeline object which has that pipeline step using joblib or pickle you get an error stating that:
"TypeError: cannot pickle 'dict_keys' object".
This attribute should be stored as a list instead. We should maybe also add a test that our transformers can be saved as a pickle file
Environment
scikit-learn == 1.1.3
Python == 3.9.13
joblib == 1.3.2
tubular == 1.1.0
Minimum reproducible code
No response
Relevant error output
TypeError: cannot pickle 'dict_keys' object
Code of Conduct
The text was updated successfully, but these errors were encountered: