Help protect the Great Barrier Reef with TensorFlow on Kaggle Join Challenge

tf_agents.policies.utils.set_bandit_policy_type

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).