Skip to content

Commit a68cc0a

Browse files
authored
fix(medcat-trainer): Fix issue with metrics failing due to missing meta values (#563)
* fix(medcat-trainer): Fix issue with metrics failing due to missing meta values * Add more details to logged warning regarding no meta values
1 parent 60a86d7 commit a68cc0a

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

medcat-trainer/webapp/api/api/metrics.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -337,7 +337,14 @@ def full_annotation_df(self) -> pd.DataFrame:
337337
continue
338338
meta_model = _meta_models[0]
339339
meta_results = self._eval(meta_model, self.mct_export)
340-
meta_values = {v: k for k, v in meta_results['meta_values'].items()}
340+
if 'meta_values' not in meta_results:
341+
logger.warning(
342+
"No meta_values found for task %s. 'meta_values' no in results, "
343+
"available keys: %s. Reverting to empty results.",
344+
meta_model_task, list(meta_results.keys()))
345+
meta_values = {}
346+
else:
347+
meta_values = {v: k for k, v in meta_results['meta_values'].items()}
341348
pred_meta_values = []
342349
counter = 0
343350
for meta_value in meta_df[meta_model_task]:

0 commit comments

Comments
 (0)