Spaces:
Running
Running
Deploy Space: update app.py
Browse files
app.py
CHANGED
|
@@ -46,9 +46,10 @@ def _dataset_tables(grouped: dict, datasets: list[str]) -> tuple[list, list]:
|
|
| 46 |
|
| 47 |
def _get_model_slugs(results_path: str) -> list[str]:
|
| 48 |
"""Return subdirs of results_path that contain a forecasts/ subdirectory."""
|
|
|
|
| 49 |
slugs = []
|
| 50 |
for p in sorted(Path(results_path).iterdir()):
|
| 51 |
-
if p.is_dir() and (p / "forecasts").exists():
|
| 52 |
slugs.append(p.name)
|
| 53 |
return slugs
|
| 54 |
|
|
|
|
| 46 |
|
| 47 |
def _get_model_slugs(results_path: str) -> list[str]:
|
| 48 |
"""Return subdirs of results_path that contain a forecasts/ subdirectory."""
|
| 49 |
+
ALLOWED_SLUGS = {"chronos_2", "moirai_1_1_large", "timer_s1", "timesfm_2_5", "tirex", "toto_2_0_1b"}
|
| 50 |
slugs = []
|
| 51 |
for p in sorted(Path(results_path).iterdir()):
|
| 52 |
+
if p.is_dir() and (p / "forecasts").exists() and p.name in ALLOWED_SLUGS:
|
| 53 |
slugs.append(p.name)
|
| 54 |
return slugs
|
| 55 |
|