... | ... |
@@ -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 |