Fix style checks

rapidsai · Jan 12, 2024 · 441364d · 441364d
1 parent 3c7746f
commit 441364d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/ci/test_patch.py b/ci/test_patch.py
@@ -2,9 +2,10 @@
 # Copyright (c) 2024, NVIDIA CORPORATION
 
 from pynvjitlink import patch
+
 patch.patch_numba_linker()
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     from numba.testing._runtests import _main
     import sys
 

diff --git a/pynvjitlink/patch.py b/pynvjitlink/patch.py
@@ -30,8 +30,12 @@
 if _numba_version_ok:
     from numba.core import config
     from numba.cuda.cudadrv import nvrtc
-    from numba.cuda.cudadrv.driver import (driver, FILE_EXTENSION_MAP, Linker,
-                                           LinkerError)
+    from numba.cuda.cudadrv.driver import (
+        driver,
+        FILE_EXTENSION_MAP,
+        Linker,
+        LinkerError,
+    )
 else:
     # Prevent the definition of PatchedLinker failing if we have no Numba
     # Linker - it won't be used anyway.