From ce181cb0a4338d5105668b07611cdc5938c260af Mon Sep 17 00:00:00 2001
From: Jason Ansel <jansel@meta.com>
Date: Thu, 2 Oct 2025 18:52:49 -0700
Subject: [PATCH] Upgrade ruff==0.13.3 pyright==1.1.406

stack-info: PR: https://github.com/pytorch/helion/pull/790, branch: jansel/stack/156
---
 .pre-commit-config.yaml |  4 ++--
 benchmarks/run.py       | 16 ++++++----------
 lint.sh                 |  2 +-
 3 files changed, 9 insertions(+), 13 deletions(-)

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 4486ad0e0..0f9b05018 100644
--- a/.pre-commit-config.yaml
+++ b/.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
diff --git a/benchmarks/run.py b/benchmarks/run.py
index 4f9f3ebfb..b2db48034 100644
--- a/benchmarks/run.py
+++ b/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)
 
diff --git a/lint.sh b/lint.sh
index 6355343ce..ad4f9946f 100755
--- a/lint.sh
+++ b/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