def end(self, session=None):
    super(SummarySaver, self).end(session=session)
    if self._summary_writer:
        self._summary_writer.flush()
def end(self, session=None):
    super(SummarySaver, self).end(session=session)
    if self._summary_writer:
        self._summary_writer.flush()