From ac4ef9e1fa156f175fefa939446cf56a97151c36 Mon Sep 17 00:00:00 2001
From: Wassim Kabalan <wastondev@gmail.com>
Date: Sun, 8 Dec 2024 22:46:46 +0100
Subject: [PATCH] update test dependencies

---
 pyproject.toml | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 3b97d20..1738bd1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -10,17 +10,20 @@ authors = [{ name = "JaxPM developers" }]
 readme = "README.md"
 requires-python = ">=3.9"
 license = { file = "LICENSE" }
-homepage = "https://github.com/DifferentiableUniverseInitiative/JaxPM"
-dependencies = ["jax_cosmo", "jax>=0.4.30", "jaxdecomp>=0.2.0"]
+urls = { "Homepage" = "https://github.com/DifferentiableUniverseInitiative/JaxPM" }
+dependencies = ["jax_cosmo", "jax>=0.4.30", "jaxdecomp>=0.2.1"]
 
 [project.optional-dependencies]
 test = [
+    "jax>=0.4.30",
     "jax_cosmo",
-    "jaxdecomp>=0.2.0",
+    "jaxdecomp>=0.2.1",
     "pytest>=8.0.0",
-    "jax[cpu]>=0.4.30",
     "pfft-python @ git+https://github.com/MP-Gadget/pfft-python",
-    "pmesh @ git+https://github.com/ASKabalan/pmesh",
+    "pmesh @ git+https://github.com/MP-Gadget/pmesh",
     "fastpm @ git+https://github.com/ASKabalan/fastpm-python",
     "diffrax"
 ]
+
+[tool.setuptools]
+packages = ["jaxpm"]