umar-sharif821 commited on
Commit
ef02d88
Β·
1 Parent(s): 09e32d2

fix: use os.getenv for env vars

Browse files
Files changed (1) hide show
  1. inference.py +5 -4
inference.py CHANGED
@@ -19,10 +19,11 @@ from env.cache import CDNCacheEnv, TASK_CONFIGS
19
  from env.models import Action, Observation
20
 
21
  # ─────────────────────────────────────────────
22
- # Config from environment
23
  # ─────────────────────────────────────────────
24
- API_BASE_URL = os.environ.get("API_BASE_URL", "https://api.openai.com/v1")
25
- MODEL_NAME = os.environ.get("MODEL_NAME", "gpt-4o-mini")
 
26
  HF_TOKEN = os.environ.get("HF_TOKEN", "")
27
 
28
  if not HF_TOKEN:
@@ -218,4 +219,4 @@ if __name__ == "__main__":
218
  print(f"[INFO] {r['task_id']}: score={r['score']} reward={r['total_reward']}", file=sys.stderr)
219
 
220
  overall = round(sum(r["score"] for r in results) / len(results), 4)
221
- print(f"[INFO] Overall score: {overall}", file=sys.stderr)
 
19
  from env.models import Action, Observation
20
 
21
  # ─────────────────────────────────────────────
22
+ # Config from environment (required by OpenEnv spec)
23
  # ─────────────────────────────────────────────
24
+ API_BASE_URL = os.getenv("API_BASE_URL", "https://api.openai.com/v1")
25
+ MODEL_NAME = os.getenv("MODEL_NAME", "gpt-4o-mini")
26
+ HF_TOKEN = os.getenv("HF_TOKEN")
27
  HF_TOKEN = os.environ.get("HF_TOKEN", "")
28
 
29
  if not HF_TOKEN:
 
219
  print(f"[INFO] {r['task_id']}: score={r['score']} reward={r['total_reward']}", file=sys.stderr)
220
 
221
  overall = round(sum(r["score"] for r in results) / len(results), 4)
222
+ print(f"[INFO] Overall score: {overall}", file=sys.stderr)