use dask for reference lines/pts timeseries data if available

fema-ffrd · Jul 2, 2024 · ba8cbad · ba8cbad
1 parent 75d8656
commit ba8cbad
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/rashdf/plan.py b/src/rashdf/plan.py
@@ -966,7 +966,13 @@ def reference_timeseries_output(self, reftype: str = "lines") -> xr.Dataset:
             group = reference_group.get(var)
             if group is None:
                 continue
-            values = group[:]
+            try:
+                import dask.array as da
+
+                # TODO: user-specified chunks?
+                values = da.from_array(group, chunks=group.chunks)
+            except ImportError:
+                values = group[:]
             units = group.attrs["Units"].decode("utf-8")
             da = xr.DataArray(
                 values,