hunk.code_window:             episode.batch_builder.count += 1
               episode._add_agent_rewards(rewards[env_id])
           if (episode.batch_builder.total() > max(1000, unroll_length * 10)
                   and log_once("large_batch_warning")):
               logger.warning(
                   "More than {} observations for {} env steps ".format(
  
hunk.labels: keep keep keep replace keep keep keep
hunk.after_edit:         if (episode.batch_builder.total() > large_batch_threshold
  
hunk.file_path: rllib/evaluation/sampler.py
hunk.type: replace
file: ray[debug]
  atoma
  flask
  