diff --git a/cuda_core/cuda/core/_utils/enum_explanations_helpers.py b/cuda_core/cuda/core/_utils/enum_explanations_helpers.py index c7927e71e42..2b110c41d6f 100644 --- a/cuda_core/cuda/core/_utils/enum_explanations_helpers.py +++ b/cuda_core/cuda/core/_utils/enum_explanations_helpers.py @@ -35,9 +35,11 @@ def _binding_version() -> tuple[int, int, int]: """Return the installed ``cuda-bindings`` version, or a conservative old value.""" try: - parts = importlib.metadata.version("cuda-bindings").split(".")[:3] + version = importlib.metadata.version("cuda-bindings") except importlib.metadata.PackageNotFoundError: return (0, 0, 0) # For very old versions of cuda-python + + parts = version.partition("+")[0].split(".")[:3] return tuple(int(v) for v in parts)