aboutsummaryrefslogtreecommitdiffhomepage
path: root/benchmarks/py_benchmark.py
diff options
context:
space:
mode:
Diffstat (limited to 'benchmarks/py_benchmark.py')
-rwxr-xr-xbenchmarks/py_benchmark.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/benchmarks/py_benchmark.py b/benchmarks/py_benchmark.py
index 48234f03..8858f4aa 100755
--- a/benchmarks/py_benchmark.py
+++ b/benchmarks/py_benchmark.py
@@ -17,8 +17,8 @@ elif sys.argv[1] != "false":
raise IOError("Need string argument \"true\" or \"false\" for whether to use cpp generated code")
# END CPP GENERATED MESSAGE
-import datasets.google_message1.benchmark_message1_proto2_pb2 as benchmark_message1_proto2_pb2
-import datasets.google_message1.benchmark_message1_proto3_pb2 as benchmark_message1_proto3_pb2
+import datasets.google_message1.proto2.benchmark_message1_proto2_pb2 as benchmark_message1_proto2_pb2
+import datasets.google_message1.proto3.benchmark_message1_proto3_pb2 as benchmark_message1_proto3_pb2
import datasets.google_message2.benchmark_message2_pb2 as benchmark_message2_pb2
import datasets.google_message3.benchmark_message3_pb2 as benchmark_message3_pb2
import datasets.google_message4.benchmark_message4_pb2 as benchmark_message4_pb2