Spaces:

c-gohlke
/

LitRL-Inference

Build error

LitRL-Inference / src /huggingface /get_environments.py

Upload folder using huggingface_hub

5b344d4 verified almost 3 years ago

642 Bytes

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

	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