refactor: Make it more robust

lablup · Jul 16, 2024 · 30226e4 · 30226e4
1 parent c4a9353
commit 30226e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ai/backend/kernel/utils.py b/src/ai/backend/kernel/utils.py
@@ -52,7 +52,7 @@ def filter(self, record: logging.LogRecord) -> bool:
                     filtered_traceback.append(tb)
             lines = ["  Traceback:"]
             for tb in filtered_traceback:
-                short_path = Path(tb.filename).relative_to(self.site_prefix)
+                short_path = tb.filename.removeprefix(self.site_prefix).removeprefix("/")
                 lines.append(f"    {short_path} (L{tb.lineno}): {tb.name}()")
             record.exc_text = "\n".join(lines)
         return True