tf_agents.bandits.specs.utils.drop_arm_observation

Drops the per-arm observation from a given trajectory (or trajectory spec).