Checked logs

.gitignore

@@ -1,14 +1,15 @@
-.env
-# Temp validation results
-validation_output.tmp
-venv/
-temp/
-__pycache__/
-*/__pycache__/
-.venv/
-.env/
-train/models/
-bess_visualization.png
-*.pth
-*.safetensors
-*.png
+.env
+# Temp validation results
+validation_output.tmp
+venv/
+temp/
+__pycache__/
+*/__pycache__/
+.venv/
+.env/
+train/models/
+bess_visualization.png
+*.pth
+*.safetensors
+*.png
+scratch/

inference.py

@@ -45,9 +45,9 @@ try:
 except ImportError:
     pass
 
-API_KEY       = os.getenv("HF_TOKEN") or os.getenv("API_KEY", "")
 API_BASE_URL  = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
 MODEL_NAME    = os.getenv("MODEL_NAME", "meta-llama/Llama-3.1-8B-Instruct")
+HF_TOKEN      = os.getenv("HF_TOKEN")
 BENCHMARK     = "bess-rl"
 TASKS         = ["easy", "medium", "hard"]
 MAX_STEPS     = 168        # 1-week simulation - well within 20-minute runtime
@@ -208,7 +208,7 @@ async def run_task(client: OpenAI, task: str) -> float:
 # Entry point
 # ---------------------------------------------------------------------------
 async def main() -> None:
-    client = OpenAI(base_url=API_BASE_URL, api_key=API_KEY)
+    client = OpenAI(base_url=API_BASE_URL, api_key=HF_TOKEN)
     for task in TASKS:
         await run_task(client, task)