Add flake8 pre-commit check and fix redirector violation

NVIDIA · Aug 1, 2024 · 7487e86 · 7487e86
1 parent e0514d7
commit 7487e86
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -1 +1,5 @@
-repos: []
+repos:
+- repo: https://github.com/PyCQA/flake8
+  rev: 7.1.0
+  hooks:
+  - id: flake8
diff --git a/site-packages/_numba_cuda_redirector.py b/site-packages/_numba_cuda_redirector.py
@@ -47,8 +47,10 @@ def ensure_initialized(self):
             return False
 
         self.numba_path = numba_search_locations[0]
-        self.numba_cuda_path = str((pathlib.Path(numba_cuda_search_locations[0]) /
-                                    'numba'))
+
+        location = numba_cuda_search_locations[0]
+        self.numba_cuda_path = str((pathlib.Path(location) / 'numba'))
+
         self.initialized = True
         return True