def add_doc_gen_task(tm, rest, count, bucket="default", kv_store=None, store_enabled=True, kv_template=None, seed=None, sizes=None, expiration=None, loop=False, monitor=False, doc_generators=None): doc_generators = doc_generators or DocumentGenerator.get_doc_generators(count, kv_template, seed, sizes) _t = task.LoadDocGeneratorTask(rest, doc_generators, bucket=bucket, kv_store=kv_store, store_enabled=store_enabled, expiration=expiration, loop=loop) return RebalanceTaskHelper.schedule_task_helper(tm, _t, monitor)