Repository: beam Updated Branches: refs/heads/master 9379ca289 -> 929a23e29
Use imports from grpc. Project: http://git-wip-us.apache.org/repos/asf/beam/repo Commit: http://git-wip-us.apache.org/repos/asf/beam/commit/05765385 Tree: http://git-wip-us.apache.org/repos/asf/beam/tree/05765385 Diff: http://git-wip-us.apache.org/repos/asf/beam/diff/05765385 Branch: refs/heads/master Commit: 0576538561d2b13a920accaf90efc4ce2f2f50f0 Parents: 9379ca2 Author: Robert Bradshaw <[email protected]> Authored: Wed Oct 4 13:39:00 2017 -0700 Committer: Robert Bradshaw <[email protected]> Committed: Wed Oct 4 13:39:00 2017 -0700 ---------------------------------------------------------------------- sdks/python/apache_beam/runners/worker/log_handler.py | 3 ++- sdks/python/apache_beam/runners/worker/log_handler_test.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/beam/blob/05765385/sdks/python/apache_beam/runners/worker/log_handler.py ---------------------------------------------------------------------- diff --git a/sdks/python/apache_beam/runners/worker/log_handler.py b/sdks/python/apache_beam/runners/worker/log_handler.py index f878943..8691184 100644 --- a/sdks/python/apache_beam/runners/worker/log_handler.py +++ b/sdks/python/apache_beam/runners/worker/log_handler.py @@ -48,7 +48,8 @@ class FnApiLogRecordHandler(logging.Handler): def __init__(self, log_service_descriptor): super(FnApiLogRecordHandler, self).__init__() self._log_channel = grpc.insecure_channel(log_service_descriptor.url) - self._logging_stub = beam_fn_api_pb2.BeamFnLoggingStub(self._log_channel) + self._logging_stub = beam_fn_api_pb2_grpc.BeamFnLoggingStub( + self._log_channel) self._log_entry_queue = queue.Queue() log_control_messages = self._logging_stub.Logging(self._write_log_entries()) http://git-wip-us.apache.org/repos/asf/beam/blob/05765385/sdks/python/apache_beam/runners/worker/log_handler_test.py ---------------------------------------------------------------------- diff --git a/sdks/python/apache_beam/runners/worker/log_handler_test.py b/sdks/python/apache_beam/runners/worker/log_handler_test.py index 4903877..9814324 100644 --- a/sdks/python/apache_beam/runners/worker/log_handler_test.py +++ b/sdks/python/apache_beam/runners/worker/log_handler_test.py @@ -45,7 +45,7 @@ class FnApiLogRecordHandlerTest(unittest.TestCase): def setUp(self): self.test_logging_service = BeamFnLoggingServicer() self.server = grpc.server(futures.ThreadPoolExecutor(max_workers=10)) - beam_fn_api_pb2.add_BeamFnLoggingServicer_to_server( + beam_fn_api_pb2_grpc.add_BeamFnLoggingServicer_to_server( self.test_logging_service, self.server) self.test_port = self.server.add_insecure_port('[::]:0') self.server.start()
