code: def test_no_override(self):
      @nnx.dataclass
      class Foo(nnx.Module):
          a: int = nnx.treenode_field()
  
      with pytest.raises(
          ValueError, match="is not compatible with return type"
      ):
          _m = Foo(a=nnx.Param(1))
  
      _m = Foo(a=nnx.TreeNode(1))
  
imports: import dataclasses
  from copy import deepcopy
  from typing import Any, TypeVar
  import jax
  import jax.numpy
  import pytest
  from flax.experimental import nnx
label_desc: Use image node_id 2 for calling the TestModuleDataclass obj's underlying member method code with example usage: obj.test_no_override() without return types