tf_agents.specs.bandit_spec_utils.drop_arm_observation

Drops the per-arm observation from a given trajectory/trajectory spec.