code: def on_train_step_end(self, tez_trainer, **kwargs):
      train_metrics = tez_trainer.metrics["train"]
      if "epoch" in train_metrics:
          del train_metrics["epoch"]
      self._train_tqdm.set_postfix(
          epoch=tez_trainer.current_epoch, **train_metrics
      )
      self._train_tqdm.update(1)
  
file_path: tez/tez/callbacks/progress.py
imports: from tqdm.auto import tqdm
  from tez.callbacks import Callback
label_desc: Use image node_id 5 for calling the Progress obj's underlying member method code with example usage: obj.on_train_step_end(tez_trainer) without return types