P-Karthik-Mohan commited on
Commit
4e17f50
Β·
1 Parent(s): 935b3b5

Fix END log format to match sample exactly :)

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -44,7 +44,7 @@ def log_step(step, action, reward, done, error=None):
44
 
45
  def log_end(success, steps, score, rewards):
46
  rewards_str = ",".join(f"{r:.2f}" for r in rewards)
47
- print(f"[END] success={str(success).lower()} steps={steps} score={score:.3f} rewards={rewards_str}", flush=True)
48
 
49
  # ── Environment helpers ───────────────────────────────────────────────────────
50
 
 
44
 
45
  def log_end(success, steps, score, rewards):
46
  rewards_str = ",".join(f"{r:.2f}" for r in rewards)
47
+ print(f"[END] success={str(success).lower()} steps={steps} rewards={rewards_str}", flush=True)
48
 
49
  # ── Environment helpers ───────────────────────────────────────────────────────
50