Merge pull request #530 from numpy/astral-sh/setup-uv@v6

Author: jorenham

.github/workflows/ci.yml

@@ -42,9 +42,7 @@ jobs:
           taplo validate --colors never --no-schema
           taplo fmt --colors never --check
 
-      - uses: astral-sh/setup-uv@v5
-        with:
-          enable-cache: true
+      - uses: astral-sh/setup-uv@v6
       - run: uv sync --only-group=list_and_test
 
       - name: ruff
@@ -68,7 +66,7 @@ jobs:
     steps:
       - uses: actions/checkout@v4
 
-      - uses: astral-sh/setup-uv@v5
+      - uses: astral-sh/setup-uv@v6
         with:
           enable-cache: true
           python-version: ${{ matrix.py }}
@@ -86,9 +84,8 @@ jobs:
     steps:
       - uses: actions/checkout@v4
 
-      - uses: astral-sh/setup-uv@v5
+      - uses: astral-sh/setup-uv@v6
         with:
-          enable-cache: true
           python-version: ${{ matrix.py }}
 
       - name: mypy
@@ -107,12 +104,9 @@ jobs:
     steps:
       - uses: actions/checkout@v4
 
-      - uses: astral-sh/setup-uv@v5
+      - uses: astral-sh/setup-uv@v6
         with:
-          enable-cache: true
           python-version: ${{ matrix.py }}
 
       - name: stubtest
-        run: >
-          uv run --no-dev --active -p ${{ matrix.py }}
-          tool/stubtest.py
+        run: uv run --active -p ${{ matrix.py }} tool/stubtest.py

tool/stubtest.py

@@ -30,14 +30,19 @@
 CWD = Path.cwd()
 TOOL_DIR = Path(__file__).parent
 SITE_DIR = Path(sysconfig.get_paths()["purelib"])
-ROOT_DIR = TOOL_DIR.parent
-
 
-ROOT_SITE_DIR = (
-    ROOT_DIR / ".venv" / "Lib" / "site-packages"
-    if sys.platform == "win32"
-    else next((ROOT_DIR / ".venv" / "lib").glob("*/site-packages"))
-)
+ROOT_DIR = TOOL_DIR.parent
+__root_site: Path | None
+if (ROOT_DIR / ".venv").is_dir():
+    __root_venv = ROOT_DIR / ".venv"
+    __root_site = (
+        __root_venv / "Lib" / "site-packages"
+        if sys.platform == "win32"
+        else next((__root_venv / "lib").glob("*/site-packages"))
+    )
+else:
+    __root_site = None
+ROOT_SITE_DIR = __root_site
 
 ALLOWLISTS = [
     "common.txt",
@@ -130,9 +135,10 @@ def _rewrite_mypy_output(line: bytes, /) -> bytes:
         if package_lib in line:
             line = line.replace(package_lib, package_src)
 
-    site_dir = str(SITE_DIR).encode()
-    if site_dir in line:
-        line = line.replace(site_dir, str(ROOT_SITE_DIR.relative_to(CWD)).encode())
+    if ROOT_SITE_DIR is not None:
+        site_dir = str(SITE_DIR).encode()
+        if site_dir in line:
+            line = line.replace(site_dir, str(ROOT_SITE_DIR.relative_to(CWD)).encode())
 
     return line