LitRL-Inference / src /huggingface /get_environments.py
c-gohlke's picture
Upload folder using huggingface_hub
bafb458
raw
history blame
423 Bytes
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