From 21ad26b59a6ae01d8ce1e2b0ea53ecec3ed1dc92 Mon Sep 17 00:00:00 2001 From: Jakob Roth <roth@mpa-garching.mpg.de> Date: Mon, 3 Jun 2024 11:54:04 +0200 Subject: [PATCH] setup: add jax and jaxbind for full install --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 5a3bb4c1..ab666ecc 100644 --- a/setup.py +++ b/setup.py @@ -146,7 +146,7 @@ setup( zip_safe=True, dependency_links=[], install_requires=["ducc0>=0.23.0", "numpy", "nifty8>=8.0"], - extras_require={"full": ("astropy", "pytest", "pytest-cov", "mpi4py", "python-casacore", "h5py", "matplotlib")}, + extras_require={"full": ("astropy", "pytest", "pytest-cov", "mpi4py", "python-casacore", "h5py", "matplotlib", "jax", "jaxlib", "jaxbind")}, ext_modules=extensions, entry_points={"console_scripts": [ -- GitLab