hunk.code_window:                 "the batch size based on environment steps, not the steps of "
                   "individual agents, which can result in unexpectedly large "
                   "batches.")
           # Check episode termination conditions
           if dones[env_id]["__all__"] or episode.length >= horizon:
               hit_horizon = (episode.length >= horizon
  
hunk.labels: keep keep replace keep keep keep keep
hunk.after_edit:                 "batches. Also, you may be in evaluation waiting for your Env "
                   "to terminate (batch_mode=`complete_episodes`). Make sure it "
                   "does at some point.")
  
hunk.file_path: rllib/evaluation/sampler.py
hunk.type: replace
file: ray[debug]