Browse code
do not initialize child-learners twice in DataSaver
Bas Nijholt authored on 15/04/2020 23:07:41
Showing 1 changed files
| ... | ... |
@@ -55,13 +55,13 @@ class DataSaver: |
| 55 | 55 |
return ( |
| 56 | 56 |
self.learner, |
| 57 | 57 |
self.arg_picker, |
| 58 |
- self._get_data(), |
|
| 58 |
+ self.extra_data, |
|
| 59 | 59 |
) |
| 60 | 60 |
|
| 61 | 61 |
def __setstate__(self, state): |
| 62 |
- learner, arg_picker, data = state |
|
| 62 |
+ learner, arg_picker, extra_data = state |
|
| 63 | 63 |
self.__init__(learner, arg_picker) |
| 64 |
- self._set_data(data) |
|
| 64 |
+ self.extra_data = extra_data |
|
| 65 | 65 |
|
| 66 | 66 |
@copy_docstring_from(BaseLearner.save) |
| 67 | 67 |
def save(self, fname, compress=True): |