Cast tuples to ndarrays

andykee · Dec 21, 2023 · 3d55b88 · 3d55b88
1 parent e10b4e8
commit 3d55b88
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lentil/propagate.py b/lentil/propagate.py
@@ -172,8 +172,8 @@ def propagate_dft(wavefront, pixelscale, shape=None, prop_shape=None,
 
     ptype_out = _propagate_ptype(wavefront.ptype, method='fraunhofer')
 
-    shape = wavefront.shape if shape is None else np.broadcast_to(shape, (2,))
-    prop_shape = shape if prop_shape is None else np.broadcast_to(prop_shape, (2,))
+    shape = np.asarray(wavefront.shape) if shape is None else np.broadcast_to(shape, (2,))
+    prop_shape = np.asarray(shape) if prop_shape is None else np.broadcast_to(prop_shape, (2,))
     shape_out = shape * oversample
     prop_shape_out = prop_shape * oversample