#915 - Annotated logs func in cli.py to take -n.

- Fixed entry point on main.py
- Commented out the print reward line in step func of primaite_env.py.
- Added jupyterlab==3.6.1 to pyproject.toml
This commit is contained in:
Chris McCarthy
2023-06-09 16:44:49 +01:00
parent 1a7d158d77
commit 149a534851
5 changed files with 39 additions and 49 deletions

View File

@@ -377,7 +377,7 @@ class Primaite(Env):
self.step_count,
self.training_config,
)
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