prashantmatlani commited on
Commit
276f522
·
1 Parent(s): c69c67b

modified inference to include proper prints

Browse files
Files changed (2) hide show
  1. app/env.py +3 -2
  2. inference.py +1 -1
app/env.py CHANGED
@@ -12,6 +12,9 @@ import sys
12
 
13
  AVAILABLE_TASKS = TASKS
14
 
 
 
 
15
  class CustomerSupportEnv:
16
 
17
  # OBTAIN TASKS FROM GRADERS.PY
@@ -278,5 +281,3 @@ class CustomerSupportEnv:
278
  "info_efficiency": round(info_eff, 3)
279
  }
280
 
281
- def get_tasks():
282
- return AVAILABLE_TASKS
 
12
 
13
  AVAILABLE_TASKS = TASKS
14
 
15
+ def get_tasks():
16
+ return AVAILABLE_TASKS
17
+
18
  class CustomerSupportEnv:
19
 
20
  # OBTAIN TASKS FROM GRADERS.PY
 
281
  "info_efficiency": round(info_eff, 3)
282
  }
283
 
 
 
inference.py CHANGED
@@ -194,7 +194,7 @@ def main():
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:
 
194
 
195
  print(f"[START] task=customer-support env=openenv model={model_name}")
196
 
197
+ #print(f"[DEBUG] Running {len(TASKS)} tasks")
198
 
199
  # RUN DISTINCT TASKS (NOT LOOP COPIES)
200
  for task in TASKS: