diff --git a/python/cugraph/cugraph/__init__.py b/python/cugraph/cugraph/__init__.py index c0f65d400ce..00ed23ee601 100644 --- a/python/cugraph/cugraph/__init__.py +++ b/python/cugraph/cugraph/__init__.py @@ -106,7 +106,7 @@ from cugraph.linear_assignment import hungarian, dense_hungarian from cugraph.layout import force_atlas2 -from raft import raft_include_test +from raft_dask import raft_include_test from cugraph.sampling import ( random_walks, diff --git a/python/cugraph/cugraph/dask/common/input_utils.py b/python/cugraph/cugraph/dask/common/input_utils.py index 9db3964c021..9dd34d6161a 100644 --- a/python/cugraph/cugraph/dask/common/input_utils.py +++ b/python/cugraph/cugraph/dask/common/input_utils.py @@ -24,7 +24,7 @@ # not available. They are necessary only when doing MG work. from cugraph.dask.common.read_utils import MissingUCXPy try: - from raft.dask.common.utils import get_client + from raft_dask.common.utils import get_client except ImportError as err: # FIXME: Generalize since err.name is arr when # libnuma.so.1 is not available diff --git a/python/cugraph/cugraph/dask/common/mg_utils.py b/python/cugraph/cugraph/dask/common/mg_utils.py index 99d80b3c659..eadd7185904 100644 --- a/python/cugraph/cugraph/dask/common/mg_utils.py +++ b/python/cugraph/cugraph/dask/common/mg_utils.py @@ -22,7 +22,7 @@ # not available. They are necessary only when doing MG work. from cugraph.dask.common.read_utils import MissingUCXPy try: - from raft.dask.common.utils import default_client + from raft_dask.common.utils import default_client except ImportError as err: # FIXME: Generalize since err.name is arr when # libnuma.so.1 is not available diff --git a/python/cugraph/cugraph/dask/comms/comms.py b/python/cugraph/cugraph/dask/comms/comms.py index dded0766e04..9485e75e8fb 100644 --- a/python/cugraph/cugraph/dask/comms/comms.py +++ b/python/cugraph/cugraph/dask/comms/comms.py @@ -15,8 +15,8 @@ # not available. They are necessary only when doing MG work. from cugraph.dask.common.read_utils import MissingUCXPy try: - from raft.dask.common.comms import Comms as raftComms - from raft.dask.common.comms import get_raft_comm_state + from raft_dask.common.comms import Comms as raftComms + from raft_dask.common.comms import get_raft_comm_state except ImportError as err: # FIXME: Generalize since err.name is arr when # libnuma.so.1 is not available