def client_connection(): (cluster, client) = setup_local_dask_cluster(p2p=True) yield client teardown_local_dask_cluster(cluster, client)
def teardown_module(): if not _is_single_gpu: teardown_local_dask_cluster(_cluster, _client)