Apply suggestions from code review

2023-07-03 10:47:26 +00:00
parent 06d5004695
commit 7fe46ef99c
3 changed files with 3 additions and 3 deletions
--- a/src/primaite/agents/rllib.py
+++ b/src/primaite/agents/rllib.py
@@ -55,7 +55,7 @@ class RLlibAgent(AgentSessionABC):
            msg = "Expected PPO or A2C agent_identifier, " f"got {self._training_config.agent_identifier.value}"
            _LOGGER.error(msg)
            raise ValueError(msg)
-        self._agent_config: PPOConfig
+        self._agent_config: Union[PPOConfig, A2CConfig]

        self._current_result: dict
        self._setup()