doosaganesh commited on
Commit
a80a354
·
verified ·
1 Parent(s): 203b157

Upload inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -175,7 +175,7 @@ def run_task(task_name: str) -> Dict[str, Any]:
175
  rewards.append(reward_val)
176
 
177
  if done:
178
- success = final_score >= 1.0
179
  break
180
 
181
  rewards_str = ",".join(f"{r:.2f}" for r in rewards)
 
175
  rewards.append(reward_val)
176
 
177
  if done:
178
+ success = final_score >= 0.95
179
  break
180
 
181
  rewards_str = ",".join(f"{r:.2f}" for r in rewards)