code: def check_no_mask_with_arm_features(
      accepts_per_arm_features: bool,
      observation_and_action_constraint_splitter: types.Splitter,
  ):
      if accepts_per_arm_features and (
          observation_and_action_constraint_splitter is not None
      ):
          raise NotImplementedError(
              "Action masking is not allowed for Bandit problems with arm features. "
              "To implement a policy that handles variable number of actions, please "
              "use the `{}` feature key.".format(
                  bandit_spec_utils.NUM_ACTIONS_FEATURE_KEY
              )
          )
  
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 11 for calling a global function with example usage: check_no_mask_with_arm_features(accepts_per_arm_features, observation_and_action_constraint_splitter) without return types