packaging update

hzheng40 · Jun 6, 2024 · 5af34cc · 5af34cc
1 parent 6366720
commit 5af34cc
Show file tree

Hide file tree

Showing 7 changed files with 19 additions and 24 deletions.
diff --git a/README.md b/README.md
@@ -5,14 +5,13 @@
 ## Dependencies
 To install, run:
 ```bash
-cd flax_rbf
 virtualenv venv
 source venv/bin/activate
-pip install -r requirements.txt
+pip install -e .
 ```
 
 ## Examples
-See a classification demo in ```flax_rbf/classification.py```.
+See a classification demo in ```src/flax_rbf/classification.py```.
 
 <p align="center">
   <img src="classification.png">

diff --git a/__init__.py b/__init__.py
diff --git a/pyproject.toml b/pyproject.toml
@@ -0,0 +1,17 @@
+[project]
+name = "flax_rbf"
+authors = [{name = "Hongrui Zheng", email = "[email protected]"}]
+version = "0.1.0"
+requires-python = ">=3.10"
+dependencies = [
+    "jax[cuda12]",
+    "chex",
+    "flax",
+    "optax",
+    "scipy",
+    "matplotlib"
+]
+
+[build-system]
+requires = ["setuptools >= 61.0"]
+build-backend = "setuptools.build_meta"
diff --git a/requirements.txt b/requirements.txt
diff --git a/flax_rbf/__init__.py → src/flax_rbf/__init__.py b/flax_rbf/__init__.py → src/flax_rbf/__init__.py
diff --git a/flax_rbf/classification.py → src/flax_rbf/classification.py b/flax_rbf/classification.py → src/flax_rbf/classification.py
diff --git a/flax_rbf/flax_rbf.py → src/flax_rbf/flax_rbf.py b/flax_rbf/flax_rbf.py → src/flax_rbf/flax_rbf.py