Spaces:
Build error
Build error
File size: 642 Bytes
5b344d4 76c534f 7011484 bafb458 5ae8333 bafb458 7011484 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from __future__ import annotations
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from huggingface_hub import HfApi
from src.constants import ENV_RESULTS_FILE_DEPTH, MODEL_REPO, MODEL_REPO_TYPE
def get_environments(hf_api: HfApi) -> list[str]:
environments: list[str] = []
files = hf_api.list_repo_files(MODEL_REPO, repo_type=MODEL_REPO_TYPE)
for file in files:
vals = file.split("/")
# e.g. ['models', 'CartPole-v1', 'results.yaml']
if len(vals) == ENV_RESULTS_FILE_DEPTH and vals[2] == "results.yaml" and vals[0] == "models":
environments.append(vals[1])
return environments
|