def METHOD_NAME(self, batch, state=None, **kwargs):
    batch_size = batch.obs.cur_step.shape[0]
    if state is None:
        state = np.zeros(batch_size)
    else:
        state = state + 2
    return Batch(
        act=np.zeros(batch_size, dtype=np.int64),
        state=state
    )