take the sqrt of the abs columns in CST files for efield beams

RadioAstronomySoftwareGroup · Aug 14, 2023 · c67d02b · c67d02b
1 parent 4ff6808
commit c67d02b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyuvdata/uvbeam/cst_beam.py b/pyuvdata/uvbeam/cst_beam.py
@@ -343,8 +343,8 @@ def read_cst_beam(
             )
             phi_phase = phi_phase.reshape((theta_axis.size, phi_axis.size), order="F")
 
-            theta_beam = theta_mag * np.exp(1j * theta_phase)
-            phi_beam = phi_mag * np.exp(1j * phi_phase)
+            theta_beam = np.sqrt(theta_mag) * np.exp(1j * theta_phase)
+            phi_beam = np.sqrt(phi_mag) * np.exp(1j * phi_phase)
 
             self.data_array[0, 0, 0, 0, :, :] = phi_beam
             self.data_array[1, 0, 0, 0, :, :] = theta_beam