episode_collector
=================

.. py:module:: episode_collector


Functions
---------

.. autoapisummary::

   episode_collector.collect_episode


Module Contents
---------------

.. py:function:: collect_episode(env: gymnasium.Env, policy: gridmind.policies.base_policy.BasePolicy, trajectory: gridmind.utils.algorithm_util.trajectory.Trajectory, obs_preprocessor: Optional[Callable] = None, record_action_prob: bool = False)