diff --git a/dask_cuda/local_cuda_cluster.py b/dask_cuda/local_cuda_cluster.py index 1b55264c0..fc82d39fc 100644 --- a/dask_cuda/local_cuda_cluster.py +++ b/dask_cuda/local_cuda_cluster.py @@ -157,11 +157,9 @@ def __init__( self, n_workers=None, threads_per_worker=1, - processes=True, memory_limit="auto", device_memory_limit=0.8, CUDA_VISIBLE_DEVICES=None, - data=None, local_directory=None, protocol=None, enable_tcp_over_ucx=False, @@ -225,7 +223,7 @@ def __init__( self.rmm_log_directory = rmm_log_directory - if not processes: + if not kwargs.pop("processes", True): raise ValueError( "Processes are necessary in order to use multiple GPUs with Dask" ) @@ -235,6 +233,7 @@ def __init__( else: self.jit_unspill = jit_unspill + data = kwargs.pop("data", None) if data is None: if self.jit_unspill: data = ( diff --git a/dask_cuda/tests/test_explicit_comms.py b/dask_cuda/tests/test_explicit_comms.py index 05edbfb8b..87beefb67 100644 --- a/dask_cuda/tests/test_explicit_comms.py +++ b/dask_cuda/tests/test_explicit_comms.py @@ -265,7 +265,6 @@ def _test_jit_unspill(protocol): dashboard_address=None, n_workers=1, threads_per_worker=1, - processes=True, jit_unspill=True, device_memory_limit="1B", ) as cluster: diff --git a/dask_cuda/tests/test_spill.py b/dask_cuda/tests/test_spill.py index dea4b3d3d..95e161468 100644 --- a/dask_cuda/tests/test_spill.py +++ b/dask_cuda/tests/test_spill.py @@ -195,7 +195,6 @@ async def test_cupy_cluster_device_spill(params): async with LocalCUDACluster( 1, scheduler_port=0, - processes=True, silence_logs=False, dashboard_address=None, asynchronous=True,