episode_collector

Functions

collect_episode(env, policy, trajectory[, ...])

Module Contents

episode_collector.collect_episode(env: gymnasium.Env, policy: gridmind.policies.base_policy.BasePolicy, trajectory: gridmind.utils.algorithm_util.trajectory.Trajectory, obs_preprocessor: Callable | None = None, record_action_prob: bool = False)[source]