yashash045 commited on
Commit
bd273e6
·
verified ·
1 Parent(s): 83ecd75

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. inference.py +3 -3
inference.py CHANGED
@@ -188,7 +188,7 @@ def parse_llm_action(text):
188
  async def run_task(client, env, task_name):
189
  rewards = []
190
  steps_taken = 0
191
- score = 0.0
192
  success = False
193
  max_steps = MAX_STEPS_PER_TASK.get(task_name, 20)
194
  max_reward = MAX_TOTAL_REWARD.get(task_name, 1.0)
@@ -263,8 +263,8 @@ async def run_task(client, env, task_name):
263
  if done:
264
  break
265
 
266
- score = sum(rewards) / max_reward if max_reward > 0 else 0.0
267
- score = min(max(score, 0.0), 1.0)
268
  success = score >= SUCCESS_SCORE_THRESHOLD
269
 
270
  except Exception as e:
 
188
  async def run_task(client, env, task_name):
189
  rewards = []
190
  steps_taken = 0
191
+ score = 0.001
192
  success = False
193
  max_steps = MAX_STEPS_PER_TASK.get(task_name, 20)
194
  max_reward = MAX_TOTAL_REWARD.get(task_name, 1.0)
 
263
  if done:
264
  break
265
 
266
+ score = sum(rewards) / max_reward if max_reward > 0 else 0.001
267
+ score = min(max(score, 0.001), 0.999)
268
  success = score >= SUCCESS_SCORE_THRESHOLD
269
 
270
  except Exception as e: