code: def _raise_batch_shape_error(distribution_name, batch_shape):
      raise ValueError(
          "`{distribution_name}` must have batch shape with length 1; "
          "got {batch_shape}. Consider using "
          "`tensorflow_probability.distributions.Independent` "
          "to manipulate batch and event shapes.".format(
              distribution_name=distribution_name,
              batch_shape=batch_shape,
          )
      )
  
file_path: agents/tf_agents/bandits/environments/random_bandit_environment.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from typing import Optional
  import tensorflow
  from tf_agents.bandits.environments import bandit_tf_environment
  from tf_agents.specs import tensor_spec
  from tf_agents.trajectories import time_step
  from tf_agents.typing import types
label_desc: Use image node_id 1 for calling a global function with example usage: _raise_batch_shape_error(distribution_name, batch_shape) without return types