hunk.code_window:                         "from the environment.")
               # Creates a new episode if this is not async return.
               # If reset is async, we will get its result in some future poll.
               elif resetted_obs != ASYNC_RESET_RETURN:
                   new_episode: Episode = active_episodes[env_id]
                   if observation_fn:
                       resetted_obs: Dict[AgentID, EnvObsType] = observation_fn(
                           agent_obs=resetted_obs,
                           worker=worker,
                           base_env=base_env,
  
hunk.labels: keep keep keep keep add keep keep keep keep keep
hunk.after_edit:                 resetted_obs = resetted_obs[env_id]
  
hunk.file_path: rllib/evaluation/sampler.py
hunk.type: add
file: imagenette2.tgz
  /data
  