ragavrida commited on
Commit
d83dda3
Β·
1 Parent(s): cf9e328

Fix: use LOCAL_IMAGE_NAME (matches submission form)

Browse files
Files changed (1) hide show
  1. inference.py +2 -2
inference.py CHANGED
@@ -6,7 +6,7 @@ MANDATORY
6
  API_BASE_URL The API endpoint for the LLM.
7
  MODEL_NAME The model identifier to use for inference.
8
  HF_TOKEN Your Hugging Face / API key.
9
- IMAGE_NAME The name of the local image to use for the environment if using from_docker_image()
10
 
11
  - The inference script must be named `inference.py` and placed in the root directory of the project
12
  - Participants must use OpenAI Client for all LLM calls using above variables
@@ -50,7 +50,7 @@ from client import CodeReviewEnv
50
 
51
  # ─── Configuration ────────────────────────────────────────────────────────────
52
 
53
- IMAGE_NAME = os.getenv("IMAGE_NAME") # If using from_docker_image()
54
  API_KEY = os.getenv("HF_TOKEN") or os.getenv("API_KEY")
55
 
56
  API_BASE_URL = os.getenv("API_BASE_URL") or "https://router.huggingface.co/v1"
 
6
  API_BASE_URL The API endpoint for the LLM.
7
  MODEL_NAME The model identifier to use for inference.
8
  HF_TOKEN Your Hugging Face / API key.
9
+ IMAGE_NAME The name of the local image to use for the environment if using from_docker_image() (also accepts LOCAL_IMAGE_NAME)
10
 
11
  - The inference script must be named `inference.py` and placed in the root directory of the project
12
  - Participants must use OpenAI Client for all LLM calls using above variables
 
50
 
51
  # ─── Configuration ────────────────────────────────────────────────────────────
52
 
53
+ IMAGE_NAME = os.getenv("LOCAL_IMAGE_NAME") or os.getenv("IMAGE_NAME") # If using from_docker_image()
54
  API_KEY = os.getenv("HF_TOKEN") or os.getenv("API_KEY")
55
 
56
  API_BASE_URL = os.getenv("API_BASE_URL") or "https://router.huggingface.co/v1"