ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


Sets the InfoFields.BANDIT_POLICY_TYPE on info to bandit_policy_type.

If policy info does not support InfoFields.BANDIT_POLICY_TYPE, this method returns info as-is (without any modification).

info Policy info on which to set bandit policy type.
bandit_policy_type Tensor containing BanditPolicyType enums or TensorSpec from create_bandit_policy_type_tensor_spec().

Policy info with modified field (if possible).