Cast acc metrics to float

AmpereComputingAI · Jan 30, 2024 · 0e70de1 · 0e70de1
1 parent 298b8aa
commit 0e70de1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/helpers.py b/utils/helpers.py
@@ -56,5 +56,5 @@ def print_accuracy_metrics(self) -> dict:
             print(f"\n{indent}ACCURACY")
             for metric in accuracy_results.keys():
                 print(f"{3 * indent}{metric}{(max_len - len(metric)) * ' '}{3 * indent}" +
-                      "= {:>7.3f}".format(accuracy_results[metric]))
+                      "= {:>7.3f}".format(float(accuracy_results[metric])))
         return accuracy_results