ray.rllib.policy.sample_batch.SampleBatch.ATTENTION_MASKS# SampleBatch.ATTENTION_MASKS = 'attention_masks'#