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