Merge pull request #419 from PyAutoLabs/feature/python-312-floor

feat: require Python 3.12

Author: Jammy2211

AGENTS.md

@@ -51,7 +51,7 @@ NUMBA_CACHE_DIR=/tmp/numba_cache MPLCONFIGDIR=/tmp/matplotlib python -m pytest t
 
 PRs must pass `pytest --cov` on the CI matrix (Python 3.12 **and** 3.13). There
 is no black/ruff/flake8 gate — formatting is advisory. (`requires-python` in
-`pyproject.toml` is `>=3.9`.)
+`pyproject.toml` is `>=3.12`.)
 
 ## Configuration & defaults
 

autoarray/operators/transformer.py

@@ -1,6 +1,5 @@
 import copy
 import numpy as np
-import sys
 import warnings
 from typing import Optional, Tuple
 
@@ -42,32 +41,6 @@ def pynufft_exception():
 
 
 def nufftax_exception():
-    # On Python 3.11 `pip install nufftax` is actively bad advice. The releases
-    # that work with our JAX stack (0.4.x) require Python >= 3.12, so pip resolves
-    # 0.6.x instead — and 0.6.0/0.6.1 both break the interferometer inversion
-    # (0.6.0 uses `jax.interpreters.batching.not_mapped`, removed in JAX 0.10;
-    # 0.6.1 cannot handle the rank-4 input `transform_mapping_matrix` produces
-    # under vmap). There is currently no nufftax release that both installs on
-    # 3.11 and works, so send 3.11 users to 3.12+ or to the pynufft backend.
-    if sys.version_info < (3, 12):
-        raise ModuleNotFoundError(
-            "\n--------------------\n"
-            "You are attempting to perform interferometer analysis with the default "
-            f"JAX-native `TransformerNUFFT`, on Python {sys.version_info.major}.{sys.version_info.minor}.\n\n"
-            "The optional library nufftax (https://github.com/GragasLab/nufftax) is not installed, "
-            "and on this Python version it CANNOT be usefully installed:\n\n"
-            "  - nufftax releases that work with PyAutoArray require Python >= 3.12.\n"
-            "  - The releases that do install here (0.6.x) are incompatible with the "
-            "JAX version PyAutoArray requires, and fail partway through a fit.\n\n"
-            "Do NOT `pip install nufftax` on this Python version. Instead either:\n\n"
-            "  1. Upgrade to Python 3.12 or newer (recommended), then "
-            "`pip install 'autoarray[optional]'`; or\n"
-            "  2. Use the legacy pynufft backend, by passing "
-            "`transformer_class=TransformerNUFFTPyNUFFT` and running "
-            "`pip install pynufft==2022.2.2`.\n\n"
-            "----------------------"
-        )
-
     raise ModuleNotFoundError(
         "\n--------------------\n"
         "You are attempting to perform interferometer analysis with the default "

pyproject.toml

@@ -8,7 +8,7 @@ dynamic = ["version"]
   description="PyAuto Data Structures"
 readme = { file = "README.md", content-type = "text/markdown" }
 license = { text = "MIT" }
-requires-python = ">=3.9"
+requires-python = ">=3.12"
 authors = [
     { name = "James Nightingale", email = "James.Nightingale@newcastle.ac.uk" },
     { name = "Richard Hayes", email = "richard@rghsoftware.co.uk" },
@@ -18,9 +18,6 @@ classifiers = [
   "Topic :: Scientific/Engineering :: Physics",
   "Natural Language :: English",
   "Operating System :: OS Independent",
-  "Programming Language :: Python :: 3.9",
-  "Programming Language :: Python :: 3.10",
-  "Programming Language :: Python :: 3.11",
   "Programming Language :: Python :: 3.12",
   "Programming Language :: Python :: 3.13"
 ]
@@ -56,7 +53,7 @@ jax = ["autonerves[jax]"]
 optional = [
     "autoarray[jax]",
     "numba",
-    "nufftax>=0.4.0,<0.5.0; python_version >= '3.12'",
+    "nufftax>=0.4.0,<0.5.0",
     "pynufft",
     # tfp provides the modified-Bessel `bessel_kve` used by the JAX Matern-kernel
     # regularization path (autoarray/inversion/regularization/matern_kernel.py).
@@ -67,11 +64,11 @@ optional = [
     "tfp-nightly==0.26.0.dev20260713"
 ]
 test = ["pytest"]
-dev = ["pytest", "black", "numba", "nufftax>=0.4.0,<0.5.0; python_version >= '3.12'", "pynufft==2022.2.2"]
+dev = ["pytest", "black", "numba", "nufftax>=0.4.0,<0.5.0", "pynufft==2022.2.2"]
 
 [tool.pytest.ini_options]
 testpaths = ["test_autoarray"]
 filterwarnings = [
     "ignore:cuda_plugin_extension:UserWarning",
     "ignore::DeprecationWarning:jax",
-]
+]

test_autoarray/conftest.py

@@ -287,13 +287,11 @@ def pytest_collection_modifyitems(config, items):
     it is not installed.
 
     The default ``Interferometer`` transformer is the JAX-native
-    ``TransformerNUFFT``, backed by ``nufftax`` — which is declared for Python
-    ``>= 3.12`` only (Python 3.9-3.11 are not officially supported). On those
-    interpreters the backend cannot be installed, so any test that builds a
-    default interferometer/transformer raises ``ModuleNotFoundError`` at
-    construction. Skip exactly those, while keeping the explicit ``DFT`` and
-    ``pynufft`` transformer tests, which have no ``nufftax`` dependency. On
-    3.12/3.13 (where ``nufftax`` is present) nothing is skipped.
+    ``TransformerNUFFT``, backed by the optional ``nufftax`` dependency. When
+    that backend is absent, any test that builds a default interferometer or
+    transformer raises ``ModuleNotFoundError`` at construction. Skip exactly
+    those tests while keeping the explicit ``DFT`` and ``pynufft`` transformer
+    tests, which have no ``nufftax`` dependency.
     """
     try:
         import nufftax  # noqa: F401
@@ -303,7 +301,7 @@ def pytest_collection_modifyitems(config, items):
         pass
 
     skip_nufftax = pytest.mark.skip(
-        reason="nufftax (default JAX interferometer backend) requires Python >= 3.12"
+        reason="nufftax (the default JAX interferometer backend) is not installed"
     )
     for item in items:
         name = item.nodeid.rsplit("::", 1)[-1]

test_autoarray/operators/test_transformer.py

@@ -308,47 +308,11 @@ def f(img_arr):
     )
 
 
-def test__nufftax_exception__pre_3_12_tells_user_to_upgrade_not_to_pip_install():
-    """
-    On Python < 3.12 there is no usable nufftax release: the versions that work
-    require >= 3.12, and the ones that do install (0.6.x) break mid-fit. The
-    error must therefore steer the user away from `pip install nufftax`.
-    """
-    import collections
-    from unittest import mock
+def test__nufftax_exception__keeps_the_plain_install_instruction():
     from autoarray.operators import transformer
 
-    version_info = collections.namedtuple(
-        "version_info", "major minor micro releaselevel serial"
-    )
-
-    with mock.patch.object(
-        transformer.sys, "version_info", version_info(3, 11, 0, "final", 0)
-    ):
-        with pytest.raises(ModuleNotFoundError) as exc:
-            transformer.nufftax_exception()
-
-    message = str(exc.value)
-    assert "Python 3.11" in message
-    assert "Do NOT `pip install nufftax`" in message
-    assert "Python 3.12 or newer" in message
-    assert "TransformerNUFFTPyNUFFT" in message
-
-
-def test__nufftax_exception__3_12_and_later_keeps_the_plain_install_instruction():
-    import collections
-    from unittest import mock
-    from autoarray.operators import transformer
-
-    version_info = collections.namedtuple(
-        "version_info", "major minor micro releaselevel serial"
-    )
-
-    with mock.patch.object(
-        transformer.sys, "version_info", version_info(3, 12, 0, "final", 0)
-    ):
-        with pytest.raises(ModuleNotFoundError) as exc:
-            transformer.nufftax_exception()
+    with pytest.raises(ModuleNotFoundError) as exc:
+        transformer.nufftax_exception()
 
     message = str(exc.value)
     assert "Install it via the command `pip install nufftax`" in message