use uv

.github/workflows/tests.yml

@@ -17,24 +17,24 @@ jobs:
 
     steps:
       - name: Checkout github repo
-        uses: actions/checkout@v4
+        uses: actions/checkout@v6
 
       - name: Set up Python
-        uses: actions/setup-python@v5
+        uses: actions/setup-python@v6
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Install uv and set the Python version
+        uses: astral-sh/setup-uv@v7
         with:
           python-version: ${{ matrix.python-version }}
-          cache: 'pip'
 
       - name: Install dependencies
-        run: |
-          python -m pip install -U pip
-          python -m pip install .
-          python -m pip install .[dev]
-          python -m pip install git+https://github.com/GalSim-developers/JAX-GalSim.git
+        run: uv sync --locked --all-extras --dev
 
       - name: Run Ruff
-        run: ruff check --output-format=github .
+        run: uv run ruff check --output-format=github .
 
-      - name: Run fast tests
+      - name: Run tests
         run: |
-          pytest --durations=0
+          uv run pytest --durations=0

.python-version

@@ -0,0 +1 @@
+/pscratch/sd/i/imendoza/miniconda3/envs/python312/bin/python

README.md

@@ -22,9 +22,6 @@ pip install -e ".[dev]"
 # Install JAX-Galsim
 pip install git+https://github.com/GalSim-developers/JAX-GalSim.git --no-deps --no-build-isolation
 
-# Install ornax
-pip install git+https://github.com/beckermr/ornax.git --no-deps --no-build-isolation
-
 # Might be necessary
 conda install -c nvidia cuda-nvcc
 ```

pyproject.toml

@@ -4,6 +4,9 @@ build-backend = "setuptools.build_meta"
 
 
 [project]
+requires-python = ">=3.10,<3.13"
+
+
 name = "BPD"
 authors = [{ name = "Ismael Mendoza" }]
 description = "Bayesian Pixel Domain method for shear inference."
@@ -19,16 +22,21 @@ dependencies = [
     "quadax>=0.2.5",
     "tensorflow-probability >=0.21.0",
     "arviz>=0.21.0",
-    "jax-tqdm",
+    "jax[cuda12]<0.7.0",
+    "jax-galsim",
 ]
-requires-python = ">=3.10,<3.13"
 
+[tool.uv.sources]
+jax-galsim = { git = "https://github.com/GalSim-developers/JAX-GalSim.git", branch = "main" }
+
+[tool.uv.pip]
+no-build-isolation-package = ["jax-galsim"]
 
 [project.optional-dependencies]
 dev = [
     "pytest",
     "typer",
-    "ruff<0.14.0",
+    "ruff",
     "ChainConsumer>=1.2.0", #v1.2.2 still requires python < 3.13
     "ipython",
     "ipykernel",
@@ -104,6 +112,7 @@ ignore = [
     "PLC0206", # too picky, not always useful
     "RUF027",  # does not account for shell scripts correctly
     "RUF052",  # I find underscore variables useful
+    "RUF067",  # I like adding things to the __init__
 ]
 preview = true
 exclude = ["*.ipynb", "scripts/one_galaxy_shear.py", "scripts/benchmarks/*.py"]