rudrapatel-1908 commited on
Commit
616561d
·
verified ·
1 Parent(s): cbd3baf

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +6 -5
inference.py CHANGED
@@ -84,7 +84,7 @@ def run_task(task_id: str) -> float:
84
 
85
  for step_num in range(1, MAX_STEPS + 1):
86
  action_str = "null"
87
- reward = 0.0
88
  error = "null"
89
 
90
  try:
@@ -116,7 +116,8 @@ def run_task(task_id: str) -> float:
116
  break
117
 
118
  # ── [END] ──
119
- score = max(0.0, min(1.0, sum(r for r in rewards if r > 0)))
 
120
  rewards_str = ",".join(f"{r:.2f}" for r in rewards)
121
  print(
122
  f"[END] success={str(success).lower()} steps={len(rewards)} "
@@ -142,15 +143,15 @@ def main():
142
  try:
143
  scores[task] = run_task(task)
144
  except Exception as e:
145
- print(f"[END] success=false steps=0 score=0.00 rewards=", flush=True)
146
- scores[task] = 0.0
147
 
148
  # Summary
149
  total = sum(scores.values())
150
  print(f"\n# FINAL SCORES", flush=True)
151
  for task, score in scores.items():
152
  print(f"# {task}: {score:.2f}", flush=True)
153
- print(f"# Total: {total:.2f} / 3.00", flush=True)
154
 
155
  if __name__ == "__main__":
156
  main()
 
84
 
85
  for step_num in range(1, MAX_STEPS + 1):
86
  action_str = "null"
87
+ reward = 0.05
88
  error = "null"
89
 
90
  try:
 
116
  break
117
 
118
  # ── [END] ──
119
+ raw_score = sum(r for r in rewards if r > 0)
120
+ score = round(max(0.01, min(0.99, raw_score)), 2)
121
  rewards_str = ",".join(f"{r:.2f}" for r in rewards)
122
  print(
123
  f"[END] success={str(success).lower()} steps={len(rewards)} "
 
143
  try:
144
  scores[task] = run_task(task)
145
  except Exception as e:
146
+ print(f"[END] success=false steps=0 score=0.01 rewards=0.01", flush=True)
147
+ scores[task] = 0.01
148
 
149
  # Summary
150
  total = sum(scores.values())
151
  print(f"\n# FINAL SCORES", flush=True)
152
  for task, score in scores.items():
153
  print(f"# {task}: {score:.2f}", flush=True)
154
+ print(f"# Total: {total:.2f} / 2.97", flush=True)
155
 
156
  if __name__ == "__main__":
157
  main()