fix metadata parsing issues with fitsio

apetri · Mar 28, 2016 · 0c69757 · 0c69757
1 parent e8aa3d1
commit 0c69757
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/lenstools/catalog/shear.py b/lenstools/catalog/shear.py
@@ -61,7 +61,21 @@ def read(cls,filename,*args,**kwargs):
 
 			if fitsio is not None:
 				with fitsio.FITS(filename,"r") as hdulist:
-					return cls(hdulist[1].read())
+
+					data = cls(hdulist[1].read())
+					header = hdulist[1].read_header()
+
+					try:
+						data.meta["NGAL"] = header["NGAL"]
+					except ValueError:
+						pass
+
+					try:
+						data.meta["AUNIT"] = header["AUNIT"].replace(" ","")
+					except ValueError:
+						pass
+
+					return data
 
 		return super(Catalog,cls).read(filename,*args,**kwargs)