[Bugfix] fix avg_std_strf

Author: Muqi1029

ai_infra_bench/modes/gen.py

@@ -81,7 +81,9 @@ def gen_export_table(
             row_values.append(f"{row_results[0][feature]:.2f}")
         row_values.append("-")
         for metric in output_metrics:
-            row_values.append(avg_std_strf(metric, row_results, precision=2))
+            row_values.append(
+                avg_std_strf(key=metric, item_list=row_results, precision=2)
+            )
         lines.append("| " + " | ".join(row_values) + " |")
 
     with open(os.path.join(output_dir, TABLE_NAME), mode="w", encoding="utf-8") as f:

ai_infra_bench/utils.py

@@ -11,7 +11,7 @@
 import time
 from dataclasses import dataclass
 from functools import wraps
-from typing import Dict, List
+from typing import Any, Dict, List
 
 import numpy as np
 import psutil
@@ -210,7 +210,7 @@ def read_jsonl(filepath: str):
 
 
 def avg_std_strf(
-    key: str, item_list: List[Dict[str, float]], *, sep=", ", precision: int = None
+    key: str, item_list: List[Dict[str, Any]], *, sep=", ", precision: int = None
 ) -> str:
     val_list = [item[key] for item in item_list]
 
@@ -219,7 +219,7 @@ def avg_std_strf(
     if not isinstance(val_list[0], (int, float)):
         return str(val_list[0])
 
-    if len(val_list) == 1 or (std := np.std(val_list, ddof=1)):
+    if len(val_list) == 1 or (std := np.std(val_list, ddof=1)) == 0:
         return format(val_list[0], fmt)
 
     avg = np.mean(val_list)