Browse code

make BalancingLearner picklable

Bas Nijholt authored on 09/04/2020 22:37:47
Showing 1 changed files
... ...
@@ -440,3 +440,16 @@ class BalancingLearner(BaseLearner):
440 440
     def _set_data(self, data):
441 441
         for l, _data in zip(self.learners, data):
442 442
             l._set_data(_data)
443
+
444
+    def __getstate__(self):
445
+        return (
446
+            self.learners,
447
+            self._cdims_default,
448
+            self.strategy,
449
+            self._get_data(),
450
+        )
451
+
452
+    def __setstate__(self, state):
453
+        learners, cdims, strategy, data = state
454
+        self.__init__(learners, cdims=cdims, strategy=strategy)
455
+        self._set_data(data)