From 2995c628a0fe9d20f946e1666e25906570efeb24 Mon Sep 17 00:00:00 2001 From: Alexandria Barghi Date: Mon, 22 Dec 2025 07:36:39 -0800 Subject: [PATCH 1/2] improve missingucxx error --- .../pylibcugraph/pylibcugraph/comms/cugraph_nccl_comms.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/python/pylibcugraph/pylibcugraph/comms/cugraph_nccl_comms.py b/python/pylibcugraph/pylibcugraph/comms/cugraph_nccl_comms.py index be2f2f6e1f..665a06b719 100644 --- a/python/pylibcugraph/pylibcugraph/comms/cugraph_nccl_comms.py +++ b/python/pylibcugraph/pylibcugraph/comms/cugraph_nccl_comms.py @@ -20,6 +20,12 @@ def __call__(self, *args, **kwargs): " but are required for MG operations" ) + def __getattr__(self, name): + raise ModuleNotFoundError( + "raft-dask and/or ucxx could not be imported" + " but are required for MG operations" + ) + nccl = MissingUCXX() inject_comms_on_handle_coll_only = MissingUCXX() From 3d1fd37778beec75259254a7d2fc2debdced62f3 Mon Sep 17 00:00:00 2001 From: Alexandria Barghi Date: Mon, 5 Jan 2026 09:51:31 -0800 Subject: [PATCH 2/2] improve error message --- .../pylibcugraph/pylibcugraph/comms/cugraph_nccl_comms.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/pylibcugraph/pylibcugraph/comms/cugraph_nccl_comms.py b/python/pylibcugraph/pylibcugraph/comms/cugraph_nccl_comms.py index 665a06b719..c825640ec0 100644 --- a/python/pylibcugraph/pylibcugraph/comms/cugraph_nccl_comms.py +++ b/python/pylibcugraph/pylibcugraph/comms/cugraph_nccl_comms.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 2024-2025, NVIDIA CORPORATION. +# SPDX-FileCopyrightText: Copyright (c) 2024-2026, NVIDIA CORPORATION. # SPDX-License-Identifier: Apache-2.0 import math @@ -17,13 +17,13 @@ class MissingUCXX: def __call__(self, *args, **kwargs): raise ModuleNotFoundError( "raft-dask and/or ucxx could not be imported" - " but are required for MG operations" + " but are required for multi-GPU operations" ) def __getattr__(self, name): raise ModuleNotFoundError( "raft-dask and/or ucxx could not be imported" - " but are required for MG operations" + " but are required for multi-GPU operations" ) nccl = MissingUCXX()