code: def test_context_none_after_init(self):
      @dataclasses.dataclass
      class DFoo(nnx.Module):
          din: int
          dout: int
          rngs: nnx.Rngs
  
          def __post_init__(self):
              self.bar = nnx.Linear(self.din, self.dout, rngs=self.rngs)
  
          def __call__(self, x):
              return self.bar(x)
  
      m = DFoo(1, 1, rngs=nnx.Rngs(0))
  
      assert hasattr(m, "bar")
      assert m.rngs is None
  
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 3 for calling the TestModuleDataclass obj's underlying member method code with example usage: obj.test_context_none_after_init() and returns: self