flatland.utils.env_edit_utils module

Contents

flatland.utils.env_edit_utils module#

class flatland.utils.env_edit_utils.SchedGen2(rcStart, rcEnd, iDir)[source]#

Bases: BaseLineGen

generate(rail: GridTransitionMap, num_agents: int, hints: dict | None = None, num_resets: int | None = None, np_random: RandomState | None = None) → Line[source]#

class flatland.utils.env_edit_utils.SchedGen3(lrcStarts, lrcTargs, liDirs)[source]#

Bases: BaseLineGen

generate(rail: GridTransitionMap, num_agents: int, hints: dict | None = None, num_resets: int | None = None, np_random: RandomState | None = None) → Line[source]#

flatland.utils.env_edit_utils.getAgentState(env)[source]#

flatland.utils.env_edit_utils.makeEnv(nAg=2, width=20, height=10, oSG=None)[source]#

flatland.utils.env_edit_utils.makeEnv2(nAg=2, shape=(20, 10), llrcPaths=[], lrcStarts=[], lrcTargs=[], liDirs=[], remove_agents_at_target=True)[source]#

flatland.utils.env_edit_utils.makeTestEnv(sName='single_alternative', nAg=2, remove_agents_at_target=True)[source]#