diff --git a/mlair/data_handler/default_data_handler.py b/mlair/data_handler/default_data_handler.py index 5a5e8d14dd70e11b9a7f04f3640a86d8b77122f3..02b15993a1a3db66c2b6e7dc564361e959ab9fa1 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]