diff --git a/quacc/evaluation/report.py b/quacc/evaluation/report.py index 56019a9..50ff5ad 100644 --- a/quacc/evaluation/report.py +++ b/quacc/evaluation/report.py @@ -280,6 +280,9 @@ class DatasetReport: return np.around([(1.0 - p, p) for p in self.prevs], decimals=2) def add(self, cr: CompReport): + if cr is None: + return + self.crs.append(cr) if self._dict is None: @@ -320,6 +323,11 @@ class DatasetReport: [self.s_dict[col][sp], cr_s_dict[col][sp]] ) + for sp in self.s_prevs: + for col, vals in self.s_dict.items(): + if sp not in vals: + vals[sp] = [] + for k, score in cr.fit_scores.items(): if k not in self.fit_scores: self.fit_scores[k] = []