flatland.envs.malfunction_effects_generators module

flatland.envs.malfunction_effects_generators module#

class flatland.envs.malfunction_effects_generators.MalfunctionEffectsGenerator(malfunction_generator)[source]#

Bases: EffectsGenerator[RailEnv]

on_episode_step_start(env: RailEnv, *args, **kwargs) RailEnv[source]#

Called by env at the beginning of step before evaluating the agent’s actions.

In the future, will receive immutable state instead of full env.

No naming similar to RLlib equivalent, see https://docs.ray.io/en/latest/rllib/rllib-callback.html

Parameters#

env args kwargs

Returns#