Fix for ruff

Signed-off-by: Thomas Parnell <tpa@zurich.ibm.com>
IBM · May 29, 2024 · ed54561 · ed54561
1 parent d5910bb
commit ed54561
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/triton_patch/custom_cache_manager.py b/triton_patch/custom_cache_manager.py
@@ -1,5 +1,11 @@
 import os
-from triton.runtime.cache import FileCacheManager, default_dump_dir, default_override_dir, default_cache_dir
+from triton.runtime.cache import (
+    FileCacheManager,
+    default_dump_dir,
+    default_override_dir,
+    default_cache_dir
+)
+
 
 class CustomCacheManager(FileCacheManager):
 
@@ -16,7 +22,8 @@ def __init__(self, key, override=False, dump=False):
             self.cache_dir = os.path.join(self.cache_dir, self.key)
         else:
             # create cache directory if it doesn't exist
-            self.cache_dir = os.getenv("TRITON_CACHE_DIR", "").strip() or default_cache_dir()
+            self.cache_dir = os.getenv("TRITON_CACHE_DIR",
+                                       "").strip() or default_cache_dir()
             self.cache_dir = self.cache_dir + "_" + str(os.getpid())
             print(f"{self.cache_dir=}")
             if self.cache_dir: