[HOTFIX] Generalize ModuleNotFoundError exception handling to ImportError for Amazon SMS libnuma.so.1 bug (#2385)

Generalize ModuleNotFoundError exception handling to ImportError for Amazon SMS libnuma.so.1 bug
#2113

This allows cugraph to be imported in a SageMaker environment without having to remove `ucx-py`

Authors:
   - Dylan Chima-Sanchez (https://github.com/betochimas)
   - Rick Ratzel (https://github.com/rlratzel)

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

Author: rlratzel

conda/recipes/libcugraph/meta.yaml

@@ -41,7 +41,7 @@ requirements:
     - libraft-headers {{ minor_version }}.*
     - libcugraphops {{ minor_version }}.*
     - librmm {{ minor_version }}.*
-    - cudf {{ minor_version }}.*
+    - libcudf {{ minor_version }}.*
     - boost-cpp {{ boost_cpp_version }}
     - nccl {{ nccl_version }}
     - ucx-proc=*=gpu

python/cugraph/cugraph/dask/common/input_utils.py

@@ -25,8 +25,10 @@
 from cugraph.dask.common.read_utils import MissingUCXPy
 try:
     from raft.dask.common.utils import get_client
-except ModuleNotFoundError as err:
-    if err.name == "ucp":
+except ImportError as err:
+    # FIXME: Generalize since err.name is arr when
+    # libnuma.so.1 is not available
+    if err.name == "ucp" or err.name == "arr":
         get_client = MissingUCXPy()
     else:
         raise

python/cugraph/cugraph/dask/common/mg_utils.py

@@ -23,8 +23,10 @@
 from cugraph.dask.common.read_utils import MissingUCXPy
 try:
     from raft.dask.common.utils import default_client
-except ModuleNotFoundError as err:
-    if err.name == "ucp":
+except ImportError as err:
+    # FIXME: Generalize since err.name is arr when
+    # libnuma.so.1 is not available
+    if err.name == "ucp" or err.name == "arr":
         default_client = MissingUCXPy()
     else:
         raise

python/cugraph/cugraph/dask/comms/comms.py

@@ -17,8 +17,10 @@
 try:
     from raft.dask.common.comms import Comms as raftComms
     from raft.dask.common.comms import get_raft_comm_state
-except ModuleNotFoundError as err:
-    if err.name == "ucp":
+except ImportError as err:
+    # FIXME: Generalize since err.name is arr when
+    # libnuma.so.1 is not available
+    if err.name == "ucp" or err.name == "arr":
         raftComms = MissingUCXPy()
         get_raft_comm_state = MissingUCXPy()
     else: