def METHOD_NAME(self, event: Event, state: State, logger: Logger) -> Optional[int]:
    x = state.batch_get_item(self.input_key)
    assert isinstance(x, Tensor), 'Multiple tensors not supported for Cutout.'
    new_x = cutout_batch(x, num_holes=self.num_holes, length=self.length, uniform_sampling=self.uniform_sampling)
    state.batch_set_item(self.input_key, new_x)