Spaces:
Sleeping
Sleeping
Update inference.py
Browse files- 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.
|
| 88 |
error = "null"
|
| 89 |
|
| 90 |
try:
|
|
@@ -116,7 +116,8 @@ def run_task(task_id: str) -> float:
|
|
| 116 |
break
|
| 117 |
|
| 118 |
# ── [END] ──
|
| 119 |
-
|
|
|
|
| 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.
|
| 146 |
-
scores[task] = 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} /
|
| 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()
|