Fix docs latex build and nits (#1297)

* fix docs latex build and nits * fix stub gen and try to clean up building
2025-10-20 01:18:12 +08:00 · 2024-07-29 11:44:06 -07:00
parent 6e06e3a904
commit aa1d6cadad
6 changed files with 36 additions and 18 deletions
--- a/setup.py
+++ b/setup.py
@@ -175,8 +175,14 @@ if __name__ == "__main__":
        package_data=package_data,
        include_package_data=True,
        extras_require={
-            "testing": ["numpy", "torch"],
-            "dev": ["pre-commit"],
+            "dev": [
+                "nanobind@git+https://github.com/wjakob/nanobind.git@2f04eac452a6d9142dedb957701bdb20125561e4",
+                "numpy",
+                "pre-commit",
+                "setuptools>=42",
+                "torch",
+                "typing_extensions",
+            ],
        },
        ext_modules=[CMakeExtension("mlx.core")],
        cmdclass={"build_ext": CMakeBuild, "generate_stubs": GenerateStubs},