Compatibility with lvk

skyreader/plot/plot.py

@@ -691,7 +691,7 @@ def bounding_box(ra, dec, theta, phi):
         plt.legend(fontsize=6, loc="lower left")
 
         # save flattened map
-        equatorial_map, column_names, column_units = prepare_flattened_map(
+        equatorial_map, column_names = prepare_flattened_map(
             equatorial_map, llh_map
         )
         if llh_map:
@@ -701,17 +701,14 @@ def bounding_box(ra, dec, theta, phi):
         filename_main = f"{unique_id}.skymap_nside_{mmap_nside}_{type_map}"
         healpy.write_map(
             self.output_dir / f"{filename_main}.fits.gz",
-            equatorial_map / healpy.nside2pixarea(
-                max_nside, degrees=False,
-            ),
+            equatorial_map,
             coord='C',
             column_names=column_names,
-            column_units=column_units,
             extra_header=fits_header,
             overwrite=True
         )
-        multiorder_map = prepare_multiorder_map(
-            grid_value, uniq_array, llh_map,
+        multiorder_map, column_names = prepare_multiorder_map(
+            grid_value, uniq_array, llh_map, column_names
         )
         multiorder_map.write_map(
             self.output_dir / f"{filename_main}.multiorder.fits.gz",

skyreader/utils/handle_map_data.py

@@ -369,21 +369,20 @@ def prepare_flattened_map(
     """
     if llh_map:
         column_names = ['2DLLH']
-        column_units = None
     else:
         # avoid excessively heavy data format for the flattened map
         equatorial_map[equatorial_map < 1e-16] = np.nanmean(
             equatorial_map[equatorial_map < 1e-16]
         )
-        column_names = ["PROBDENSITY"]
-        column_units = ["sr-1"]
-    return equatorial_map, column_names, column_units
+        column_names = ["PROBABILITY"]
+    return equatorial_map, column_names
 
 
 def prepare_multiorder_map(
     grid_value: np.ndarray,
     uniq_array: np.ndarray,
     llh_map: bool,
+    column_names: List[str]
 ) -> mhealpy.HealpixMap:
     """
     Create the mhealpix map that needs to be saved keeping
@@ -406,4 +405,5 @@ def prepare_multiorder_map(
             uniq_array,
             unit="sr-1"
         )
-    return multiorder_map
+        column_names = ["PROBDENSITY"]
+    return multiorder_map, column_names