Graheet commited on
Commit
4d67e80
·
1 Parent(s): 3ad2480

Align END log format with validator: use score field

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -213,7 +213,7 @@ def log_end(success: bool, steps: int, rewards: List[float], final_score: float)
213
 
214
  rewards_text = ",".join(f"{reward:.2f}" for reward in rewards)
215
  print(
216
- f"[END] success={str(success).lower()} steps={steps} rewards={rewards_text} final_score={final_score:.4f}",
217
  flush=True,
218
  )
219
 
 
213
 
214
  rewards_text = ",".join(f"{reward:.2f}" for reward in rewards)
215
  print(
216
+ f"[END] success={str(success).lower()} steps={steps} score={final_score:.2f} rewards={rewards_text}",
217
  flush=True,
218
  )
219