Deploy Space: update src

src/utils.py

@@ -122,6 +122,17 @@ def load_all_results(root_dir: str = "results") -> pd.DataFrame:
         return pd.DataFrame(columns=RESULT_COLUMNS)
     df = df.sort_values(by=["model", "dataset"]).reset_index(drop=True)
     df["model"] = df["model"].replace(LEGACY_MODEL_NAMES)
+    
+    ALLOWED_MODELS = {
+        "Toto-2.0-1B",
+        "Moirai-1.1-Large",
+        "Chronos-2",
+        "Timer-S1",
+        "TiRex",
+        "TimesFM-2.5"
+    }
+    df = df[df["model"].isin(ALLOWED_MODELS)]
+
     parts = df["dataset"].astype(str).str.split("/", expand=True)
     if parts.shape[1] < 3:
         for idx in range(parts.shape[1], 3):
@@ -627,9 +638,10 @@ def _today_start_utc8() -> datetime:
 
 def get_dataset_last_eval_time(ds_base: str, root_dir: str = "results") -> datetime | None:
     latest_dt: datetime | None = None
+    ALLOWED_SLUGS = {"chronos_2", "moirai_1_1_large", "timer_s1", "timesfm_2_5", "tirex", "toto_2_0_1b"}
     # Scan model directories
     for model_dir in Path(root_dir).iterdir():
-        if not model_dir.is_dir():
+        if not model_dir.is_dir() or model_dir.name not in ALLOWED_SLUGS:
             continue
         forecasts_dir = model_dir / "forecasts"
         if not forecasts_dir.is_dir():
@@ -708,7 +720,10 @@ def make_domain_pie_chart(root_dir: str = "results") -> object | None:
     # Fallback to current snapshots count if history is empty or fails
     if not timestamp_counts:
         seen_datasets: set[str] = set()
+        ALLOWED_SLUGS = {"chronos_2", "moirai_1_1_large", "timer_s1", "timesfm_2_5", "tirex", "toto_2_0_1b"}
         for model_dir in sorted(Path(root_dir).iterdir()):
+            if not model_dir.is_dir() or model_dir.name not in ALLOWED_SLUGS:
+                continue
             forecasts_dir = model_dir / "forecasts"
             if not forecasts_dir.is_dir():
                 continue