From bb8b41a5ec1bbf65dbc26f15d396c0039f17003e Mon Sep 17 00:00:00 2001 From: Chris McCarthy Date: Fri, 28 Jul 2023 14:02:17 +0100 Subject: [PATCH] #1711 - Removed the legacy bools from the RLlibAgent constructor in primaite_session.py --- src/primaite/primaite_session.py | 8 +------- tests/test_full_legacy_config_session.py | 2 +- 2 files changed, 2 insertions(+), 8 deletions(-) diff --git a/src/primaite/primaite_session.py b/src/primaite/primaite_session.py index f7495c48..2cb0d5bd 100644 --- a/src/primaite/primaite_session.py +++ b/src/primaite/primaite_session.py @@ -160,13 +160,7 @@ class PrimaiteSession: elif self._training_config.agent_framework == AgentFramework.RLLIB: _LOGGER.debug(f"PrimaiteSession Setup: Agent Framework = {AgentFramework.RLLIB}") # Ray RLlib Agent - self._agent_session = RLlibAgent( - self._training_config_path, - self._lay_down_config_path, - self.session_path, - self.legacy_training_config, - self.legacy_lay_down_config, - ) + self._agent_session = RLlibAgent(self._training_config_path, self._lay_down_config_path, self.session_path) else: # Invalid AgentFramework diff --git a/tests/test_full_legacy_config_session.py b/tests/test_full_legacy_config_session.py index 33f827f1..066ff72c 100644 --- a/tests/test_full_legacy_config_session.py +++ b/tests/test_full_legacy_config_session.py @@ -20,7 +20,7 @@ def test_legacy_training_config_run_session(legacy_file): legacy_training_config_path = TEST_CONFIG_ROOT / "legacy_conversion" / "legacy_training_config.yaml" legacy_lay_down_config_path = TEST_CONFIG_ROOT / "legacy_conversion" / legacy_file - # Run a PrimAITE session using the paths of both the converted training and lay down config files + # Run a PrimAITE session using legacy training and lay down config file paths run( legacy_training_config_path, legacy_lay_down_config_path,