Fix lint error on main (#281)

Author: jansel

.github/workflows/lint.yml

@@ -49,7 +49,7 @@ jobs:
       - name: Install pre-commit
         run: |
           python -m pip install --upgrade pip
-          pip install pre-commit
+          pip install .'[dev]'
 
       - name: Run pre-commit
         run: pre-commit run --all-files

benchmarks/run.py

@@ -243,7 +243,9 @@ def main() -> None:
     tb_args, unknown_args = tb_parser.parse_known_args(tritonbench_args)
 
     # Register the Helion kernel with tritonbench BEFORE importing the operator
-    from tritonbench.utils.triton_op import register_benchmark
+    from tritonbench.utils.triton_op import (  # type: ignore[reportMissingImports]
+        register_benchmark,
+    )
 
     # Create the benchmark method
     def create_helion_method(

pyproject.toml

@@ -20,6 +20,7 @@ dependencies = [
     "torch>=2.7.0",
     "typing_extensions>=4.0.0",
     "filecheck",
+    "psutil"
 ]
 
 [project.optional-dependencies]