Properly import protos

6 years ago · f3b57e35ec
parent b9659d58da
commit f3b57e35ec
3 changed files with 5 additions and 4 deletions
--- a/examples/python/multiprocessing/BUILD
+++ b/examples/python/multiprocessing/BUILD
@ -20,6 +20,7 @@ load("@org_pubref_rules_protobuf//python:rules.bzl", "py_proto_library")
 py_proto_library(
    name = "prime_proto",
    protos = ["prime.proto",],
+    deps = [requirement("protobuf")],
 )

 py_binary(
--- a/examples/python/multiprocessing/client.py
+++ b/examples/python/multiprocessing/client.py
@ -26,8 +26,8 @@ import sys

 import grpc

-import prime_pb2
-import prime_pb2_grpc
+from examples.python.multiprocessing import prime_pb2
+from examples.python.multiprocessing import prime_pb2_grpc

 _PROCESS_COUNT = 8
 _MAXIMUM_CANDIDATE = 10000
--- a/examples/python/multiprocessing/server.py
+++ b/examples/python/multiprocessing/server.py
@ -29,8 +29,8 @@ import sys

 import grpc

-import prime_pb2
-import prime_pb2_grpc
+from examples.python.multiprocessing import prime_pb2
+from examples.python.multiprocessing import prime_pb2_grpc

 _LOGGER = logging.getLogger(__name__)