diff --git a/src/primaite/environment/primaite_env.py b/src/primaite/environment/primaite_env.py index bc52c887..03798c3b 100644 --- a/src/primaite/environment/primaite_env.py +++ b/src/primaite/environment/primaite_env.py @@ -368,7 +368,7 @@ class Primaite(Env): self.step_count, self.config_values, ) - # print(f" Step {self.step_count} Reward: {str(reward)}") + print(f" Step {self.step_count} Reward: {str(reward)}") self.total_reward += reward if self.step_count == self.episode_steps: self.average_reward = self.total_reward / self.step_count