code: def on_epoch_end(self):
      raise NotImplementedError
  
file_path: labml/helpers/labml_helpers/metrics/__init__.py
imports: from abc import ABC
label_desc: Use image node_id 5 for calling the StateModule obj's underlying member method code with example usage: obj.on_epoch_end() without return types