import os

from huggingface_hub import HfApi

# Info to change for your repository
# ----------------------------------

# A read/write user access token for a user in the org
# You can create these at https://huggingface.co/settings/tokens
# This must be set as a 'Secret' in the leaderboard Space settings.
TOKEN = os.environ.get("TOKEN")

# Change to your org name
OWNER = "Bias-Leaderboard"
DEVICE = "cuda:0" # "cpu" or "cuda:0" if you add compute
LIMIT = 20 # !!!! Should be None for actual evaluations!!!
# ----------------------------------

# Define some input/output variables.
# Don't forget to create a results and requests Dataset for your org
# Leaderboard Space
REPO_ID = f"{OWNER}/leaderboard"
# Leaderboard input Dataset
QUEUE_REPO = f"{OWNER}/requests"
# Leaderboard output Dataset
RESULTS_REPO = f"{OWNER}/results"

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

# Local caches to read previously-submitted/computed results
EVAL_REQUESTS_PATH = os.path.join(CACHE_PATH, "eval-queue")
EVAL_RESULTS_PATH = os.path.join(CACHE_PATH, "eval-results")
EVAL_REQUESTS_PATH_BACKEND = os.path.join(CACHE_PATH, "eval-queue-bk")
EVAL_RESULTS_PATH_BACKEND = os.path.join(CACHE_PATH, "eval-results-bk")

API = HfApi(token=TOKEN)