From 7823fdaf98bec4216644d05c6535f6d06ab1b01b Mon Sep 17 00:00:00 2001
From: Wassim Kabalan <wastondev@gmail.com>
Date: Fri, 6 Dec 2024 18:57:16 +0100
Subject: [PATCH] Upgrade setup.py to pyproject

---
 pyproject.toml | 26 ++++++++++++++++++++++++++
 setup.py       | 11 -----------
 2 files changed, 26 insertions(+), 11 deletions(-)
 create mode 100644 pyproject.toml
 delete mode 100644 setup.py

diff --git a/pyproject.toml b/pyproject.toml
new file mode 100644
index 0000000..3b97d20
--- /dev/null
+++ b/pyproject.toml
@@ -0,0 +1,26 @@
+[build-system]
+requires = ["setuptools", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "JaxPM"
+version = "0.0.1"
+description = "A dead simple FastPM implementation in JAX"
+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"]
+
+[project.optional-dependencies]
+test = [
+    "jax_cosmo",
+    "jaxdecomp>=0.2.0",
+    "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",
+    "fastpm @ git+https://github.com/ASKabalan/fastpm-python",
+    "diffrax"
+]
diff --git a/setup.py b/setup.py
deleted file mode 100644
index b9b6b29..0000000
--- a/setup.py
+++ /dev/null
@@ -1,11 +0,0 @@
-from setuptools import find_packages, setup
-
-setup(
-    name='JaxPM',
-    version='0.0.1',
-    url='https://github.com/DifferentiableUniverseInitiative/JaxPM',
-    author='JaxPM developers',
-    description='A dead simple FastPM implementation in JAX',
-    packages=find_packages(),
-    install_requires=['jax', 'jax_cosmo', 'jaxdecomp'],
-)