Apply suggestions from PR review.

2024-03-04 09:58:57 +00:00
parent a6031d568d
commit d1480e4477
16 changed files with 25 additions and 19 deletions
--- a/src/primaite/game/game.py
+++ b/src/primaite/game/game.py
@@ -414,7 +414,7 @@ class PrimaiteGame:
            reward_function = RewardFunction.from_config(reward_function_cfg)

            # CREATE AGENT
-            if agent_type == "probabilistic_agent":
+            if agent_type == "ProbabilisticAgent":
                # TODO: implement non-random agents and fix this parsing
                settings = agent_cfg.get("agent_settings")
                new_agent = ProbabilisticAgent(