last diff didn't commit properly?

MWATelescope · Feb 14, 2025 · 1a70bd7 · 1a70bd7
1 parent 1e41bce
commit 1a70bd7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/demo/11_allsky.py b/demo/11_allsky.py
@@ -353,6 +353,8 @@ def main():
             continue
     if len(axes_to_average):
         data = np.mean(data, axis=(*axes_to_average,))
+        for ax in sorted(axes_to_average, reverse=True):
+            wcs = wcs.dropaxis(ax)
 
     # With WCS we can write the image to FITS
     hdu = fits.PrimaryHDU(header=wcs.to_header(), data=np.abs(data))