Now the evaluations are running for trainz and fzboost

LSSTDESC · Aug 4, 2023 · bb92b92 · bb92b92
1 parent ad3223a
commit bb92b92
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/rail/evaluation/evaluator.py b/src/rail/evaluation/evaluator.py
@@ -136,4 +136,6 @@ def run(self):
             out_table['CDE_stat'] = [value.statistic]
             out_table['CDE_pval'] = [value.p_value]
 
-        self.add_data('output', out_table)
+	# Converting any possible None to NaN to write it
+        out_table_to_write = {key: np.array(val).astype(float) for key, val in out_table.items()}
+        self.add_data('output', out_table_to_write)