Change get_action signature for agents

2024-03-04 10:43:38 +00:00
parent d1480e4477
commit ac9d550e9b
7 changed files with 27 additions and 28 deletions
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -328,7 +328,7 @@ class ControlledAgent(AbstractAgent):
        )
        self.most_recent_action: Tuple[str, Dict]

-    def get_action(self, obs: None, reward: float = 0.0, timestep: Optional[int] = None) -> Tuple[str, Dict]:
+    def get_action(self, obs: None, timestep: int = 0) -> Tuple[str, Dict]:
        """Return the agent's most recent action, formatted in CAOS format."""
        return self.most_recent_action