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

#1711 - Removed the legacy bools from the RLlibAgent constructor in primaite_session.py

Browse Source
This commit is contained in:
Chris McCarthy
2023-07-28 14:02:17 +01:00
parent 0fb9268f44
commit bb8b41a5ec
2 changed files with 2 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/primaite/primaite_session.py
View File

@@ -160,13 +160,7 @@ class PrimaiteSession:
elif self._training_config.agent_framework == AgentFramework.RLLIB: elif self._training_config.agent_framework == AgentFramework.RLLIB:
_LOGGER.debug(f"PrimaiteSession Setup: Agent Framework = {AgentFramework.RLLIB}") _LOGGER.debug(f"PrimaiteSession Setup: Agent Framework = {AgentFramework.RLLIB}")
# Ray RLlib Agent # Ray RLlib Agent
self._agent_session = RLlibAgent( self._agent_session = RLlibAgent(self._training_config_path, self._lay_down_config_path, self.session_path)
self._training_config_path,
self._lay_down_config_path,
self.session_path,
self.legacy_training_config,
self.legacy_lay_down_config,
)
else: else:
# Invalid AgentFramework # Invalid AgentFramework

2
tests/test_full_legacy_config_session.py
View File

@@ -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_training_config_path = TEST_CONFIG_ROOT / "legacy_conversion" / "legacy_training_config.yaml"
legacy_lay_down_config_path = TEST_CONFIG_ROOT / "legacy_conversion" / legacy_file 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( run(
legacy_training_config_path, legacy_training_config_path,
legacy_lay_down_config_path, legacy_lay_down_config_path,