     base_env (BaseEnv): The base environment that is being used in the training process.</span>
<span class="sd">            policies (Dict[str, Policy]): The policies that are being used in the training process.</span>
<span class="sd">            episode (MultiAgentEpisode): The episode object that is being processed.</span>
<span class="sd">            env_index (int): The index of the environment within the worker task.</span>
<span class="sd">            **kwargs: additional arguments that can be pass