Type annotations for mlx.core module (#512)

This commit is contained in:
Danilo Peixoto
2024-01-21 17:53:12 -03:00
committed by GitHub
parent 7a34e46677
commit ecb174ca9d
2 changed files with 67 additions and 8 deletions

View File

@@ -148,7 +148,7 @@ if __name__ == "__main__":
where="python", exclude=["src", "tests", "tests.*"]
)
package_dir = {"": "python"}
package_data = {"mlx": ["lib/*", "include/*", "share/*"]}
package_data = {"mlx": ["lib/*", "include/*", "share/*"], "mlx.core": ["*.pyi"]}
setup(
name="mlx",