src/envs.py

import os

from huggingface_hub import HfApi

# clone / pull the lmeh eval data
TOKEN = os.environ.get("TOKEN", None)

OWNER = "latticeflow"
REPO_ID = f"{OWNER}/compl-ai-board"
QUEUE_REPO = f"{OWNER}/requests"
RESULTS_REPO = f"{OWNER}/results"

CACHE_PATH = os.getenv("HF_HOME", ".")

# Local caches
EVAL_REQUESTS_PATH = os.path.join(CACHE_PATH, "requests")
EVAL_RESULTS_PATH = os.path.join(CACHE_PATH, "results")

API = HfApi(token=TOKEN)