code: def _raise_batch_shape_error(tensor_name, batch_shape):
      raise ValueError(
          "`{tensor_name}` must have batch shape with length 1; "
          "got {batch_shape}.".format(
              tensor_name=tensor_name, batch_shape=batch_shape
          )
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import gin
  import tensorflow
  from tf_agents.bandits.environments import non_stationary_stochastic_environment
  from tf_agents.specs import tensor_spec
  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(tensor_name, batch_shape) without return types