Dolphin-Syndrom commited on
Commit
7ce5739
·
1 Parent(s): ab42912

fix: update log_end score formatting to .3f to strictly match sample script implementation

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -115,7 +115,7 @@ def log_step(step: int, action: str, reward: float, done: bool, error: Optional[
115
  def log_end(success: bool, steps: int, score: float, rewards: List[float]) -> None:
116
  rewards_str = ",".join(f"{r:.2f}" for r in rewards)
117
  print(
118
- f"[END] success={str(success).lower()} steps={steps} score={score:.2f} rewards={rewards_str}",
119
  flush=True,
120
  )
121
 
 
115
  def log_end(success: bool, steps: int, score: float, rewards: List[float]) -> None:
116
  rewards_str = ",".join(f"{r:.2f}" for r in rewards)
117
  print(
118
+ f"[END] success={str(success).lower()} steps={steps} score={score:.3f} rewards={rewards_str}",
119
  flush=True,
120
  )
121