diff options
Diffstat (limited to 'benchmarks/consumer_performance.py')
| -rwxr-xr-x | benchmarks/consumer_performance.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/benchmarks/consumer_performance.py b/benchmarks/consumer_performance.py index 5ffd3f5..d7580ce 100755 --- a/benchmarks/consumer_performance.py +++ b/benchmarks/consumer_performance.py @@ -29,7 +29,7 @@ def start_brokers(n): replicas = min(n, 3) print('-> {0} Brokers [{1} partitions / {2} replicas]'.format(n, partitions, replicas)) brokers = [ - KafkaFixture.instance(i, zk.host, zk.port, zk_chroot='', + KafkaFixture.instance(i, zk, zk_chroot='', partitions=partitions, replicas=replicas) for i in range(n) ] @@ -148,7 +148,7 @@ def get_args_parser(): help='Topic for consumer test', default='kafka-python-benchmark-test') parser.add_argument( - '--num-records', type=long, + '--num-records', type=int, help='number of messages to consume', default=1000000) parser.add_argument( |