Python support up to 3.14

Author: FlorianPfaff

.github/workflows/tests.yml

@@ -18,7 +18,7 @@ jobs:
     strategy:
       matrix:
         backend: [numpy, pytorch, jax]
-        python-version: ["3.10", "3.11", "3.12", "3.13", "3.14"]
+        python-version: ["3.11", "3.12", "3.13", "3.14"]
     permissions:
       checks: write
       pull-requests: write
@@ -27,6 +27,24 @@ jobs:
       - name: Check out repository
         uses: actions/checkout@v4
 
+      - name: Install system build dependencies (when binary not available)
+        run: |
+          sudo apt-get update
+          sudo apt-get install -y \
+          gfortran pkg-config ninja-build \
+          libopenblas-dev liblapack-dev \
+          libfftw3-dev
+        
+      - name: Install LLVM toolchain for llvmlite (cp314 only)
+        if: ${{ matrix.python-version == '3.14' }}
+        run: |
+          sudo apt-get install -y \
+            llvm-20 llvm-20-dev clang-20 \
+            libedit-dev libcurl4-openssl-dev cmake
+          echo "LLVM_CONFIG=/usr/lib/llvm-20/bin/llvm-config" >> $GITHUB_ENV
+          echo "CC=clang-20" >> $GITHUB_ENV
+          echo "CXX=clang++-20" >> $GITHUB_ENV
+
       - name: Set up Python
         uses: actions/setup-python@v5
         with:
@@ -39,11 +57,10 @@ jobs:
           poetry env use python
           poetry install --extras "healpy_support"
           
-      - name: Install pytorch specific dependencies
+      - name: Install pytorch specific dependencies afterward
         if: ${{ matrix.backend == 'pytorch' }}
         run: |
           poetry install --extras "healpy_support" --extras "pytorch_support"
-          poetry run python -m pip install torch==2.1.0+cpu torchaudio==2.1.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
           
       - name: Run tests
         run: |
@@ -55,7 +72,7 @@ jobs:
       - name: Upload test result artifact
         uses: actions/upload-artifact@v4
         with:
-          name: ${{ matrix.backend }}-test-results
+          name: ${{ matrix.backend }}-${{ matrix.python-version }}-test-results
           path: junit_test_results_${{ matrix.backend }}.xml
 
   publish-results:

poetry.lock

@@ -6,7 +6,7 @@ authors = ["Florian Pfaff <pfaff@ias.uni-stuttgart.de>"]
 version = "0.8.0"
 
 [tool.poetry.dependencies]
-python = ">=3.10,<3.13"
+python = ">=3.11,<3.15"
 numpy = "*"
 scipy = "^1.14.1"
 matplotlib = "*"

pyproject.toml

@@ -280,6 +280,11 @@ def _create_backend_module(self, backend_name: str):
         new_module = types.ModuleType(self._path)
         new_module.__file__ = getattr(backend, "__file__", None)
 
+        # expose chosen backend
+        new_module.__backend_name__ = backend_name
+        new_module.BACKEND_NAME = backend_name
+        new_module.get_backend_name = staticmethod(lambda: backend_name)
+
         for module_name, attributes in BACKEND_ATTRIBUTES.items():
             if module_name:
                 try:
@@ -336,5 +341,8 @@ def exec_module(self, module):
             module.set_default_dtype("float64")
         logging.info(f"Using {BACKEND_NAME} backend")
 
-# Insert at the *front* of sys.meta_path to ensure it takes precedence
-sys.meta_path.insert(0, BackendImporter("pyrecest.backend"))
+TARGET = "pyrecest.backend"
+if not any(isinstance(f, BackendImporter) and getattr(f, "_path", None) == TARGET
+           for f in sys.meta_path):
+    # put it in front so it intercepts 'pyrecest.backend'
+    sys.meta_path.insert(0, BackendImporter(TARGET))

pyrecest/_backend/__init__.py

@@ -64,7 +64,7 @@ def normalize(self, verify: bool | None = None) -> "AbstractCustomDistribution":
         :returns: A copy of the original distribution, with the PDF normalized.
         """
         assert (
-            pyrecest.backend.__name__ == "pyrecest.numpy"
+            pyrecest.backend.__backend_name__ == "numpy"
         ), "Only supported for numpy backend"
         cd = copy.deepcopy(self)
 

pyrecest/distributions/abstract_custom_distribution.py

@@ -82,7 +82,7 @@ def sample_metropolis_hastings(
         # jscpd:ignore-end
         """Metropolis Hastings sampling algorithm."""
         assert (
-            pyrecest.backend.__name__ != "pyrecest.jax"
+            pyrecest.backend.__backend_name__ != "jax"
         ), "Not supported on this backend"
         if proposal is None or start_point is None:
             raise NotImplementedError(

pyrecest/distributions/abstract_manifold_specific_distribution.py

@@ -300,9 +300,9 @@ def mode_numerical(self, starting_point=None):
         m : ndarray
           The mode of the distribution.
         """
-        assert pyrecest.backend.__name__ in (
-            "pyrecest.numpy",
-            "pyrecest.pytorch",
+        assert pyrecest.backend.__backend_name__ in (
+            "numpy",
+            "pytorch",
         ), "Not supported for this backend."
         if starting_point is None:
             starting_point = concatenate(

pyrecest/distributions/cart_prod/abstract_hypercylindrical_distribution.py

@@ -313,7 +313,7 @@ def from_distribution(
         else:
             xs = arange(
                 0.0, 2.0 * pi, 2.0 * pi / n
-            )  # Like linspace without endpoint but with compatbiility for pytroch
+            )  # Like linspace without endpoint but with compatbiility for pytorch
             fvals = distribution.pdf(xs)
             if transformation == "identity":
                 pass

pyrecest/distributions/circle/circular_fourier_distribution.py

@@ -69,7 +69,7 @@ def pdf(self, xs):
         x = where(x < 0, x + 2.0 * pi, x)
         x -= self.mu
         max_iterations: int = 1000
-        if pyrecest.backend.__name__ != "pyrecest.jax":
+        if pyrecest.backend.__backend_name__ != "jax":
             n_inputs = xs.shape[0]
             result = zeros(n_inputs)
 

pyrecest/distributions/circle/wrapped_normal_distribution.py

@@ -151,9 +151,9 @@ def integrate_fun_over_domain(
         )
 
     def mode_numerical(self):
-        assert pyrecest.backend.__name__ in (
-            "pyrecest.numpy",
-            "pyrecest.pytorch",
+        assert pyrecest.backend.__backend_name__ in (
+            "numpy",
+            "pytorch",
         ), "Not supported for this backend."
 
         def objective_function_2d(s):