prashantmatlani commited on
Commit
8e156dc
·
1 Parent(s): ecf4764

implemented controlled stochasticity

Browse files
Files changed (1) hide show
  1. inference.py +3 -3
inference.py CHANGED
@@ -51,8 +51,8 @@ def format_action(action: dict) -> str:
51
  # =========================
52
  def run_single_task(task):
53
 
54
- task_name = task["name"]
55
- task_type = task["type"]
56
 
57
  env = CustomerSupportEnv()
58
  obs = env.reset()
@@ -175,7 +175,7 @@ def main():
175
  # 🚨 CRITICAL: One START per task (validator reads this)
176
  for task in TASKS:
177
 
178
- task_name = task["name"]
179
 
180
  print(f"[START] task={task_name} env={benchmark} model={model_name}")
181
 
 
51
  # =========================
52
  def run_single_task(task):
53
 
54
+ task_name = task["id"]
55
+ task_type = task["difficulty"]
56
 
57
  env = CustomerSupportEnv()
58
  obs = env.reset()
 
175
  # 🚨 CRITICAL: One START per task (validator reads this)
176
  for task in TASKS:
177
 
178
+ task_name = task["id"]
179
 
180
  print(f"[START] task={task_name} env={benchmark} model={model_name}")
181