Update raft import (#2729)

This PR updates the way `raft` is imported following changes on the `raft` new conda packages.

closes #2727

Authors:
  - Joseph Nke (https://github.com/jnke2016)

Approvers:
  - Rick Ratzel (https://github.com/rlratzel)
  - Brad Rees (https://github.com/BradReesWork)

URL: #2729

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,

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

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

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