docs: def draw_summary(self, history, title=""):
          """Inserts a text summary at the top that lists the number of steps and total
          training time."""
          # Generate summary string
          time_str = str(history.get_total_time()).split(".")[0]  # remove microseconds
          summary = "Step: {}      Time: {}".format(history.step, time_str)
          if title:
              summary = title + "\n\n" + summary
          self.figure.suptitle(summary)