add kilonova score to csv

nuztf/base_scanner.py

@@ -23,9 +23,6 @@
 from astropy.time import Time
 from gwemopt.ztf_tiling import get_quadrant_ipix
 from matplotlib.backends.backend_pdf import PdfPages
-from tqdm import tqdm
-from ztfquery import fields as ztfquery_fields
-
 from nuztf.ampel_api import (
     ampel_api_acknowledge_chunk,
     ampel_api_cone,
@@ -41,6 +38,8 @@
 from nuztf.observations import get_obs_summary
 from nuztf.plot import lightcurve_from_alert
 from nuztf.utils import cosmo
+from tqdm import tqdm
+from ztfquery import fields as ztfquery_fields
 
 DEBUG = False
 RATELIMIT_CALLS = 10
@@ -551,14 +550,25 @@ def create_overview_table(self, outfile=None):
         mags = []
         crossmatches = []
 
-        data = {"ztf_id": [], "RA": [], "Dec": [], "mag": [], "xmatch": []}
+        data = {
+            "ztf_id": [],
+            "RA": [],
+            "Dec": [],
+            "mag": [],
+            "xmatch": [],
+            "kilonova_score": [],
+        }
 
         for ztf_id, alert in tqdm(sorted(self.cache.items())):
             data["ztf_id"].append(ztf_id)
             data["RA"].append(alert["candidate"]["ra"])
             data["Dec"].append(alert["candidate"]["dec"])
             data["mag"].append(alert["candidate"]["magpsf"])
             data["xmatch"].append(get_cross_match_info(raw=alert, logger=self.logger))
+            if alert.get("kilonova_eval") is not None:
+                data["kilonova_score"].append(alert["kilonova_eval"]["kilonovaness"])
+            else:
+                data["kilonova_score"].append(None)
 
         df = pandas.DataFrame.from_dict(data)