#1522: create_random_red_agent -> _create_random_red_agent + converting NodeStateInstructionRed into a dataclass

This commit is contained in:
Czar Echavez
2023-07-03 13:36:14 +01:00
parent befd183b2c
commit a7913487b8
2 changed files with 5 additions and 19 deletions

View File

@@ -278,7 +278,7 @@ class Primaite(Env):
# Create a random red agent to use for this episode
if self.training_config.random_red_agent:
self.create_random_red_agent()
self._create_random_red_agent()
# Reset counters and totals
self.total_reward = 0
@@ -1236,7 +1236,7 @@ class Primaite(Env):
combined_action_dict = {**acl_action_dict, **new_node_action_dict}
return combined_action_dict
def create_random_red_agent(self):
def _create_random_red_agent(self):
"""Decide on random red agent for the episode to be called in env.reset()."""
# Reset the current red iers and red node pol
self.red_iers = {}