code: def on_batch_end(self, training_state, snapshot=False):
      training_state.step_time_total += (
          time.time() - training_state.step_time
      )
      if snapshot:
          self.snapshot_termlogs(training_state)
      else:
          self.print_termlogs(training_state)
  
imports: from __future__ import division, print_function, absolute_import
  import time
  import sys
label_desc: Use image node_id 6 for calling the TermLogger obj's underlying member method code with example usage: obj.on_batch_end(training_state, snapshot) without return types