add extra_require with libs for running tests (#36)

This commit is contained in:
Diogo
2023-12-06 15:21:48 -05:00
committed by GitHub
parent 2440fe0124
commit d15dead35e
3 changed files with 5 additions and 1 deletions

View File

@@ -145,6 +145,9 @@ if __name__ == "__main__":
package_dir=package_dir,
package_data=package_data,
include_package_data=True,
extras_require={
"testing": ["numpy", "torch"]
},
ext_modules=[CMakeExtension("mlx.core")],
cmdclass={"build_ext": CMakeBuild},
zip_safe=False,