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

Add typehint for agent config class

Browse Source
This commit is contained in:
Marek Wolan
2023-07-17 11:21:29 +01:00
parent 2bb71623fa
commit 432da5ca90
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/primaite/agents/rllib.py
View File

@@ -66,6 +66,7 @@ class RLlibAgent(AgentSessionABC):
msg = f"Expected RLLIB agent_framework, " f"got {self._training_config.agent_framework}" msg = f"Expected RLLIB agent_framework, " f"got {self._training_config.agent_framework}"
_LOGGER.error(msg) _LOGGER.error(msg)
raise ValueError(msg) raise ValueError(msg)
self._agent_config_class: Union[PPOConfig, A2CConfig]
if self._training_config.agent_identifier == AgentIdentifier.PPO: if self._training_config.agent_identifier == AgentIdentifier.PPO:
self._agent_config_class = PPOConfig self._agent_config_class = PPOConfig
elif self._training_config.agent_identifier == AgentIdentifier.A2C: elif self._training_config.agent_identifier == AgentIdentifier.A2C: