#917 - Integrated both SB3 and RLlib agents into PrimaiteSession

2023-06-19 20:27:08 +01:00
parent c09874edbe
commit 3670f16766
13 changed files with 726 additions and 688 deletions
--- a/src/primaite/environment/primaite_env.py
+++ b/src/primaite/environment/primaite_env.py
@@ -435,7 +435,6 @@ class Primaite(Env):
            _action: The action space from the agent
        """
        # At the moment, actions are only affecting nodes
-
        if self.training_config.action_type == ActionType.NODE:
            self.apply_actions_to_nodes(_action)
        elif self.training_config.action_type == ActionType.ACL: