flatland.evaluators.trajectory_evaluator module

flatland.evaluators.trajectory_evaluator module#

class flatland.evaluators.trajectory_evaluator.TrajectoryEvaluator(trajectory: Trajectory, callbacks: FlatlandCallbacks | None = None)[source]#

Bases: object

evaluate(start_step: int | None = None, end_step: int | None = None, snapshot_interval=0, tqdm_kwargs: dict | None = None, skip_rewards_dones_infos: bool = False) RailEnv[source]#

Parameters

end_stepint

stop evaluation at intermediate step excl.

renderingbool

render while evaluating

snapshot_intervalint

interval to write pkl snapshots to outputs/serialised_state subdirectory (not serialised_state subdirectory directly). 1 means at every step. 0 means never.

tqdm_kwargs: dict

additional kwargs for tqdm

skip_rewards_dones_infosbool

skip verification of rewards/dones/infos