Fix 2D arrays (#16)

AlecThomson · Feb 5, 2025 · aef0458 · aef0458
1 parent 54f083c
commit aef0458
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rm_lite/utils/synthesis.py b/rm_lite/utils/synthesis.py
@@ -699,6 +699,8 @@ def rmsynth_nufft(
                 complex_pol_arr.shape[1] * complex_pol_arr.shape[2],
             ),
         )
+    else:
+        complex_pol_arr_2d = complex_pol_arr
 
     # Create a complex polarised cube, B&dB Eqns. (8) and (14)
     # Array has dimensions [nFreq, nY * nX]