self multiplication issue with dask array

NREL · Nov 20, 2024 · 8027b19 · 8027b19
1 parent 420c229
commit 8027b19
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/sup3r/bias/bias_transforms.py b/sup3r/bias/bias_transforms.py
@@ -1094,8 +1094,9 @@ def local_presrat_bc(
         # If no trend, it doesn't make sense to correct for zero rate or
         # apply the k-factor, but limit to QDM only.
         if not no_trend:
-            subset = np.where(subset < bias_tau_fut, 0, subset)
-            subset *= k_factor[:, :, nt : nt + 1]
+            subset = np.where(
+                subset < bias_tau_fut, 0, subset * k_factor[:, :, nt : nt + 1]
+            )
 
         data_unbiased[:, :, subset_idx] = subset
 

diff --git a/sup3r/utilities/interpolation.py b/sup3r/utilities/interpolation.py
@@ -65,7 +65,17 @@ def _lin_interp(cls, lev_samps, var_samps, level):
             0,
             da.map_blocks(lambda x, y: x / y, (level - lev_samps[0]), diff),
         )
-        return (1 - alpha) * var_samps[0] + alpha * var_samps[1]
+        indices = 'ijk'[:lev_samps[0].ndim]
+        return da.blockwise(
+            lambda x, y, a: x * (1 - a) + y * a,
+            indices,
+            var_samps[0],
+            indices,
+            var_samps[1],
+            indices,
+            alpha,
+            indices,
+        )
 
     @classmethod
     def _log_interp(cls, lev_samps, var_samps, level):