tf_agents.trajectories.policy_step.set_log_probability

Sets the CommonFields.LOG_PROBABILITY on info to be log_probability.