From bb92b92c7e6fefa2782f3d8dd9e49db59e0a9911 Mon Sep 17 00:00:00 2001 From: Josue De Santiago Date: Fri, 4 Aug 2023 15:04:11 -0700 Subject: [PATCH] Now the evaluations are running for trainz and fzboost --- src/rail/evaluation/evaluator.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/rail/evaluation/evaluator.py b/src/rail/evaluation/evaluator.py index 88d9ce43..b9fcd9d4 100644 --- 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)