Deploy Space: update src

src/utils.py

@@ -672,6 +672,12 @@ def make_domain_pie_chart(root_dir: str = "results") -> object | None:
     for _, row in props.iterrows():
         domain_map[str(row["dataset"]).lower()] = display_domain(str(row["domain"]))
 
+    def get_domain(ds_name: str) -> str:
+        ds_lower = ds_name.lower()
+        if ds_lower not in domain_map:
+            domain_map[ds_lower] = display_domain(lookup_dataset_domain(ds_name, root_dir))
+        return domain_map[ds_lower]
+
     today_start = _today_start_utc8()
     utc8 = today_start.tzinfo
 
@@ -693,7 +699,7 @@ def make_domain_pie_chart(root_dir: str = "results") -> object | None:
                         if eval_dt.astimezone(utc8) >= today_start:
                             ds_base = entry.get("dataset", "")
                             pred_len = entry.get("prediction_length", 0)
-                            domain = domain_map.get(ds_base.lower(), "Other")
+                            domain = get_domain(ds_base)
                             timestamp_counts[domain] = timestamp_counts.get(domain, 0) + pred_len
                             domain_datasets.setdefault(domain, set()).add(dataset_short_name(ds_base))
         except Exception:
@@ -727,7 +733,7 @@ def make_domain_pie_chart(root_dir: str = "results") -> object | None:
                     continue
                 seen_datasets.add(ds_base)
 
-                domain = domain_map.get(ds_base.lower(), "Other")
+                domain = get_domain(ds_base)
                 n_ts = len(snap.get("context", [])) + len(snap.get("actuals", []))
                 timestamp_counts[domain] = timestamp_counts.get(domain, 0) + n_ts
                 domain_datasets.setdefault(domain, set()).add(dataset_short_name(ds_base))