From ed45658a650ec8bcf215df6a3b70ca2e4a218c5d Mon Sep 17 00:00:00 2001 From: Damon Douglas Date: Fri, 27 Dec 2024 11:54:24 -0800 Subject: [PATCH] Add worker_id attribute --- sdks/python/apache_beam/runners/worker/sdk_worker.py | 1 + 1 file changed, 1 insertion(+) diff --git a/sdks/python/apache_beam/runners/worker/sdk_worker.py b/sdks/python/apache_beam/runners/worker/sdk_worker.py index 0bcf43756f6..b4a2a04997f 100644 --- a/sdks/python/apache_beam/runners/worker/sdk_worker.py +++ b/sdks/python/apache_beam/runners/worker/sdk_worker.py @@ -900,6 +900,7 @@ def __init__(self, state_cache, worker_id, credentials=None): self._throwing_state_handler = ThrowingStateHandler() self._credentials = credentials self._state_cache = state_cache + self._worker_id = worker_id def create_state_handler(self, api_service_descriptor): # type: (endpoints_pb2.ApiServiceDescriptor) -> CachingStateHandler