Declare a slim dependency set for the minimal wheel build

The minimal build mode trimmed the packaged files but not the declared dependencies. Because `pyproject.toml` `[project.dependencies]` is static, a normal `pip install` of the minimal wheel still pulled `coremltools`, `scikit-learn`, `pandas`, `hydra-core`, `omegaconf` and the test-only deps (`expecttest`, `hypothesis`, `kgb`, `parameterized`), which is most of the real install size. The CI smoke test only passed because it used `pip install --no-deps`.

Make `dependencies` dynamic in `pyproject.toml` and compute `install_requires` in `setup.py` per build mode. The full wheel keeps the existing set; the minimal wheel declares only what `executorch.exir` needs to export a `.pte`: `flatbuffers`, `numpy`, `packaging`, `pyyaml`, `ruamel.yaml`, `sympy`, `tabulate`, `typing-extensions`. `torch` is assumed present, as before.

Update `test_minimal_wheel.sh` to drop `--no-deps` so dependency resolution is exercised, and assert the wheel METADATA declares the slim set and none of the heavy deps.

Co-authored-by: Jacob Szwejbka <20804483+JacobSzwejbka@users.noreply.github.com>
Co-authored-by: Claude <noreply@anthropic.com>

Author: 3 people

.ci/scripts/test_minimal_wheel.sh

@@ -40,6 +40,7 @@ WHEEL_FILE="$(find "${REPO_ROOT}/dist" -maxdepth 1 -name 'executorch-*.whl' | he
 test -n "${WHEEL_FILE}"
 
 python - "${WHEEL_FILE}" <<'PY'
+import re
 import sys
 import zipfile
 
@@ -66,27 +67,77 @@ extensions = [
 ]
 if extensions:
     raise AssertionError(f"{wheel_file} unexpectedly contains extensions: {extensions}")
+
+# The minimal wheel must declare only the slim runtime dependency set: none of
+# the heavy full-wheel deps, and all of the minimal ones.
+metadata_name = next(
+    (name for name in names if name.endswith(".dist-info/METADATA")), None
+)
+if metadata_name is None:
+    raise AssertionError(f"{wheel_file} has no METADATA")
+with zipfile.ZipFile(wheel_file) as wheel:
+    metadata_text = wheel.read(metadata_name).decode("utf-8")
+
+
+def _dist_name(requirement):
+    name = re.split(r"[ ;\[<>=!~(]", requirement.strip(), maxsplit=1)[0]
+    return re.sub(r"[-_.]+", "-", name).lower()
+
+
+declared = {
+    _dist_name(line.split(":", 1)[1])
+    for line in metadata_text.splitlines()
+    if line.startswith("Requires-Dist:")
+}
+heavy = {
+    "coremltools",
+    "scikit-learn",
+    "pandas",
+    "hydra-core",
+    "omegaconf",
+    "expecttest",
+    "hypothesis",
+    "kgb",
+    "parameterized",
+    "pytorch-tokenizers",
+}
+leaked = sorted(declared & heavy)
+if leaked:
+    raise AssertionError(f"{wheel_file} METADATA declares heavy deps: {leaked}")
+expected = {
+    "flatbuffers",
+    "numpy",
+    "packaging",
+    "pyyaml",
+    "ruamel-yaml",
+    "sympy",
+    "tabulate",
+    "typing-extensions",
+}
+missing = sorted(expected - declared)
+if missing:
+    raise AssertionError(f"{wheel_file} METADATA missing minimal deps: {missing}")
 PY
 
 deactivate
 
 "${PYTHON_EXECUTABLE}" -m venv "${TEST_VENV}"
 source "${TEST_VENV}/bin/activate"
 python -m pip install --upgrade pip
+# torch and torchvision are needed to export a model but are intentionally not
+# declared as wheel dependencies (consumers are expected to bring their own).
 python -m pip install \
-  "flatbuffers" \
-  "numpy>=2.0.0" \
-  "packaging" \
-  "pyyaml" \
-  "ruamel.yaml" \
-  "sympy" \
-  "tabulate" \
   "torch" \
   "torchvision" \
-  "typing-extensions>=4.10.0" \
   --index-url https://download.pytorch.org/whl/cpu \
   --extra-index-url https://pypi.org/simple
-python -m pip install --no-deps "${WHEEL_FILE}"
+# Install the minimal wheel WITHOUT --no-deps so pip resolves its declared
+# dependencies: this proves the slim set is correct and resolvable, and that
+# none of the heavy full-wheel deps get pulled in.
+python -m pip install \
+  "${WHEEL_FILE}" \
+  --index-url https://download.pytorch.org/whl/cpu \
+  --extra-index-url https://pypi.org/simple
 
 python - <<'PY'
 from pathlib import Path

pyproject.toml

@@ -16,6 +16,9 @@ name = "executorch"
 dynamic = [
   # setup.py will set the version.
   'version',
+  # setup.py sets dependencies, which vary by build mode (the
+  # EXECUTORCH_BUILD_MINIMAL wheel declares a slimmer runtime set).
+  'dependencies',
 ]
 description = "On-device AI across mobile, embedded and edge for PyTorch"
 readme = "README-wheel.md"
@@ -51,30 +54,11 @@ classifiers = [
 ]
 
 requires-python = ">=3.10,<3.14"
-dependencies=[
-  "expecttest",
-  "flatbuffers",
-  "hypothesis",
-  "kgb",
-  "mpmath==1.3.0",
-  "numpy>=2.0.0; python_version >= '3.10'",
-  "packaging",
-  "pandas>=2.2.2; python_version >= '3.10'",
-  "parameterized",
-  "pytorch-tokenizers",
-  "pyyaml",
-  "ruamel.yaml",
-  "sympy",
-  "tabulate",
-  # See also third-party/TARGETS for buck's typing-extensions version.
-  "typing-extensions>=4.10.0",
-  # Keep this version in sync with: ./backends/apple/coreml/scripts/install_requirements.sh
-  "coremltools==9.0; platform_system == 'Darwin' or platform_system == 'Linux'",
-  # scikit-learn is used to support palettization in the coreml backend
-  "scikit-learn==1.7.1",
-  "hydra-core>=1.3.0",
-  "omegaconf>=2.3.0",
-]
+
+# Runtime dependencies are declared dynamically (see `dynamic` above) and
+# computed in setup.py, so the EXECUTORCH_BUILD_MINIMAL wheel can ship a slimmer
+# set than the full wheel. See `_base_dependencies()` / `_minimal_dependencies()`
+# in setup.py.
 
 [project.optional-dependencies]
 cortex_m = [

setup.py

@@ -165,6 +165,58 @@ def _minimal_packages() -> List[str]:
     )
 
 
+def _base_dependencies() -> List[str]:
+    """Runtime dependencies for the full wheel.
+
+    Declared here rather than in pyproject.toml (where `dependencies` is marked
+    dynamic) so the minimal build can ship a slimmer set. Keep in sync with the
+    project's runtime needs.
+    """
+    return [
+        "expecttest",
+        "flatbuffers",
+        "hypothesis",
+        "kgb",
+        "mpmath==1.3.0",
+        "numpy>=2.0.0; python_version >= '3.10'",
+        "packaging",
+        "pandas>=2.2.2; python_version >= '3.10'",
+        "parameterized",
+        "pytorch-tokenizers",
+        "pyyaml",
+        "ruamel.yaml",
+        "sympy",
+        "tabulate",
+        # See also third-party/TARGETS for buck's typing-extensions version.
+        "typing-extensions>=4.10.0",
+        # Keep this version in sync with: ./backends/apple/coreml/scripts/install_requirements.sh
+        "coremltools==9.0; platform_system == 'Darwin' or platform_system == 'Linux'",
+        # scikit-learn is used to support palettization in the coreml backend.
+        "scikit-learn==1.7.1",
+        "hydra-core>=1.3.0",
+        "omegaconf>=2.3.0",
+    ]
+
+
+def _minimal_dependencies() -> List[str]:
+    """Runtime dependencies for the minimal (AOT export only) wheel.
+
+    Only what executorch.exir needs to lower and serialize a .pte. torch is
+    intentionally omitted (consumers bring their own). Keep in sync with
+    .ci/scripts/test_minimal_wheel.sh.
+    """
+    return [
+        "flatbuffers",
+        "numpy>=2.0.0; python_version >= '3.10'",
+        "packaging",
+        "pyyaml",
+        "ruamel.yaml",
+        "sympy",
+        "tabulate",
+        "typing-extensions>=4.10.0",
+    ]
+
+
 class Version:
     """Static strings that describe the version of the pip package."""
 
@@ -931,6 +983,9 @@ def run(self):  # noqa C901
 setup_kwargs = {}
 if _is_minimal_build():
     setup_kwargs["packages"] = _minimal_packages()
+    setup_kwargs["install_requires"] = _minimal_dependencies()
+else:
+    setup_kwargs["install_requires"] = _base_dependencies()
 
 
 setup(