Call compute for each kube if necessery

CMCC-Foundation · Jan 19, 2024 · 1836d60 · 1836d60
1 parent f1771da
commit 1836d60
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/drivers/intake_geokube/base.py b/drivers/intake_geokube/base.py
@@ -98,7 +98,11 @@ def _process_geokube_dataset(
         )
         if not query:
             self.log.info("query is empty!")
-            return dataset.compute() if compute else dataset
+            if compute:
+                return dataset.apply(
+                    lambda dc: dc.compute() if isinstance(dc, Delayed) else dc
+                )            
+            return dataset
         if isinstance(dataset, Dataset):
             self.log.info("filtering with: %s", query.filters)
             dataset = dataset.filter(**query.filters)