code: def test_rng_stream_pytree(self):
      rngs = nnx.Rngs(params=0, dropout=1)
      stream = rngs.fork()["params"]
  
      stream2 = jax.tree_map(lambda x: x, stream)
  
      assert stream.key is stream2.key
  
imports: from typing import Any
  import jax
  import jax.numpy
  import pytest
  from flax.experimental import nnx
  from flax.experimental.nnx.nnx.rnglib import _stable_hash
label_desc: Use image node_id 15 for calling the TestRngs obj's underlying member method code with example usage: obj.test_rng_stream_pytree() without return types