Use divmod.

Co-authored-by: jakirkham <[email protected]>
rapidsai · Dec 11, 2024 · 8080bc8 · 8080bc8
1 parent 07118bd
commit 8080bc8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ci/notebook_list.py b/ci/notebook_list.py
@@ -37,8 +37,8 @@ def _get_cuda_version_string():
     status, version = runtime.getLocalRuntimeVersion()
     if status != runtime.cudaError_t.cudaSuccess:
         raise RuntimeError("Could not get CUDA runtime version.")
-    major = version // 1000
-    minor = (version % 1000) // 10
+    major, minor = divmod(version, 1000)
+    minor //= 10
     return f"{major}.{minor}"