- class invertedai.cosimulation.BasicCosimulation(location: str, ego_agent_mask: Optional[List[bool]] = None, monitor_infractions: bool = False, get_birdview: bool = False, random_seed: Optional[int] = None, **kwargs)#
Stateful wrapper around the Inverted AI API to simplify co-simulation. All arguments to
initialize()can be passed to the constructor here and a sufficient combination of them must be passed as required by
initialize(). This wrapper caches static agent attributes and propagates the recurrent state, so that only states of ego agents and NPCs need to be exchanged with it to perform co-simulation. Typically, each time step requires a single call to
self.npc_states()and a single call to
This wrapper only supports a minimal co-simulation functionality. For more advanced use cases, call
location – Location name as expected by
ego_agent_mask – List indicating which agent is ego, meaning that it is controlled by you externally. The order in this list should be the same as that used in arguments to
monitor_infraction – Whether to monitor driving infractions, at a small increase in latency and payload size.
get_birdview – Whether to render the bird’s eye view of the simulation state at each time step. It drastically increases the payload received from Inverted AI servers and therefore slows down the simulation - use only for debugging.
random_seed – Controls the stochastic aspects of simulation for reproducibility.
- property agent_count: int#
The total number of agents, both ego and NPCs.
- property birdview: Image#
If get_birdview was set in the constructor, this is the image showing the current state of the simulation.
- property ego_agent_mask: List[bool]#
Lists which agents are ego, which means that you control them externally. It can be updated during the simulation, but see caveats in user guide regarding the quality of resulting predictions.
- property ego_states#
Returns the predicted states of ego agents in order. The NPC agents are excluded.
- property infractions: Optional[List[InfractionIndicators]]#
If monitor_infractions was set in the constructor, lists infractions currently committed by each agent, including ego agents.
- property location: str#
Location name as recognized by Inverted AI API.
- property npc_states: List[AgentState]#
Returns the predicted states of NPCs (non-ego agents) in order. The predictions for ego agents are excluded.
- step(current_ego_agent_states: List[AgentState]) None #
drive()to advance the simulation by one time step. Current states of ego agents need to be provided to synchronize with your local simulator.
current_ego_agent_states – States of ego agents before the step.
None - call
self.npc_states()to retrieve predictions.