code: def __init__(self):
      self.state = None
  
file_path: labml/helpers/labml_helpers/metrics/simple_state.py
imports: from typing import Generic, TypeVar, Optional
  from .None import StateModule
label_desc: Use image node_id 1 to create a new SimpleState object from inherited base classes: Generic with example: obj = SimpleState()