Catch TypeError & define i-band in the SLSN kernel with a dummy value

Author: JulienPeloton

fink_science/ztf/superluminous/kernel.py

@@ -19,7 +19,8 @@
 curdir = os.path.dirname(os.path.abspath(__file__))
 
 classifier_path = curdir + "/data/models/superluminous_classifier.joblib"
-band_wave_aa = {1: 4770.0, 2: 6231.0}
+# Declare i-band with dummy value
+band_wave_aa = {1: 4770.0, 2: 6231.0, 3: 0.0}
 temperature = "sigmoid"
 bolometric = "bazin"
 min_points_total = 7

fink_science/ztf/superluminous/slsn_classifier.py

@@ -378,7 +378,7 @@ def fit_rainbow(lc, rainbow_model):
 
         return list(result[:-1]) + list(result[:-1] / errors) + [result[-1]]
 
-    except RuntimeError:
+    except (TypeError, RuntimeError):
         return [np.nan] * (2 * len(rainbow_model.names) + 1)