Support config= parameter in gen_cluster
@@ -726,7 +726,7 @@ def iscoroutinefunction(f):
def gen_cluster(ncores=[('127.0.0.1', 1), ('127.0.0.1', 2)],
scheduler='127.0.0.1', timeout=10, security=None,
Worker=Worker, client=False, scheduler_kwargs={},
- worker_kwargs={}, active_rpc_timeout=1):
+ worker_kwargs={}, active_rpc_timeout=1, config={}):
from distributed import Client
""" Coroutine test with small cluster
@@ -763,6 +763,7 @@ def gen_cluster(ncores=[('127.0.0.1', 1), ('127.0.0.1', 2)],
with check_active_rpc(loop, active_rpc_timeout):
@gen.coroutine
def coro():
+ with dask.config.set(config):
for i in range(5):
try:
s, ws = yield start_cluster(
