Fix bug in getting required_scenarios

mlcommons · Jan 4, 2025 · 0f27253 · 0f27253
1 parent 58fd6bc
commit 0f27253
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/add_results_summary.py b/add_results_summary.py
@@ -299,8 +299,14 @@ def get_table_header(division, category):
 
                     version = data[model]["Offline"]["version"]
                     acc_target = checker.MODEL_CONFIG[version]["accuracy-target"][model]
-                    required_scenarios_datacenter = checker.MODEL_CONFIG[version]["required-scenarios-datacenter"][model]
-                    required_scenarios_edge = checker.MODEL_CONFIG[version]["required-scenarios-edge"][model]
+                    if model in checker.MODEL_CONFIG[version]["required-scenarios-datacenter"]:
+                        required_scenarios_datacenter = checker.MODEL_CONFIG[version]["required-scenarios-datacenter"][model]
+                    else:
+                        required_scenarios_datacenter = []
+                    if model in checker.MODEL_CONFIG[version]["required-scenarios-edge"]:
+                        required_scenarios_edge = checker.MODEL_CONFIG[version]["required-scenarios-edge"][model]
+                    else:
+                        required_scenarios_edge = []
 
                     i = 0
                     acc_targets = []