enable CI with TRITON_INTERPRET=1; format trace.py using pre-commit

Author: mark14wu

.github/workflows/python-app.yml

@@ -5,11 +5,13 @@ name: Python application
 
 on:
   push:
-    branches-ignore:
-      - '**'
+    branches:
+      - main
+      - keren/v2.0
   pull_request:
-    branches-ignore:
-      - '**'
+    branches:
+      - main
+      - keren/v2.0
 
 permissions:
   contents: read
@@ -54,6 +56,8 @@ jobs:
         pip install -e .
 
     - name: Test with pytest
+      env: |
+        TRITON_INTERPRET=1
       run: |
         cd triton_viz
         python -m pytest tests

triton_viz/core/trace.py

@@ -34,7 +34,11 @@ def add_client(self, new_client: Union[Client, str]) -> None:
         new_client_instance = self._normalize_client(new_client)
         self.client_manager.add_clients([new_client_instance])
 
-    def __init__(self, kernel: Union[JITFunction, InterpretedFunction], client: Union[str, Client]) -> None:
+    def __init__(
+        self,
+        kernel: Union[JITFunction, InterpretedFunction],
+        client: Union[str, Client],
+    ) -> None:
         self.fn = kernel
         if isinstance(kernel, InterpretedFunction):
             self.interpreter_fn = kernel
@@ -91,7 +95,9 @@ def decorator(kernel) -> Trace:
             trace.add_client(clients)
             return trace
 
-        raise TypeError(f"Expected JITFunction, InterpretedFunction or Trace, got {type(kernel)}")
+        raise TypeError(
+            f"Expected JITFunction, InterpretedFunction or Trace, got {type(kernel)}"
+        )
 
     return decorator