#2869 - Some additional test updates to amend failures. Pre-commit tbd and some cleanup

This commit is contained in:
Charlie Crane
2024-12-17 12:21:28 +00:00
parent 436a986458
commit 3b1b74fb3a
8 changed files with 27 additions and 13 deletions

View File

@@ -9,6 +9,7 @@ from gymnasium.core import ObsType
from primaite.game.agent.scripted_agents.interface import AbstractAgent, AbstractScriptedAgent
__all__ = ("AbstractTAPAgent")
class AbstractTAPAgent(AbstractScriptedAgent, identifier="Abstract_TAP"):
"""Base class for TAP agents to inherit from."""
@@ -22,11 +23,6 @@ class AbstractTAPAgent(AbstractScriptedAgent, identifier="Abstract_TAP"):
starting_node_name: Optional[str] = None
# @property
# def next_execution_timestep(self) -> int:
# """Returns the agents next execution timestep."""
# return self.next_execution_timestep
@abstractmethod
def get_action(self, obs: ObsType, timestep: int = 0) -> Tuple[str, Dict]:
"""Return an action to be taken in the environment."""