diff options
Diffstat (limited to 'fg21sim')
| -rw-r--r-- | fg21sim/sky.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/fg21sim/sky.py b/fg21sim/sky.py index 8f15ca8..2c176ff 100644 --- a/fg21sim/sky.py +++ b/fg21sim/sky.py @@ -171,9 +171,11 @@ class SkyPatch: os.makedirs(outdir) logger.info("Created output directory: %s" % outdir) image = self.data.reshape(self.ysize, self.xsize) - header = self.wcs.to_header() - header.extend(self.header, update=True) - hdu = fits.PrimaryHDU(data=image, header=self.header) + if hasattr(self, "header"): + header = self.header.copy(strip=True) + wcs_header = self.wcs.to_header() + header.extend(wcs_header, update=True) + hdu = fits.PrimaryHDU(data=image, header=header) hdu.writeto(outfile, clobber=clobber, checksum=checksum) logger.info("Write sky map to file: %s" % outfile) @@ -364,7 +366,9 @@ class SkyHealpix: if not os.path.exists(outdir): os.makedirs(outdir) logger.info("Created output directory: %s" % outdir) - write_fits_healpix(outfile, self.data, header=self.header, + if hasattr(self, "header"): + header = self.header + write_fits_healpix(outfile, self.data, header=header, clobber=clobber, checksum=checksum) logger.info("Write sky map to file: %s" % outfile) |