code: def create_chosen_arm_features_info_spec(
      observation_spec: types.NestedTensorSpec,
  ) -> types.NestedTensorSpec:
      """Creates the chosen arm features info spec from the arm observation spec."""
      arm_spec = observation_spec[bandit_spec_utils.PER_ARM_FEATURE_KEY]
      return tensor_spec.remove_outer_dims_nest(arm_spec, 1)
  
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 10 for calling a global function with example usage: create_chosen_arm_features_info_spec(observation_spec) and returns: tensor_spec