Mohammed-Altaf commited on
Commit
59a045c
·
1 Parent(s): f5f530a

change logging of score to int

Browse files
Files changed (1) hide show
  1. helpers/logging.py +1 -1
helpers/logging.py CHANGED
@@ -16,4 +16,4 @@ def log_step(step: int, action: Union[dict, str], reward: float, done: bool, err
16
 
17
  def log_end(success: bool, steps: int, score: float, rewards: List[float]) -> None:
18
  rewards_str = ",".join(f"{r:.2f}" for r in rewards)
19
- print(f"[END] success={str(success).lower()} steps={steps} score={score:.3f} rewards={rewards_str}\n", flush=True)
 
16
 
17
  def log_end(success: bool, steps: int, score: float, rewards: List[float]) -> None:
18
  rewards_str = ",".join(f"{r:.2f}" for r in rewards)
19
+ print(f"[END] success={str(success).lower()} steps={steps} score={int(score)} rewards={rewards_str}\n", flush=True)