prashantmatlani commited on
Commit
8f59669
·
1 Parent(s): c4b20ec

changed task to task_id

Browse files
Files changed (1) hide show
  1. inference.py +4 -2
inference.py CHANGED
@@ -177,7 +177,7 @@ def run_single_task(task):
177
  # =========================
178
  print(f"\n")
179
  print(json.dumps({
180
- "task": task_name,
181
  "score": round(score, 4)
182
  }))
183
  print(f"\n")
@@ -194,7 +194,9 @@ def main():
194
 
195
  print(f"[START] task=customer-support env=openenv model={model_name}")
196
 
197
- # RUN DISTINCT TASKS (NOT LOOP COPIES)
 
 
198
  for task in TASKS:
199
  run_single_task(task)
200
 
 
177
  # =========================
178
  print(f"\n")
179
  print(json.dumps({
180
+ "task_id": task_name,
181
  "score": round(score, 4)
182
  }))
183
  print(f"\n")
 
194
 
195
  print(f"[START] task=customer-support env=openenv model={model_name}")
196
 
197
+ print(f"\n[DEBUG] Running {len(TASKS)} tasks\n")
198
+
199
+ # RUN DISTINCT TASKS (NOT LOOP COPIES)
200
  for task in TASKS:
201
  run_single_task(task)
202