do not initialize child-learners twice in DataSaver

python-adaptive · Apr 23, 2020 · 29aea69 · 29aea69
1 parent 5308939
commit 29aea69
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/adaptive/learner/data_saver.py b/adaptive/learner/data_saver.py
@@ -55,13 +55,13 @@ def __getstate__(self):
         return (
             self.learner,
             self.arg_picker,
-            self._get_data(),
+            self.extra_data,
         )
 
     def __setstate__(self, state):
-        learner, arg_picker, data = state
+        learner, arg_picker, extra_data = state
         self.__init__(learner, arg_picker)
-        self._set_data(data)
+        self.extra_data = extra_data
 
     @copy_docstring_from(BaseLearner.save)
     def save(self, fname, compress=True):