diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index e0da7fe..dc50b55 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -485,11 +485,11 @@ def _visualise_metrics(visualise_precision_recall: callable, "max_f1_score_micro": max_f1_score_micro, "precision_at_max_f1_micro": precision_at_max_f1_micro, "recall_at_max_f1_micro": recall_at_max_f1_micro, - "ose_at_max_f1_micro": ose_at_max_f1_micro, + "ose_at_max_f1_micro": int(ose_at_max_f1_micro), "max_f1_score_macro": max_f1_score_macro, "precision_at_max_f1_macro": precision_at_max_f1_macro, "recall_at_max_f1_macro": recall_at_max_f1_macro, - "ose_at_max_f1_macro": ose_at_max_f1_macro, + "ose_at_max_f1_macro": int(ose_at_max_f1_macro), }, file)