[BEAM-3150] Fix threading for progress reporting in SDKHarness
Project: http://git-wip-us.apache.org/repos/asf/beam/repo Commit: http://git-wip-us.apache.org/repos/asf/beam/commit/4403b348 Tree: http://git-wip-us.apache.org/repos/asf/beam/tree/4403b348 Diff: http://git-wip-us.apache.org/repos/asf/beam/diff/4403b348 Branch: refs/heads/master Commit: 4403b3484eeab06fc0a9badd1b1bb4aa682f7bbc Parents: a5254e7 Author: Ankur Goenka <[email protected]> Authored: Tue Nov 7 14:02:29 2017 -0800 Committer: Robert Bradshaw <[email protected]> Committed: Tue Nov 14 17:41:57 2017 -0800 ---------------------------------------------------------------------- sdks/python/apache_beam/runners/worker/sdk_worker.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/beam/blob/4403b348/sdks/python/apache_beam/runners/worker/sdk_worker.py ---------------------------------------------------------------------- diff --git a/sdks/python/apache_beam/runners/worker/sdk_worker.py b/sdks/python/apache_beam/runners/worker/sdk_worker.py index 55ecbcc..fec844e 100644 --- a/sdks/python/apache_beam/runners/worker/sdk_worker.py +++ b/sdks/python/apache_beam/runners/worker/sdk_worker.py @@ -66,7 +66,8 @@ class SdkHarness(object): for work_request in control_stub.Control(get_responses()): logging.info('Got work %s', work_request.instruction_id) request_type = work_request.WhichOneof('request') - if request_type == ['process_bundle_progress']: + # WhichOneOf returns the name of the set field as a single string + if request_type in ['process_bundle_progress']: thread_pool = self._progress_thread_pool else: thread_pool = self._default_work_thread_pool
