hunk.code_window:     def extra_compute_grad_fetches(self):
           return {
               "td_error": self.loss.td_error,
               LEARNER_STATS_KEY: self.loss.stats,
           }
       @override(PolicyGraph)
       def get_state(self):
  
hunk.labels: keep keep keep replace keep keep keep keep
hunk.after_edit:             LEARNER_STATS_KEY: self.stats_fetches,
  
hunk.file_path: python/ray/rllib/agents/dqn/dqn_policy_graph.py
hunk.type: replace
file: .posts { list-style-type: none; }
  
  .posts li { margin-bottom: 30px; }
  