From 27c29842572bd5a68bdd3a087e7a40392d01bbef Mon Sep 17 00:00:00 2001
From: Felix Kleinert <f.kleinert@fz-juelich.de>
Date: Tue, 6 Apr 2021 10:41:37 +0200
Subject: [PATCH] add dask.compute in build
---
mlair/data_handler/default_data_handler.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/mlair/data_handler/default_data_handler.py b/mlair/data_handler/default_data_handler.py
index 5a5e8d14..02b15993 100644
--- a/mlair/data_handler/default_data_handler.py
+++ b/mlair/data_handler/default_data_handler.py
@@ -64,7 +64,10 @@ class DefaultDataHandler(AbstractDataHandler):
sp_keys = {k: copy.deepcopy(kwargs[k]) for k in cls._requirements if k in kwargs}
sp = cls.data_handler(station, **sp_keys)
dp_args = {k: copy.deepcopy(kwargs[k]) for k in cls.own_args("id_class") if k in kwargs}
- return cls(sp, **dp_args)
+ try:
+ return dask.compute(cls(sp, **dp_args))[0]
+ except Exception:
+ return cls(sp, **dp_args)
def _create_collection(self):
return [self.id_class]
--
GitLab