hunk.code_window:             (SampleBatch.NEXT_OBS, self.obs_tp1),
               (SampleBatch.DONES, self.done_mask),
               (PRIO_WEIGHTS, self.importance_weights),
           ]
           TFPolicyGraph.__init__(
               self,
               observation_space,
               action_space,
  
hunk.labels: keep keep keep add keep keep keep keep
hunk.after_edit:         LearningRateSchedule.__init__(self, self.config["lr"],
                                         self.config["lr_schedule"])
  
hunk.file_path: python/ray/rllib/agents/dqn/dqn_policy_graph.py
hunk.type: add
file: package org.ray.streaming.api.function.impl;
  
  import org.ray.streaming.api.function.Function;
  
  /**
   * Interface of aggregate functions.
   *
   * @param <I> Type of the input data.
   * @param <A> Type of the intermediate data.
   * @param <O> Type of the output data.
   */
  public interface AggregateFunction<I, A, O> extends Function {
  
    A createAccumulator();
  
    void add(I value, A accumulator);
  
    O getResult(A accumulator);
  
    A merge(A a, A b);
  
    void retract(A acc, I value);
  }
  