Add missing imports (related to NVIDIA#769). These got lost due to merging with -X ours

rwgk · rwgk · commit 29628eaef60e · 2025-07-23T15:03:15.000-07:00
diff --git a/cuda_bindings/tests/test_cuda.py b/cuda_bindings/tests/test_cuda.py
@@ -10,6 +10,7 @@
 
 import cuda.bindings.driver as cuda
 import cuda.bindings.runtime as cudart
+from cuda.bindings import driver
 
 
 def driverVersionLessThan(target):
diff --git a/cuda_bindings/tests/test_cudart.py b/cuda_bindings/tests/test_cudart.py
@@ -9,6 +9,7 @@
 
 import cuda.bindings.driver as cuda
 import cuda.bindings.runtime as cudart
+from cuda.bindings import runtime
 
 
 def isSuccess(err):
