Fix names.

Co-authored-by: Matthew Murray <[email protected]>
rapidsai · Dec 9, 2024 · 3124c0e · 3124c0e
1 parent 4871535
commit 3124c0e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/python/rmm/rmm/_cuda/gpu.py b/python/rmm/rmm/_cuda/gpu.py
@@ -1,7 +1,6 @@
 # Copyright (c) 2020-2024, NVIDIA CORPORATION.
 
-import cuda.bindings.driver as driver
-import cuda.bindings.runtime as runtime
+from cuda.bindings import driver, runtime
 
 
 class CUDARuntimeError(RuntimeError):