class_docstr: A helper class to run and time callbacks in TorchTNT.
class_docstr_tok: ["A","helper","class","to","run","and","time","callbacks","in","TorchTNT","."]
code: def on_predict_end(self, state: State, unit: TPredictUnit) -> None:
      fn_name = "on_predict_end"
      callbacks = self._callbacks.get(fn_name, [])
      for cb in callbacks:
          cb.on_predict_end(state, unit)
  
imports: import logging
  from functools import partial
  from typing import Dict, List, Type, Union
  from unittest.mock import Mock
  from torchtnt.framework.callback import Callback
  from torchtnt.framework.state import State
  from torchtnt.framework.unit import TEvalUnit, TPredictUnit, TTrainUnit
label_desc: Use image node_id 23 for calling the CallbackHandler obj's underlying member method code with example usage: obj.on_predict_end(state, unit) without return types