fix bug

lm-sys · Oct 24, 2024 · b27d617 · b27d617
1 parent 7ca0d43
commit b27d617
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/fastchat/serve/monitor/monitor.py b/fastchat/serve/monitor/monitor.py
@@ -895,6 +895,8 @@ def get_model_name(model_key):
 
     combined_table = []
     for category in elo_results.keys():
+        if category not in key_to_category_name:
+            continue
         df = elo_results[category]["leaderboard_table_df"].copy()
         # remove deprecated models
         df = df.loc[~df.index.isin(deprecated_model_name)]