Skip to content

Commit

Permalink
Update version error message
Browse files Browse the repository at this point in the history
Update version error message to match the minimum support UCX version
and provide instructions to check where UCX is being loaded from.
  • Loading branch information
pentschev committed Sep 11, 2024
1 parent 2ba56e5 commit 26192c9
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions ucp/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -108,10 +108,13 @@ def _is_mig_device(handle):
os.environ["UCX_PROTO_ENABLE"] = "n"


__ucx_min_version__ = "1.15.0"
__ucx_version__ = "%d.%d.%d" % get_ucx_version()

if get_ucx_version() < (1, 15, 0):
if get_ucx_version() < tuple([int(i) for i in __ucx_min_version__.split(".")]):
raise ImportError(
f"Support for UCX {__ucx_version__} has ended. Please upgrade to "
"1.11.1 or newer."
f"{__ucx_min_version__} or newer. If you believe the wrong version "
"is being loaded, please check the path from where UCX is loaded "
"by rerunning with the environment variable `UCX_LOG_LEVEL=debug`."
)

0 comments on commit 26192c9

Please sign in to comment.