fixup! perf: Prune function bodies from source before building AST

Author: pawamoy

.github/workflows/ci.yml

@@ -75,9 +75,6 @@ jobs:
     - name: Install dependencies
       run: make setup
 
-    - name: Install prune-source accelerator
-      run: uv sync --extra native
-
     - name: Check if the documentation builds correctly
       run: make check-docs
 
@@ -153,11 +150,6 @@ jobs:
         UV_RESOLUTION: ${{ matrix.resolution }}
       run: make setup
 
-    - name: Install prune-source accelerator
-      env:
-        UV_RESOLUTION: ${{ matrix.resolution }}
-      run: uv sync --extra native
-
     - name: Download objects inventory
       uses: actions/download-artifact@v8
       with:

README.md

@@ -32,7 +32,7 @@ uv tool install griffe
 For faster static analysis, install the optional `prune-source` Rust/Ruff accelerator:
 
 ```bash
-pip install "griffe[native]"
+pip install "griffe[faster]"
 ```
 
 Griffe automatically imports the accelerator when available and falls back to its full CPython AST

docs/installation.md

@@ -2,18 +2,18 @@
 
 Griffe is a Python package, so you can install it with your favorite Python package installer or dependency manager.
 
-## Install the native parser accelerator
+## Install the parser accelerator
 
 Static analysis can use the optional `prune-source` Rust parser accelerator:
 
 === ":simple-python: pip"
     ```bash
-    pip install "griffe[native]"
+    pip install "griffe[faster]"
     ```
 
 === ":simple-astral: uv"
     ```bash
-    uv add "griffe[native]"
+    uv add "griffe[faster]"
     ```
 
 To install only the library, replace `griffe` with `griffelib` in either command. The separately

packages/griffelib/pyproject.toml

@@ -37,9 +37,9 @@ classifiers = [
 ]
 
 [project.optional-dependencies]
-# The 'native' extra installs the Rust/Ruff parser accelerator. Griffe automatically falls back to
+# The 'faster' extra installs the Rust/Ruff parser accelerator. Griffe automatically falls back to
 # CPython's parser when the accelerator is unavailable or an extension needs concrete AST nodes.
-native = ["prune-source>=0.1.0,<0.2"]
+faster = ["prune-source>=0.1.0,<0.2"]
 # The 'pypi' extra provides dependencies needed for the load_pypi functionality
 # to download and inspect packages from PyPI.
 pypi = ["pip>=24.0", "platformdirs>=4.2", "wheel>=0.42"]

packages/griffelib/src/griffe/_internal/agents/parser.py

@@ -17,8 +17,6 @@
 from __future__ import annotations
 
 import ast
-import sys
-from importlib import import_module
 from typing import TYPE_CHECKING, Final
 
 if TYPE_CHECKING:

packages/griffelib/tests/test_parser.py

@@ -54,7 +54,7 @@ def test_native_parser_is_used_without_node_aware_hooks(monkeypatch: pytest.Monk
     """Use Ruff when installed extensions do not require the concrete source AST."""
     calls = 0
 
-    def prune(code: str, python_minor: int) -> str:  # noqa: ARG001
+    def prune(code: str) -> str:
         nonlocal calls
         calls += 1
         return code
@@ -69,7 +69,7 @@ def test_native_parser_is_skipped_without_functions(monkeypatch: pytest.MonkeyPa
     """Avoid paying for a second parse when no implementation suite can be removed."""
     calls = 0
 
-    def prune(code: str, python_minor: int) -> str:  # noqa: ARG001
+    def prune(code: str) -> str:
         nonlocal calls
         calls += 1
         return code
@@ -84,7 +84,7 @@ def test_native_parser_is_skipped_for_stub_files(monkeypatch: pytest.MonkeyPatch
     """Avoid pruning stub files, whose function bodies contain no implementations."""
     calls = 0
 
-    def prune(code: str, python_minor: int) -> str:  # noqa: ARG001
+    def prune(code: str) -> str:
         nonlocal calls
         calls += 1
         return code
@@ -99,7 +99,7 @@ def test_node_aware_hooks_force_cpython_fallback(monkeypatch: pytest.MonkeyPatch
     """Keep complete CPython nodes available to visit-time extension callbacks."""
     calls = 0
 
-    def prune(code: str, python_minor: int) -> str:  # noqa: ARG001
+    def prune(code: str) -> str:
         nonlocal calls
         calls += 1
         return code
@@ -114,7 +114,7 @@ def test_custom_visitors_can_disable_native_pruning(monkeypatch: pytest.MonkeyPa
     """Keep the complete tree available to Visitor subclasses with custom traversal."""
     calls = 0
 
-    def prune(code: str, python_minor: int) -> str:  # noqa: ARG001
+    def prune(code: str) -> str:
         nonlocal calls
         calls += 1
         return code
@@ -131,7 +131,7 @@ class CustomVisitor(Visitor):
 
 def test_cpython_retries_an_invalid_native_result(monkeypatch: pytest.MonkeyPatch) -> None:
     """Never expose a diagnostic caused by native source pruning."""
-    monkeypatch.setattr(parser_module, "prune_source", lambda code, python_minor: "def invalid(: pass")
+    monkeypatch.setattr(parser_module, "prune_source", lambda code: "def invalid(: pass")
 
     node = parser_module._compile_module("def valid(): pass", filename="module.py", extensions=Extensions())
 
@@ -140,7 +140,7 @@ def test_cpython_retries_an_invalid_native_result(monkeypatch: pytest.MonkeyPatc
 
 def test_cpython_compiles_untouched_source_when_native_declines(monkeypatch: pytest.MonkeyPatch) -> None:
     """Use the normal source when Ruff cannot prune it."""
-    monkeypatch.setattr(parser_module, "prune_source", lambda code, python_minor: None)
+    monkeypatch.setattr(parser_module, "prune_source", lambda code: None)
 
     node = parser_module._compile_module("def valid(): pass", filename="module.py", extensions=Extensions())
 

pyproject.toml

@@ -64,8 +64,8 @@ expression = "get_version()"
 dependencies = ["griffelib=={{version}}", "griffecli=={{version}}"]
 
 [tool.hatch.metadata.hooks.uv-dynamic-versioning.optional-dependencies]
-# The 'native' extra re-exports griffelib[native] for backward compatibility.
-native = ["griffelib[native]=={{version}}"]
+# The 'faster' extra re-exports griffelib[faster].
+faster = ["griffelib[faster]=={{version}}"]
 # The 'pypi' extra re-exports griffelib[pypi] for backward compatibility.
 pypi = ["griffelib[pypi]=={{version}}"]
 
@@ -105,6 +105,7 @@ ci = [
     "duty>=1.6",
     "griffe-inherited-docstrings>=1.1.2",
     "jsonschema>=4.18",
+    "prune-source>=0.1",
     "pysource-codegen>=0.7",
     "pysource-minimize>=0.10",
     "pytest>=8.2",