def __enter__(self): from gcloud.datastore.batch import _BATCHES _BATCHES.push(self._transaction) return self._transaction