fix submodule stubs (#1492)

2025-10-20 01:18:12 +08:00 · 2024-10-15 16:23:37 -07:00
parent 3f86399922
commit f9f8c167d4
3 changed files with 24 additions and 2 deletions
--- a/setup.py
+++ b/setup.py
@@ -141,6 +141,8 @@ class GenerateStubs(Command):
            "nanobind.stubgen",
            "-m",
            "mlx.core",
+            "-p",
+            "python/mlx/_stub_patterns.txt",
        ]
        subprocess.run(stub_cmd + ["-r", "-O", out_path])
        # Run again without recursive to specify output file name