diff options
Diffstat (limited to 'benchmarks/py_benchmark.py')
| -rwxr-xr-x | benchmarks/py_benchmark.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/benchmarks/py_benchmark.py b/benchmarks/py_benchmark.py index 48234f03..ba7a3470 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 @@ -115,3 +115,4 @@ class Benchmark: if __name__ == "__main__": for i in range(2, len(sys.argv)): run_one_test(sys.argv[i]) + |