diff options
Diffstat (limited to 'benchmarks/producer_performance.py')
| -rwxr-xr-x | benchmarks/producer_performance.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/benchmarks/producer_performance.py b/benchmarks/producer_performance.py index 0c29cbc..c0de6fd 100755 --- a/benchmarks/producer_performance.py +++ b/benchmarks/producer_performance.py @@ -26,7 +26,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) ] @@ -130,7 +130,7 @@ def get_args_parser(): help='Topic name for test', default='kafka-python-benchmark-test') parser.add_argument( - '--num-records', type=long, + '--num-records', type=int, help='number of messages to produce', default=1000000) parser.add_argument( |