fix an indexing error in quickview

bhazelton · bhazelton · commit fab41eb2d58d · 2025-10-28T14:10:04.000-07:00
diff --git a/PyFHD/io/pyfhd_quickview.py b/PyFHD/io/pyfhd_quickview.py
@@ -245,7 +245,7 @@ def quickview(
                     high=np.max(beam_mask_test),
                 )
             beam_mask0 = np.zeros([obs_out["dimension"], obs_out["elements"]])
-            beam_mask0.flat[beam_i] = 1
+            beam_mask0[beam_i] = 1
             beam_avg += beam_base_out[pol_i] ** 2
             beam_mask *= beam_mask0
     beam_avg /= min(obs["n_pol"], 2)

Original file line number	Diff line number	Diff line change
`@@ -245,7 +245,7 @@ def quickview(`
`245`	`245`	`high=np.max(beam_mask_test),`
`246`	`246`	`)`
`247`	`247`	`beam_mask0 = np.zeros([obs_out["dimension"], obs_out["elements"]])`
`248`		`- beam_mask0.flat[beam_i] = 1`
	`248`	`+ beam_mask0[beam_i] = 1`
`249`	`249`	`beam_avg += beam_base_out[pol_i] ** 2`
`250`	`250`	`beam_mask *= beam_mask0`
`251`	`251`	`beam_avg /= min(obs["n_pol"], 2)`