src/envs.py

import os

from huggingface_hub import HfApi

# Info to change for your repository
# ----------------------------------
TOKEN = os.environ.get("TOKEN")  # A read/write token for your org

OWNER = "open-rl-leaderboard"  # Change to your org - don't forget to create a results and request file

# For evaluations
DEVICE = "cpu"  # "cuda:0" if you add compute, for evaluations
LIMIT = 20  # !!!! Should be None for actual evaluations!!!

# For lighteval evaluations
ACCELERATOR = "cpu"
REGION = "us-east-1"
VENDOR = "aws"
# ----------------------------------

REPO_ID = f"{OWNER}/leaderboard"
RESULTS_REPO = f"{OWNER}/results"

# If you setup a cache later, just change HF_HOME
CACHE_PATH = os.getenv("HF_HOME", ".")

# Local caches
RESULTS_PATH = os.path.join(CACHE_PATH, "eval-results")

REFRESH_RATE = 1 * 60  # 1 min
NUM_LINES_VISUALIZE = 300

API = HfApi(token=TOKEN)