diff --git a/medcat-trainer/webapp/api/api/metrics.py b/medcat-trainer/webapp/api/api/metrics.py index b6e17e44b..b421e8381 100644 --- a/medcat-trainer/webapp/api/api/metrics.py +++ b/medcat-trainer/webapp/api/api/metrics.py @@ -337,7 +337,14 @@ def full_annotation_df(self) -> pd.DataFrame: continue meta_model = _meta_models[0] meta_results = self._eval(meta_model, self.mct_export) - meta_values = {v: k for k, v in meta_results['meta_values'].items()} + if 'meta_values' not in meta_results: + logger.warning( + "No meta_values found for task %s. 'meta_values' no in results, " + "available keys: %s. Reverting to empty results.", + meta_model_task, list(meta_results.keys())) + meta_values = {} + else: + meta_values = {v: k for k, v in meta_results['meta_values'].items()} pred_meta_values = [] counter = 0 for meta_value in meta_df[meta_model_task]: