def step_end(self, step, output):
    super(GraphDump, self).step_end(step, output)
    self._data[step] = output
def step_end(self, step, output):
    super(GraphDump, self).step_end(step, output)
    self._data[step] = output