Just call use_future_array_shapes -- if it's already set it's a no-op

RadioAstronomySoftwareGroup · Feb 29, 2024 · 2fe795b · 2fe795b
1 parent ceba263
commit 2fe795b
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/pyuvsim/uvsim.py b/pyuvsim/uvsim.py
@@ -469,8 +469,7 @@ def uvdata_to_task_iter(task_ids, input_uv, catalog, beam_list, beam_dict, Nsky_
         raise TypeError("catalog must be a SkyModelData object.")
 
     # use future array shapes
-    if not input_uv.future_array_shapes:
-        input_uv.use_future_array_shapes()
+    input_uv.use_future_array_shapes()
 
     # Splitting the catalog for memory's sake.
     Nsrcs_total = catalog.Ncomponents
@@ -722,8 +721,7 @@ def run_uvdata_uvsim(
     if not ((input_uv.Npols == 4) and (input_uv.polarization_array.tolist() == [-5, -6, -7, -8])):
         raise ValueError("input_uv must have XX,YY,XY,YX polarization")
 
-    if not input_uv.future_array_shapes:
-        input_uv.use_future_array_shapes()
+    input_uv.use_future_array_shapes()
 
     input_order = input_uv.blt_order
     if input_order != ("time", "baseline"):