Skip to content

Commit

Permalink
add non p2p dask client
Browse files Browse the repository at this point in the history
  • Loading branch information
jnke2016 committed Jul 18, 2024
1 parent 17355f8 commit 41474a1
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions python/cugraph/cugraph/tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,6 +52,21 @@ def dask_client():
stop_dask_client(dask_client, dask_cluster)


@pytest.fixture(scope="module")
def dask_client_non_p2p():
# start_dask_client will check for the SCHEDULER_FILE and
# DASK_WORKER_DEVICES env vars and use them when creating a client if
# set. start_dask_client will also initialize the Comms singleton.
dask_client, dask_cluster = start_dask_client(
worker_class=IncreasedCloseTimeoutNanny,
p2p=False
)

yield dask_client

stop_dask_client(dask_client, dask_cluster)


@pytest.fixture(scope="module")
def scratch_dir():
# This should always be set if doing MG testing, since temporary
Expand Down

0 comments on commit 41474a1

Please sign in to comment.