class_docstr: A callback which profiles user code using `PyTorch Profiler <https://pytorch.org/docs/stable/profiler.html/>`_.
  
  Args:
      profiler: a torch.profiler.profile context manager which will be used
code: def on_eval_step_end(self, state: State, unit: TEvalUnit) -> None:
      self.profiler.step()
  
file_path: tnt/torchtnt/framework/callbacks/pytorch_profiler.py
imports: import torch
  from torchtnt.framework.callback import Callback
  from torchtnt.framework.state import EntryPoint, State
  from torchtnt.framework.unit import TEvalUnit, TPredictUnit, TTrainUnit
label_desc: Use image node_id 6 for calling the PyTorchProfiler obj's underlying member method code with example usage: obj.on_eval_step_end(state, unit) without return types