Skip to content

Commit

Permalink
make DataSaver picklable
Browse files Browse the repository at this point in the history
  • Loading branch information
basnijholt committed Apr 9, 2020
1 parent 2a88cff commit b2965d8
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions adaptive/learner/data_saver.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,18 @@ def _set_data(self, data):
learner_data, self.extra_data = data
self.learner._set_data(learner_data)

def __getstate__(self):
return (
self.learner,
self.arg_picker,
self._get_data(),
)

def __setstate__(self, state):
learner, arg_picker, data = state
self.__init__(learner, arg_picker)
self._set_data(data)

@copy_docstring_from(BaseLearner.save)
def save(self, fname, compress=True):
# We copy this method because the 'DataSaver' is not a
Expand Down

0 comments on commit b2965d8

Please sign in to comment.