File size: 423 Bytes
bafb458
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11


def get_environments() -> 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