python · AlexWaygood · May 2, 2023 · May 1, 2023 · May 1, 2023 · May 1, 2023
diff --git a/requirements-tests.txt b/requirements-tests.txt
@@ -12,6 +12,7 @@ pre-commit-hooks==4.4.0                           # must match .pre-commit-confi
 pycln==2.1.3                                      # must match .pre-commit-config.yaml
 pytype==2023.4.27; platform_system != "Windows" and python_version < "3.11"
 pyyaml==6.0
+stubdefaulter==0.1.0
 termcolor>=2.3
 tomli==2.0.1
 tomlkit==0.11.8

diff --git a/scripts/create_baseline_stubs.py b/scripts/create_baseline_stubs.py
@@ -50,6 +50,11 @@ def run_stubgen(package: str, output: str) -> None:
     subprocess.run(["stubgen", "-o", output, "-p", package, "--export-less"], check=True)
 
 
+def run_stubdefaulter(stub_dir: str) -> None:
+    print(f"Running stubdefaulter: stubdefaulter --packages {stub_dir}")
+    subprocess.run(["stubdefaulter", "--packages", stub_dir])
+
+
 def run_black(stub_dir: str) -> None:
     print(f"Running black: black {stub_dir}")
     subprocess.run(["black", stub_dir])
@@ -153,6 +158,7 @@ def main() -> None:
 
     run_stubgen(package, stub_dir)
 
+    run_stubdefaulter(stub_dir)
     run_isort(stub_dir)
     run_black(stub_dir)