From fc6625d210e59925194f2d0f239bc78b0f7f653a Mon Sep 17 00:00:00 2001
From: Felix Kleinert <f.kleinert@fz-juelich.de>
Date: Fri, 5 Mar 2021 13:53:09 +0100
Subject: [PATCH] remove dask from defaultdh
---
mlair/data_handler/default_data_handler.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mlair/data_handler/default_data_handler.py b/mlair/data_handler/default_data_handler.py
index e4367184..8553aec7 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()
--
GitLab