Browse code

make Learner2D's datastructures identical before and after pickling

Bas Nijholt authored on 14/04/2020 11:42:53
Showing 1 changed files
... ...
@@ -712,10 +712,12 @@ class Learner2D(BaseLearner):
712 712
             self.function,
713 713
             self.bounds,
714 714
             self.loss_per_triangle,
715
+            self._stack,
715 716
             self._get_data(),
716 717
         )
717 718
 
718 719
     def __setstate__(self, state):
719
-        function, bounds, loss_per_triangle, data = state
720
+        function, bounds, loss_per_triangle, _stack, data = state
720 721
         self.__init__(function, bounds, loss_per_triangle)
721 722
         self._set_data(data)
723
+        self._stack = _stack