浏览代码

fixed simulator type label

Snej69420 4 月之前
父节点
当前提交
d63e40319b
共有 1 个文件被更改,包括 1 次插入 和 4 次删除
  1. 1 4
      benchmarks/compare.py

+ 1 - 4
benchmarks/compare.py

@@ -43,10 +43,7 @@ def get_style_config(user_input):
 
 def get_sim_shorthand(sim_id):
     sim_id_lower = str(sim_id).lower()
-    for key, shorthand in SIMULATOR_MAPPING.items():
-        if key.lower() in sim_id_lower:
-            return shorthand
-    return f"({sim_id})"
+    return SIMULATOR_MAPPING.get(sim_id_lower, f"({sim_id})")
 
 def clean_data(df):
     df["total_time"] = (