cambrian.envs.step_fns

Step fns. These can be used to modify the observation and info dictionaries.

Functions

respawn_agent(env, agent)	Respawn agent at given position.
step_respawn_agents_if_close_to_agents(env, obs, info, ...)	Keywords Args:
step_add_agent_qpos_to_info(env, obs, info, *[, ...])	Add agent qpos to observation.
step_combined(env, obs, info, **step_fns)	Combine multiple step functions.

Module Contents

respawn_agent(env, agent)

Respawn agent at given position.

step_respawn_agents_if_close_to_agents(env, obs, info, *, distance_threshold, for_agents=None, to_agents=None, from_agents=None)

Keywords Args:

for_agents: List of agent names to check for proximity. to_agents: List of agent names to check distance to from_agents: List of agent names to check distance from

step_add_agent_qpos_to_info(env, obs, info, *, for_agents=None)

Add agent qpos to observation.

step_combined(env, obs, info, **step_fns)

Combine multiple step functions.