Update ci/notebook_list.py

rapidsai · Jan 9, 2025 · c7472d4 · c7472d4
1 parent 1123cf2
commit c7472d4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ci/notebook_list.py b/ci/notebook_list.py
@@ -17,6 +17,8 @@
 import glob
 from pathlib import Path
 
+from cuda.bindings import runtime
+
 # for adding another run type and skip file name add to this dictionary
 runtype_dict = {
     "all": "",