Update pynvjitlink/patch.py

Co-authored-by: Graham Markall <[email protected]>
rapidsai · Feb 19, 2025 · 5d3448f · 5d3448f
1 parent 3a203d8
commit 5d3448f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pynvjitlink/patch.py b/pynvjitlink/patch.py
@@ -55,7 +55,7 @@
 
     import numba_cuda
 
-    numba_cuda_ver = (int(x) for x in numba_cuda.__version__.split("."))
+    numba_cuda_ver = tuple(int(x) for x in numba_cuda.__version__.split("."))
     if numba_cuda_ver < (0, 2, 0):
         suggestion = "Instead, use NUMBA_CUDA_ENABLE_PYNVJITLINK environment variable to enable pynvjitlink features."
     else: