tf_agents.policies.utils.set_bandit_policy_type

Sets the InfoFields.BANDIT_POLICY_TYPE on info to bandit_policy_type.

tf_agents.policies.utils.set_bandit_policy_type(
    info: Optional[Union[tf_agents.policies.utils.PolicyInfo, tf_agents.policies.utils.PerArmPolicyInfo]],
    bandit_policy_type: tf_agents.typing.types.Tensor
) -> Union[tf_agents.policies.utils.PolicyInfo, tf_agents.policies.utils.PerArmPolicyInfo]

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

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

Returns
Policy info with modified field (if possible).