Fix type of arg

rapidsai · Jan 13, 2022 · 84994cb · 84994cb
1 parent abd3022
commit 84994cb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/rmm/_cuda/gpu.py b/python/rmm/_cuda/gpu.py
@@ -4,7 +4,7 @@
 
 
 class CUDARuntimeError(RuntimeError):
-    def __init__(self, status: cuda.CUresult.CUDA_SUCCESS):
+    def __init__(self, status: cudart.cudaError_t):
         self.status = status
         _, name = cudart.cudaGetErrorName(status)
         _, msg = cudart.cudaGetErrorString(status)