diff --git a/mlair/data_handler/default_data_handler.py b/mlair/data_handler/default_data_handler.py index e4367184830b94f910cf42ae7887585ba958009d..8553aec7883e28c1a958d4551e9690ffeff6ccda 100644 --- a/mlair/data_handler/default_data_handler.py +++ b/mlair/data_handler/default_data_handler.py @@ -85,7 +85,7 @@ class DefaultDataHandler(AbstractDataHandler): self._cleanup() if fresh_store is True else None data = {"X": self._X, "Y": self._Y, "X_extreme": self._X_extreme, "Y_extreme": self._Y_extreme} with open(self._save_file, "wb") as f: - pickle.dump(dask.compute(data), f) + pickle.dump(data, f) logging.debug(f"save pickle data to {self._save_file}") self._reset_data()