Upgrade ruff==0.13.3 pyright==1.1.406

.pre-commit-config.yaml

@@ -20,7 +20,7 @@ repos:
 
 -   repo: https://github.com/astral-sh/ruff-pre-commit
     # Ruff version.
-    rev: v0.13.1
+    rev: v0.13.3
     hooks:
       # Run the linter.
       - id: ruff
@@ -44,7 +44,7 @@ repos:
           - tomli
 
 -   repo: https://github.com/RobertCraigie/pyright-python
-    rev: v1.1.405
+    rev: v1.1.406
     hooks:
       - id: pyright
         language: system

benchmarks/run.py

@@ -26,7 +26,7 @@
 import dataclasses
 import functools
 import gc
-import importlib
+import importlib.util
 import json
 import logging
 import os
@@ -39,16 +39,16 @@
 from typing import Any
 from typing import Callable
 
-if TYPE_CHECKING:
-    from tritonbench.utils.triton_op import BenchmarkOperator
-    from tritonbench.utils.triton_op import BenchmarkOperatorMetrics
-
 import torch
 from torch.utils._pytree import tree_leaves
 from torch.utils._pytree import tree_map
 
 from helion._utils import counters
 
+if TYPE_CHECKING:
+    from tritonbench.utils.triton_op import BenchmarkOperator
+    from tritonbench.utils.triton_op import BenchmarkOperatorMetrics
+
 try:
     from tritonbench.utils.env_utils import get_nvidia_gpu_model
     from tritonbench.utils.env_utils import is_cuda
@@ -572,12 +572,8 @@ def get_system_memory_gb() -> float:
 def check_and_setup_tritonbench() -> None:
     """Check if tritonbench is installed and install it from GitHub if not."""
     # Check if tritonbench is already installed
-    try:
-        import tritonbench  # pyright: ignore[reportMissingImports]
-
+    if importlib.util.find_spec("tritonbench") is not None:
         return  # Already installed
-    except ImportError:
-        pass
 
     print("Tritonbench not found. Installing...", file=sys.stderr)
 

lint.sh

@@ -9,7 +9,7 @@ fi
 if [ "$ACTION" = "install" ];
 then
     set -ex
-    pip install ruff==0.13.1 pyright==1.1.405
+    pip install ruff==0.13.3 pyright==1.1.406
     exit 0
 fi