paza-bench / src /envs.py
muchai-mercy's picture
pazabench version 2 release
2cdca7d
Raw
History Blame Contribute Delete
1.05 kB
import os
from dotenv import load_dotenv
from huggingface_hub import HfApi
load_dotenv()
TOKEN = os.environ.get("HF_TOKEN")
OWNER = os.environ.get("OWNER")
REPO_ID = os.environ.get("REPO_ID")
QUEUE_REPO = os.environ.get("SUBMISSIONS_REPO")
RESULTS_REPO = os.environ.get("RESULTS_REPO")
# If you setup a cache later, just change HF_HOME
CACHE_PATH=os.getenv("HF_HOME", ".")
# Local caches
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")
# Only create API if token is available
API = HfApi(token=TOKEN) if TOKEN else None
# Flag to enable/disable HuggingFace Hub features
HF_ENABLED = TOKEN is not None and os.environ.get("DISABLE_HF_HUB", "").lower() != "true"
# Telemetry: Set TELEMETRY_ENABLED=true to enable Azure Application Insights
TELEMETRY_ENABLED = os.environ.get("TELEMETRY_ENABLED", "false").lower() == "true"