correcting mistake of adding subok keyword to broadcast shapes instea…

…d of broadcast_to
sun-data · May 24, 2024 · dc3b6f0 · dc3b6f0
1 parent b454d27
commit dc3b6f0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/regridding/_regrid/_regrid_from_weights.py b/regridding/_regrid/_regrid_from_weights.py
@@ -52,16 +52,16 @@ def regrid_from_weights(
     :func:`regridding.regrid_from_weights`
     """
 
-    shape_input = np.broadcast_shapes(values_input.shape, shape_input, subok=True)
+    shape_input = np.broadcast_shapes(values_input.shape, shape_input)
 
     ndim_input = len(shape_input)
     axis_input = _util._normalize_axis(axis_input, ndim=ndim_input)
 
     shape_orthogonal = (
         1 if i in axis_input else shape_input[i] for i in range(-len(shape_input), 0)
     )
-    weights = np.broadcast_to(np.array(weights), shape_orthogonal)
-    values_input = np.broadcast_to(values_input, shape_input)
+    weights = np.broadcast_to(np.array(weights), shape_orthogonal, subok=True)
+    values_input = np.broadcast_to(values_input, shape_input, subok=True)
 
     if values_output is None:
         shape_output = np.broadcast_shapes(