You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@beam.apache.org by ro...@apache.org on 2017/11/15 01:42:29 UTC

[2/2] beam git commit: [BEAM-3150] Fix threading for progress reporting in SDKHarness

[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 <go...@goenka.svl.corp.google.com>
Authored: Tue Nov 7 14:02:29 2017 -0800
Committer: Robert Bradshaw <ro...@gmail.com>
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