zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-24 09:21:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add extra_require with libs for running tests (#36)

Browse Source
This commit is contained in:
Diogo 2023-12-06 15:21:48 -05:00 committed by GitHub
parent 2440fe0124
commit d15dead35e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -8,6 +8,7 @@ __pycache__/
# Metal libraries # Metal libraries
*.metallib *.metallib
venv/
# Distribution / packaging # Distribution / packaging
python/mlx/share python/mlx/share

2
docs/src/install.rst
View File

@ -60,7 +60,7 @@ For developing use an editable install:
To make sure the install is working run the tests with: To make sure the install is working run the tests with:
.. code-block:: shell .. code-block:: shell
pip install ".[testing]"
python -m unittest discover python/tests python -m unittest discover python/tests
C++ API C++ API

3
setup.py
View File

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