From b04962e0eed73aa188ccc4a52ba06aa18a9b88e8 Mon Sep 17 00:00:00 2001
From: beckermr <becker.mr@gmail.com>
Date: Sat, 7 Feb 2026 21:09:58 -0600
Subject: [PATCH 01/18] test: use pytest-split for faster tests

---
 .github/workflows/python_package.yaml | 56 ++++++++++++++++++++++++++-
 .github/workflows/release.yml         |  2 +-
 2 files changed, 55 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/python_package.yaml b/.github/workflows/python_package.yaml
index af985d14..4964fb6a 100644
--- a/.github/workflows/python_package.yaml
+++ b/.github/workflows/python_package.yaml
@@ -21,6 +21,9 @@ jobs:
       fail-fast: false
       matrix:
         python-version: ["3.12"]
+        group: [1, 2, 3, 4]
+    env:
+      NUM_SPLITS: 4
 
     steps:
       - uses: actions/checkout@v6
@@ -37,18 +40,67 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          python -m pip install pytest pytest-codspeed pytest-randomly
+          python -m pip install pytest pytest-codspeed pytest-randomly pytest-split
           python -m pip install .
 
+      - name: Restore test durations
+        uses: actions/cache/restore@cdf6c1fa76f9f475f3d7449005a359c84ca0f306  # v5.0.3
+        with:
+          path: .test_durations
+          key: test-durations-${{ github.ref }}-${{ github.sha }}
+          restore-keys: |
+            test-durations-${{ github.ref }}-
+            test-durations-
+
+      - name: Post-process test durations
+        run: |
+          if [[ -f .test_durations ]]; then
+            cp .test_durations .test_durations.${{ matrix.group }}
+          fi
+          ls -lah .test_durations*
+          echo " "
+          cat .test_durations*
+
       - name: Test with pytest
         run: |
-          pytest -vv --durations=100 --randomly-seed=42
+          pytest \
+            -vv \
+            --durations=100 \
+            --randomly-seed=42 \
+            --splits ${NUM_SPLITS} --group ${{ matrix.group }} \
+            --store-durations \
+            --durations-path=.test_durations.${{ matrix.group }} \
+            --splitting-algorithm least_duration \
+            --clean-durations
+
+      - name: Upload test durations
+        uses: actions/upload-artifact@v6
+        with:
+          name: test-durations-${{ matrix.group }}
+          path: .test_durations.${{ matrix.group }}
+          include-hidden-files: true
 
   build-status:
     needs: build
     runs-on: ubuntu-latest
 
     steps:
+      - name: Download test duration artifacts
+        uses: actions/download-artifact@v7
+        with:
+          pattern: test-durations-*
+
+      - name: Combine test durations
+        run: |
+          # see https://stackoverflow.com/a/71416016/1745538
+          jq 'reduce inputs as $i (.; . + $i)' test-durations-*/.test_durations.* > .test_durations
+
+      - name: Save test durations
+        uses: actions/cache/save@v5
+        with:
+          path: .test_durations
+          key: test-durations-${{ github.ref }}-${{ github.sha }}
+
       - name: check status
         run: |
           echo "Builds all passed!"
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index 82f8dbd9..e9e9275e 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -28,9 +28,9 @@ jobs:
 
       - name: Build release distributions
         run: |
-          # NOTE: put your own distribution build steps here.
           python -m pip install build
           python -m build
+          python -m build --sdist .
 
       - name: Upload distributions
         uses: actions/upload-artifact@v4

From a2497b60ff996e3f4d66373aef476539c1f672c4 Mon Sep 17 00:00:00 2001
From: beckermr <becker.mr@gmail.com>
Date: Sat, 7 Feb 2026 21:12:27 -0600
Subject: [PATCH 02/18] fix: make sure we handle cache miss properly

---
 .github/workflows/python_package.yaml | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/python_package.yaml b/.github/workflows/python_package.yaml
index 4964fb6a..73a7e689 100644
--- a/.github/workflows/python_package.yaml
+++ b/.github/workflows/python_package.yaml
@@ -56,10 +56,10 @@ jobs:
         run: |
           if [[ -f .test_durations ]]; then
             cp .test_durations .test_durations.${{ matrix.group }}
+            ls -lah .test_durations*
+            echo " "
+            cat .test_durations*
           fi
-          ls -lah .test_durations*
-          echo " "
-          cat .test_durations*
 
       - name: Test with pytest
         run: |

From 178b5990a629a3103296d33a277ba82b8ed73edb Mon Sep 17 00:00:00 2001
From: beckermr <becker.mr@gmail.com>
Date: Sat, 7 Feb 2026 21:29:30 -0600
Subject: [PATCH 03/18] debug: try to make the dir

---
 .github/workflows/python_package.yaml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/.github/workflows/python_package.yaml b/.github/workflows/python_package.yaml
index 73a7e689..0fa8b546 100644
--- a/.github/workflows/python_package.yaml
+++ b/.github/workflows/python_package.yaml
@@ -63,6 +63,7 @@ jobs:
 
       - name: Test with pytest
         run: |
+          mkdir output
           pytest \
             -vv \
             --durations=100 \

From aa6fee33ffe90d47b4e3ea89676dab98f2ae99cf Mon Sep 17 00:00:00 2001
From: beckermr <becker.mr@gmail.com>
Date: Sat, 7 Feb 2026 21:41:57 -0600
Subject: [PATCH 04/18] fix: esnure we run tests in tests dir

---
 .github/workflows/python_package.yaml | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/python_package.yaml b/.github/workflows/python_package.yaml
index 0fa8b546..a80363d5 100644
--- a/.github/workflows/python_package.yaml
+++ b/.github/workflows/python_package.yaml
@@ -63,16 +63,17 @@ jobs:
 
       - name: Test with pytest
         run: |
-          mkdir output
+          pushd tests
           pytest \
             -vv \
             --durations=100 \
             --randomly-seed=42 \
             --splits ${NUM_SPLITS} --group ${{ matrix.group }} \
             --store-durations \
-            --durations-path=.test_durations.${{ matrix.group }} \
+            --durations-path=../.test_durations.${{ matrix.group }} \
             --splitting-algorithm least_duration \
             --clean-durations
+          popd
 
       - name: Upload test durations
         uses: actions/upload-artifact@v6

From 4812fb55b0ac83230f8bba66f882bacface7d80e Mon Sep 17 00:00:00 2001
From: beckermr <becker.mr@gmail.com>
Date: Sat, 7 Feb 2026 21:46:13 -0600
Subject: [PATCH 05/18] fix: always make output dir

---
 .github/workflows/python_package.yaml | 4 +---
 tests/conftest.py                     | 3 +++
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/python_package.yaml b/.github/workflows/python_package.yaml
index a80363d5..73a7e689 100644
--- a/.github/workflows/python_package.yaml
+++ b/.github/workflows/python_package.yaml
@@ -63,17 +63,15 @@ jobs:
 
       - name: Test with pytest
         run: |
-          pushd tests
           pytest \
             -vv \
             --durations=100 \
             --randomly-seed=42 \
             --splits ${NUM_SPLITS} --group ${{ matrix.group }} \
             --store-durations \
-            --durations-path=../.test_durations.${{ matrix.group }} \
+            --durations-path=.test_durations.${{ matrix.group }} \
             --splitting-algorithm least_duration \
             --clean-durations
-          popd
 
       - name: Upload test durations
         uses: actions/upload-artifact@v6
diff --git a/tests/conftest.py b/tests/conftest.py
index 49507d78..36f6d71f 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -18,6 +18,9 @@
 # Identify the path to this current file
 test_directory = os.path.dirname(os.path.abspath(__file__))
 
+# make the output directory for the test output
+os.makedirs(os.path.join("..", "output"), exist_ok=True)
+
 # Loading which tests to run
 with open(os.path.join(test_directory, "galsim_tests_config.yaml"), "r") as f:
     test_config = yaml.safe_load(f)

From f0fce762c5d18c74c214724cf21f6722f55660d7 Mon Sep 17 00:00:00 2001
From: beckermr <becker.mr@gmail.com>
Date: Sat, 7 Feb 2026 21:48:38 -0600
Subject: [PATCH 06/18] fix: use simulation for benchmarks

---
 .github/workflows/benchmarks.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/benchmarks.yaml b/.github/workflows/benchmarks.yaml
index ee17aac1..b42b0267 100644
--- a/.github/workflows/benchmarks.yaml
+++ b/.github/workflows/benchmarks.yaml
@@ -42,4 +42,4 @@ jobs:
         with:
           token: ${{ secrets.CODSPEED_TOKEN }}
           run: pytest -vvs --codspeed
-          mode: instrumentation
+          mode: simulation

From 32851702952f85d63cdc7906bb71e966f10abae0 Mon Sep 17 00:00:00 2001
From: beckermr <becker.mr@gmail.com>
Date: Sat, 7 Feb 2026 21:49:34 -0600
Subject: [PATCH 07/18] test: cleanup benchmark ci

---
 .github/workflows/benchmarks.yaml | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/benchmarks.yaml b/.github/workflows/benchmarks.yaml
index b42b0267..5630ceef 100644
--- a/.github/workflows/benchmarks.yaml
+++ b/.github/workflows/benchmarks.yaml
@@ -24,6 +24,10 @@ jobs:
 
     steps:
       - uses: actions/checkout@v6
+        with:
+          fetch-depth: 0
+          submodules: recursive
+          fetch-tags: true
 
       - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@v6
@@ -35,11 +39,10 @@ jobs:
           python -m pip install --upgrade pip
           python -m pip install pytest pytest-codspeed
           python -m pip install .
-          git submodule update --init --recursive
 
       - name: Run benchmarks
         uses: CodSpeedHQ/action@v4
         with:
           token: ${{ secrets.CODSPEED_TOKEN }}
           run: pytest -vvs --codspeed
-          mode: simulation
+          mode: walltime

From f67ce3a5b6551a03e765774994d79680ff11c76d Mon Sep 17 00:00:00 2001
From: beckermr <becker.mr@gmail.com>
Date: Sat, 7 Feb 2026 22:12:07 -0600
Subject: [PATCH 08/18] test: try the bare-metal runner

---
 .github/workflows/benchmarks.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/benchmarks.yaml b/.github/workflows/benchmarks.yaml
index 5630ceef..c9206d29 100644
--- a/.github/workflows/benchmarks.yaml
+++ b/.github/workflows/benchmarks.yaml
@@ -16,7 +16,7 @@ on:
 
 jobs:
   build:
-    runs-on: ubuntu-latest
+    runs-on: codspeed-macro
     strategy:
       fail-fast: false
       matrix:

From cba7543bc3ec3440e5b75a28eea63ea9412e51d6 Mon Sep 17 00:00:00 2001
From: beckermr <becker.mr@gmail.com>
Date: Sat, 7 Feb 2026 22:16:43 -0600
Subject: [PATCH 09/18] test: fix benchmarks

---
 .github/workflows/benchmarks.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/benchmarks.yaml b/.github/workflows/benchmarks.yaml
index c9206d29..6682eb58 100644
--- a/.github/workflows/benchmarks.yaml
+++ b/.github/workflows/benchmarks.yaml
@@ -16,7 +16,7 @@ on:
 
 jobs:
   build:
-    runs-on: codspeed-macro
+    runs-on: ubuntu-latest
     strategy:
       fail-fast: false
       matrix:
@@ -45,4 +45,4 @@ jobs:
         with:
           token: ${{ secrets.CODSPEED_TOKEN }}
           run: pytest -vvs --codspeed
-          mode: walltime
+          mode: simulation

From edd4fc2c2b0c869b6f14445848426439e6ac2bce Mon Sep 17 00:00:00 2001
From: "Matthew R. Becker" <beckermr@users.noreply.github.com>
Date: Sat, 7 Feb 2026 23:09:51 -0600
Subject: [PATCH 10/18] Update conftest.py

---
 tests/conftest.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/conftest.py b/tests/conftest.py
index 36f6d71f..9bccd20e 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -19,7 +19,7 @@
 test_directory = os.path.dirname(os.path.abspath(__file__))
 
 # make the output directory for the test output
-os.makedirs(os.path.join("..", "output"), exist_ok=True)
+os.makedirs("output", exist_ok=True)
 
 # Loading which tests to run
 with open(os.path.join(test_directory, "galsim_tests_config.yaml"), "r") as f:

From 50a58fd9ef2fcfd9142670a9cdd89a73cd0212ae Mon Sep 17 00:00:00 2001
From: beckermr <becker.mr@gmail.com>
Date: Sun, 8 Feb 2026 05:10:54 -0600
Subject: [PATCH 11/18] perf: use splits for benchmarks too

---
 .github/workflows/benchmarks.yaml | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/benchmarks.yaml b/.github/workflows/benchmarks.yaml
index 6682eb58..7aed13bb 100644
--- a/.github/workflows/benchmarks.yaml
+++ b/.github/workflows/benchmarks.yaml
@@ -20,7 +20,10 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: ["3.12",]
+        python-version: ["3.12"]
+        group: [1, 2, 3, 4]
+    env:
+      NUM_SPLITS: 4
 
     steps:
       - uses: actions/checkout@v6
@@ -37,12 +40,12 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          python -m pip install pytest pytest-codspeed
+          python -m pip install pytest pytest-codspeed pytest-split
           python -m pip install .
 
       - name: Run benchmarks
         uses: CodSpeedHQ/action@v4
         with:
           token: ${{ secrets.CODSPEED_TOKEN }}
-          run: pytest -vvs --codspeed
+          run: pytest -vvs --codspeed --splits ${{ env.NUM_SPLITS }} --group ${{ matrix.group }}
           mode: simulation

From b1e33019333e0f8fb64a045d8483c4a9628a3123 Mon Sep 17 00:00:00 2001
From: beckermr <becker.mr@gmail.com>
Date: Sun, 8 Feb 2026 05:32:57 -0600
Subject: [PATCH 12/18] fix: these are not benchmarks

---
 tests/jax/test_spergel_comp_galsim.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/jax/test_spergel_comp_galsim.py b/tests/jax/test_spergel_comp_galsim.py
index d0b3dddd..1411fdab 100644
--- a/tests/jax/test_spergel_comp_galsim.py
+++ b/tests/jax/test_spergel_comp_galsim.py
@@ -113,7 +113,7 @@ def _run():
 
 
 @pytest.mark.parametrize("kind", ["compile", "run"])
-def test_spergel_comp_galsim_perf_conv(benchmark, kind):
+def test_spergel_comp_galsim_perf_conv(kind):
     dt = _run_time_test(kind, lambda: _run_spergel_bench_conv_jit().block_until_ready())
     print(f"\njax-galsim time: {dt:0.4g} ms")
 
@@ -125,7 +125,7 @@ def test_spergel_comp_galsim_perf_conv(benchmark, kind):
 
 
 @pytest.mark.parametrize("kind", ["compile", "run"])
-def test_spergel_comp_galsim_perf_kvalue(benchmark, kind):
+def test_spergel_comp_galsim_perf_kvalue(kind):
     dt = _run_time_test(
         kind, lambda: _run_spergel_bench_kvalue_jit().block_until_ready()
     )
@@ -139,7 +139,7 @@ def test_spergel_comp_galsim_perf_kvalue(benchmark, kind):
 
 
 @pytest.mark.parametrize("kind", ["compile", "run"])
-def test_spergel_comp_galsim_perf_xvalue(benchmark, kind):
+def test_spergel_comp_galsim_perf_xvalue(kind):
     dt = _run_time_test(
         kind, lambda: _run_spergel_bench_xvalue_jit().block_until_ready()
     )

From 9b55688ef492dbf3b6f667029ac314aaffe90b55 Mon Sep 17 00:00:00 2001
From: beckermr <becker.mr@gmail.com>
Date: Sun, 8 Feb 2026 05:33:35 -0600
Subject: [PATCH 13/18] chore: fix pre-commit

---
 .pre-commit-config.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index fdcfd90c..ac38ca37 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -13,7 +13,7 @@ repos:
   - repo: https://github.com/astral-sh/ruff-pre-commit
     rev: v0.14.14
     hooks:
-      - id: ruff
+      - id: ruff-check
         args: [ --fix ]
         exclude: tests/GalSim/|tests/Coord/|tests/jax/galsim/|dev/notebooks/
       - id: ruff-format

From c08cf0d8e17da3d98db055e17f4f938db67a570c Mon Sep 17 00:00:00 2001
From: beckermr <becker.mr@gmail.com>
Date: Sun, 8 Feb 2026 05:52:59 -0600
Subject: [PATCH 14/18] perf: try without splits

---
 .github/workflows/benchmarks.yaml |  9 +++++----
 tests/jax/test_benchmarks.py      | 24 ++++++++++++------------
 tests/jax/test_image_wrapping.py  |  2 +-
 tests/jax/test_interpolant_jax.py |  3 ++-
 4 files changed, 20 insertions(+), 18 deletions(-)

diff --git a/.github/workflows/benchmarks.yaml b/.github/workflows/benchmarks.yaml
index 7aed13bb..203b39b7 100644
--- a/.github/workflows/benchmarks.yaml
+++ b/.github/workflows/benchmarks.yaml
@@ -21,9 +21,9 @@ jobs:
       fail-fast: false
       matrix:
         python-version: ["3.12"]
-        group: [1, 2, 3, 4]
-    env:
-      NUM_SPLITS: 4
+    #     group: [1, 2, 3, 4]
+    # env:
+    #   NUM_SPLITS: 4
 
     steps:
       - uses: actions/checkout@v6
@@ -40,12 +40,13 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          python -m pip install pytest pytest-codspeed pytest-split
+          python -m pip install pytest pytest-codspeed
           python -m pip install .
 
       - name: Run benchmarks
         uses: CodSpeedHQ/action@v4
         with:
           token: ${{ secrets.CODSPEED_TOKEN }}
+          # run: pytest -vvs --codspeed --splits ${{ env.NUM_SPLITS }} --group ${{ matrix.group }}
           run: pytest -vvs --codspeed --splits ${{ env.NUM_SPLITS }} --group ${{ matrix.group }}
           mode: simulation
diff --git a/tests/jax/test_benchmarks.py b/tests/jax/test_benchmarks.py
index 4d35e8a7..b48091ec 100644
--- a/tests/jax/test_benchmarks.py
+++ b/tests/jax/test_benchmarks.py
@@ -33,7 +33,7 @@ def _run():
     return tr.dt
 
 
-@pytest.mark.parametrize("kind", ["compile", "run"])
+@pytest.mark.parametrize("kind", ["run"])
 @pytest.mark.parametrize(
     "conserve_dc", [True, False], ids=["conserve_dc", "no_conserve_dc"]
 )
@@ -53,7 +53,7 @@ def test_benchmarks_lanczos_interp(benchmark, kind, conserve_dc, method):
     print(f"time: {dt:0.4g} ms", end=" ")
 
 
-@pytest.mark.parametrize("kind", ["compile", "run"])
+@pytest.mark.parametrize("kind", ["run"])
 def test_benchmarks_interpolated_image(benchmark, kind):
     gal = jgs.Gaussian(fwhm=1.2)
     im_gal = gal.drawImage(nx=32, ny=32, scale=0.2)
@@ -174,7 +174,7 @@ def _run_spergel_bench_conv(gsmod):
 _run_spergel_bench_conv_jit = jax.jit(partial(_run_spergel_bench_conv, jgs))
 
 
-@pytest.mark.parametrize("kind", ["compile", "run"])
+@pytest.mark.parametrize("kind", ["run"])
 def test_benchmark_spergel_conv(benchmark, kind):
     dt = _run_benchmarks(
         benchmark, kind, lambda: _run_spergel_bench_conv_jit().block_until_ready()
@@ -190,7 +190,7 @@ def _run_spergel_bench_xvalue(gsmod):
 _run_spergel_bench_xvalue_jit = jax.jit(partial(_run_spergel_bench_xvalue, jgs))
 
 
-@pytest.mark.parametrize("kind", ["compile", "run"])
+@pytest.mark.parametrize("kind", ["run"])
 def test_benchmark_spergel_xvalue(benchmark, kind):
     dt = _run_benchmarks(
         benchmark, kind, lambda: _run_spergel_bench_xvalue_jit().block_until_ready()
@@ -206,7 +206,7 @@ def _run_spergel_bench_kvalue(gsmod):
 _run_spergel_bench_kvalue_jit = jax.jit(partial(_run_spergel_bench_kvalue, jgs))
 
 
-@pytest.mark.parametrize("kind", ["compile", "run"])
+@pytest.mark.parametrize("kind", ["run"])
 def test_benchmark_spergel_kvalue(benchmark, kind):
     dt = _run_benchmarks(
         benchmark, kind, lambda: _run_spergel_bench_kvalue_jit().block_until_ready()
@@ -219,7 +219,7 @@ def _run_spergel_bench_init():
     return jgs.Spergel(nu=-0.6, half_light_radius=3.4).scale_radius
 
 
-@pytest.mark.parametrize("kind", ["compile", "run"])
+@pytest.mark.parametrize("kind", ["run"])
 def test_benchmark_spergel_init(benchmark, kind):
     dt = _run_benchmarks(
         benchmark, kind, lambda: _run_spergel_bench_init().block_until_ready()
@@ -232,7 +232,7 @@ def _run_gaussian_bench_init():
     return jgs.Gaussian(half_light_radius=3.4).sigma
 
 
-@pytest.mark.parametrize("kind", ["compile", "run"])
+@pytest.mark.parametrize("kind", ["run"])
 def test_benchmark_gaussian_init(benchmark, kind):
     dt = _run_benchmarks(
         benchmark, kind, lambda: _run_gaussian_bench_init().block_until_ready()
@@ -247,7 +247,7 @@ def _run_benchmark_interpimage_flux_frac(img):
     return jgs.interpolatedimage._flux_frac(img.array, x, y, cenx, ceny)
 
 
-@pytest.mark.parametrize("kind", ["compile", "run"])
+@pytest.mark.parametrize("kind", ["run"])
 def test_benchmark_interpimage_flux_frac(benchmark, kind):
     obj = jgs.Gaussian(half_light_radius=0.9).shear(g1=0.1, g2=0.2)
     img = obj.drawImage(nx=55, ny=55, scale=0.2, method="no_pixel")
@@ -265,7 +265,7 @@ def _run_benchmark_rng_discard(rng):
     return rng._state.key
 
 
-@pytest.mark.parametrize("kind", ["compile", "run"])
+@pytest.mark.parametrize("kind", ["run"])
 def test_benchmark_rng_discard(benchmark, kind):
     rng = jgs.BaseDeviate(seed=42)
     dt = _run_benchmarks(
@@ -278,7 +278,7 @@ def _run_benchmark_invert_ab_noraise(u, v, ab):
     return jgs.fitswcs._invert_ab_noraise(u, v, ab)[0]
 
 
-@pytest.mark.parametrize("kind", ["compile", "run"])
+@pytest.mark.parametrize("kind", ["run"])
 def test_benchmark_invert_ab_noraise(benchmark, kind):
     u = jnp.arange(1000).astype(jnp.float64)
     v = jnp.arange(1000).astype(jnp.float64)
@@ -295,7 +295,7 @@ def _run_benchmark_moffat_init():
     return jgs.Moffat(beta=2.5, half_light_radius=0.6, trunc=1.2).scale_radius
 
 
-@pytest.mark.parametrize("kind", ["compile", "run"])
+@pytest.mark.parametrize("kind", ["run"])
 def test_benchmark_moffat_init(benchmark, kind):
     dt = _run_benchmarks(
         benchmark, kind, lambda: _run_benchmark_moffat_init().block_until_ready()
@@ -307,7 +307,7 @@ def _run_benchmark_spergel_calcfluxrad():
     return jgs.spergel.calculateFluxRadius(1e-10, 2.0)
 
 
-@pytest.mark.parametrize("kind", ["compile", "run"])
+@pytest.mark.parametrize("kind", ["run"])
 def test_benchmark_spergel_calcfluxrad(benchmark, kind):
     dt = _run_benchmarks(
         benchmark,
diff --git a/tests/jax/test_image_wrapping.py b/tests/jax/test_image_wrapping.py
index 4bb6a4da..63136a4b 100644
--- a/tests/jax/test_image_wrapping.py
+++ b/tests/jax/test_image_wrapping.py
@@ -1,7 +1,7 @@
 import jax
 import numpy as np
 import pytest
-from galsim_test_helpers import timer
+from galsim.utilities import timer
 
 import jax_galsim as galsim
 from jax_galsim.core.wrap_image import (
diff --git a/tests/jax/test_interpolant_jax.py b/tests/jax/test_interpolant_jax.py
index 73c2bb81..2be425c1 100644
--- a/tests/jax/test_interpolant_jax.py
+++ b/tests/jax/test_interpolant_jax.py
@@ -11,7 +11,8 @@
 import jax
 import numpy as np
 import pytest
-from galsim_test_helpers import assert_raises, timer
+from galsim.utilities import timer
+from numpy.testing import assert_raises
 from scipy.special import sici
 
 import jax_galsim as galsim

From 2e4a11fce2cecf6732f0c28111d39647edc04311 Mon Sep 17 00:00:00 2001
From: beckermr <becker.mr@gmail.com>
Date: Sun, 8 Feb 2026 05:55:22 -0600
Subject: [PATCH 15/18] fix: bug in call

---
 .github/workflows/benchmarks.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/benchmarks.yaml b/.github/workflows/benchmarks.yaml
index 203b39b7..e7bc2c33 100644
--- a/.github/workflows/benchmarks.yaml
+++ b/.github/workflows/benchmarks.yaml
@@ -48,5 +48,5 @@ jobs:
         with:
           token: ${{ secrets.CODSPEED_TOKEN }}
           # run: pytest -vvs --codspeed --splits ${{ env.NUM_SPLITS }} --group ${{ matrix.group }}
-          run: pytest -vvs --codspeed --splits ${{ env.NUM_SPLITS }} --group ${{ matrix.group }}
+          run: pytest -vvs --codspeed
           mode: simulation

From 443f68f7da52299a85a523a7070e7b295be6283e Mon Sep 17 00:00:00 2001
From: beckermr <becker.mr@gmail.com>
Date: Sun, 8 Feb 2026 06:19:18 -0600
Subject: [PATCH 16/18] style: remove dead code

---
 .github/workflows/benchmarks.yaml | 4 ----
 1 file changed, 4 deletions(-)

diff --git a/.github/workflows/benchmarks.yaml b/.github/workflows/benchmarks.yaml
index e7bc2c33..b47e505b 100644
--- a/.github/workflows/benchmarks.yaml
+++ b/.github/workflows/benchmarks.yaml
@@ -21,9 +21,6 @@ jobs:
       fail-fast: false
       matrix:
         python-version: ["3.12"]
-    #     group: [1, 2, 3, 4]
-    # env:
-    #   NUM_SPLITS: 4
 
     steps:
       - uses: actions/checkout@v6
@@ -47,6 +44,5 @@ jobs:
         uses: CodSpeedHQ/action@v4
         with:
           token: ${{ secrets.CODSPEED_TOKEN }}
-          # run: pytest -vvs --codspeed --splits ${{ env.NUM_SPLITS }} --group ${{ matrix.group }}
           run: pytest -vvs --codspeed
           mode: simulation

From 7503f8fed785534411300877188c70050a58a789 Mon Sep 17 00:00:00 2001
From: beckermr <becker.mr@gmail.com>
Date: Sun, 8 Feb 2026 06:23:22 -0600
Subject: [PATCH 17/18] fix: sdist is already going

---
 .github/workflows/release.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index e9e9275e..82f8dbd9 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -28,9 +28,9 @@ jobs:
 
       - name: Build release distributions
         run: |
+          # NOTE: put your own distribution build steps here.
           python -m pip install build
           python -m build
-          python -m build --sdist .
 
       - name: Upload distributions
         uses: actions/upload-artifact@v4

From eb21f94ce0c233ea7f79171ce785b7017d6d2f2d Mon Sep 17 00:00:00 2001
From: beckermr <becker.mr@gmail.com>
Date: Sun, 8 Feb 2026 06:28:25 -0600
Subject: [PATCH 18/18] chore: this is not used

---
 tests/output/README.md | 1 -
 1 file changed, 1 deletion(-)
 delete mode 100644 tests/output/README.md

diff --git a/tests/output/README.md b/tests/output/README.md
deleted file mode 100644
index 7e5b8f4b..00000000
--- a/tests/output/README.md
+++ /dev/null
@@ -1 +0,0 @@
-Folder for storing test outputs
\ No newline at end of file