diff --git a/benchmarks/python_e2e/cugraph_dask_funcs.py b/benchmarks/python_e2e/cugraph_dask_funcs.py index c8f48936647..9ff22a9f248 100644 --- a/benchmarks/python_e2e/cugraph_dask_funcs.py +++ b/benchmarks/python_e2e/cugraph_dask_funcs.py @@ -182,6 +182,9 @@ def setup(dask_scheduler_file=None, rmm_pool_size=None): def teardown(client, cluster=None): Comms.destroy() - client.close() + # Shutdown the connected scheduler and workers + # therefore we will no longer rely on killing the dask cluster ID + # for MNMG runs + client.shutdown() if cluster: cluster.close() diff --git a/python/cugraph/cugraph/tests/conftest.py b/python/cugraph/cugraph/tests/conftest.py index 5c507ed28c6..046aa629286 100644 --- a/python/cugraph/cugraph/tests/conftest.py +++ b/python/cugraph/cugraph/tests/conftest.py @@ -58,7 +58,10 @@ def dask_client(): yield client Comms.destroy() - client.close() + # Shut down the connected scheduler and workers + # therefore we will no longer rely on killing the dask cluster ID + # for MNMG runs + client.shutdown() if cluster: cluster.close() print("\ndask_client fixture: client.close() called")