src/envs.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# flake8: noqa E501

import os

from huggingface_hub import HfApi

# Info to change for your repository
# ----------------------------------
TOKEN = os.environ.get("HF_TOKEN")
DISCORD_WEBHOOK_URL = os.environ.get("DISCORD_WEBHOOK_URL")
OWNER = os.environ.get("OWNER")
# ----------------------------------

REPO_ID = os.environ.get("REPO_ID")
REQUESTS_REPO = os.environ.get("REQUESTS_REPO")
RESULTS_REPO = os.environ.get("RESULTS_REPO")

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

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

API = HfApi(token=TOKEN)