code: def create_bandit_policy_type_tensor_spec(
      shape: types.Shape,
  ) -> types.BoundedTensorSpec:
      """Create tensor spec for bandit policy type."""
      return tensor_spec.BoundedTensorSpec(
          shape=shape,
          dtype=tf.int32,
          minimum=BanditPolicyType.UNKNOWN,
          maximum=BanditPolicyType.FALCON,
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import collections
  from typing import Optional, Sequence, Text, Union
  import tensorflow
  from tf_agents.specs import bandit_spec_utils
  from tf_agents.specs import tensor_spec
  from tf_agents.trajectories import policy_step
  from tf_agents.typing import types
  from tf_agents.utils import common
label_desc: Use image node_id 3 for calling a global function with example usage: create_bandit_policy_type_tensor_spec(shape) and returns: tensor_spec