code: def dataclass(cls: tp.Type[A]) -> tp.Type[A]:
      ...
  
file_path: flax/flax/experimental/nnx/nnx/dataclasses.py
imports: import dataclasses
  import typing
  import typing_extensions
  from flax.experimental import nnx
  from flax.experimental.nnx.nnx import pytreelib, variables
label_desc: Use image node_id 5 for calling a global function with example usage: dataclass(cls) without return types