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

Merged PR 62: Make reward calculation consider red POL

Browse Source 
Check out the linked bug ticket to understand the issue.

The fix was very simple- just changing which variable is passed to the reward calculation funciton.

Related work items: #1442
This commit is contained in:
Marek Wolan
2023-06-06 12:27:55 +00:00
parent c276a31b9c cdd710d672
commit a987ffb745
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):
# 5. Calculate reward signal (for RL) # 5. Calculate reward signal (for RL)
reward = calculate_reward_function( reward = calculate_reward_function(
self.nodes_post_pol, self.nodes_post_pol,
self.nodes_post_blue, self.nodes_post_red,
self.nodes_reference, self.nodes_reference,
self.green_iers, self.green_iers,
self.red_iers, self.red_iers,