#2869 - Minor changes trying to make pytest happy

2025-01-06 11:38:07 +00:00
parent 505eab6ed9
commit 66d309871f
11 changed files with 17 additions and 23 deletions
--- a/src/primaite/game/agent/interface.py
+++ b/src/primaite/game/agent/interface.py
@@ -51,9 +51,9 @@ class AbstractAgent(BaseModel):

    history: List[AgentHistoryItem] = []
    config: "AbstractAgent.ConfigSchema"
-    action_manager: ActionManager
-    observation_manager: ObservationManager
-    reward_function: RewardFunction
+    action_manager: "ActionManager"
+    observation_manager: "ObservationManager"
+    reward_function: "RewardFunction"

    class ConfigSchema(BaseModel):
        """