code: def test_set_attr(self):
      state = nnx.State({"a": nnx.Param(1), "b": {"c": nnx.Param(2)}})
  
      state.a = 3
      state.b.c = 4
  
      assert state["a"] == 3
      assert state["b"]["c"] == 4
  
file_path: flax/flax/experimental/nnx/tests/test_state.py
imports: from absl.testing.absltest import TestCase
  from flax.experimental import nnx
label_desc: Use image node_id 3 for calling the StateTest obj's underlying member method code with example usage: obj.test_set_attr() without return types