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