From 4c3f295dc726b342661f7b846e02aee5fbddea56 Mon Sep 17 00:00:00 2001 From: Charles Coulombe Date: Fri, 15 May 2026 14:23:23 -0400 Subject: [PATCH] Refactor version retrieval for cuda-bindings partition the local version if present --- cuda_core/cuda/core/_utils/enum_explanations_helpers.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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)