Change "auto" behaviour for dask mode

If driver exposes dask_reader default mode to "concurrency".
opendatacube · Jun 25, 2024 · 70a8e20 · 70a8e20
1 parent fc278b3
commit 70a8e20
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/odc/loader/_builder.py b/odc/loader/_builder.py
@@ -179,8 +179,11 @@ def __init__(
         chunks = {**chunks}
         chunks.update(dict(zip(["time", "y", "x"], chunk_tyx)))
         if mode == "auto":
-            # "mem" unless overwriten by env var
-            mode = _default_dask_mode()
+            if rdr.dask_reader is not None:
+                mode = "concurrency"
+            else:
+                # "mem" unless overwritten by env var
+                mode = _default_dask_mode()
 
         self.cfg = cfg
         self.template = template