add_custom_init_fn

Signature:

def add_custom_init_fn(cls: Any) -> None:
  ...