code: def test_bias_dims_invalid(self):
      mod = bias.Bias(bias_dims=[1, 5])
      with self.assertRaisesRegex(
          ValueError, "5 .* out of range for input of rank 3"
      ):
          mod(jnp.ones([1, 2, 3]))
  
file_path: dm-haiku/haiku/_src/bias_test.py
imports: from absl.testing import absltest
  from haiku._src import bias
  from haiku._src import test_utils
  from haiku._src import transform
  import jax.numpy
  import numpy
label_desc: Use image node_id 6 for calling the BiasTest obj's underlying member method code with example usage: obj.test_bias_dims_invalid() without return types