code: def test_non_mapping_rngs(self):
      def f():
          mod = transform_flax.lift(ParamsRNG(), name="foo")
          return mod(rngs=jax.random.PRNGKey(42))
  
      f = transform.transform(f)
      with self.assertRaisesRegex(
          flax.errors.InvalidRngError, "should be a dictionary"
      ):
          f.init(None)
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import flax.errors
  import flax.linen
  from haiku._src import base
  from haiku._src import module
  from haiku._src import transform
  from haiku._src.flax import transform_flax
  import jax
  import jax.numpy
label_desc: Use image node_id 10 for calling the TransformFlaxTest obj's underlying member method code with example usage: obj.test_non_mapping_rngs() and returns: mod