oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

893 -

Browse Source
This commit is contained in:
SunilSamra
2023-06-06 13:47:07 +01:00
parent af44b99b6f
commit 69c5c9458b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/primaite/environment/primaite_env.py
View File

@@ -368,7 +368,7 @@ class Primaite(Env):
self.step_count, self.step_count,
self.config_values, 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 self.total_reward += reward
if self.step_count == self.episode_steps: if self.step_count == self.episode_steps:
self.average_reward = self.total_reward / self.step_count self.average_reward = self.total_reward / self.step_count