From bedcf9a18ecb65884aae5388282fc27b9f551312 Mon Sep 17 00:00:00 2001 From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com> Date: Fri, 30 Jan 2026 14:53:00 +0000 Subject: [PATCH 1/3] Initial plan From 0f13b2f03ec63ceace02454a03a7c35d800cebcc Mon Sep 17 00:00:00 2001 From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com> Date: Fri, 30 Jan 2026 14:54:32 +0000 Subject: [PATCH 2/3] Update FlatLambdaCDM docstring to reflect JAX-compatible implementation Co-authored-by: Jammy2211 <23455639+Jammy2211@users.noreply.github.com> --- autogalaxy/cosmology/model.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/autogalaxy/cosmology/model.py b/autogalaxy/cosmology/model.py index 4cccf234e..0c452892e 100644 --- a/autogalaxy/cosmology/model.py +++ b/autogalaxy/cosmology/model.py @@ -399,15 +399,15 @@ def __init__( Ob0: float = 0.04897, ): """ - A wrapper for the astropy `FlatLambdaCDM` cosmology class, which allows it to be used for modeling such - that the cosmological parameters are free parameters which can be fitted for. + A JAX-compatible implementation of the FlatLambdaCDM cosmology model, which allows it to be used for + modeling such that the cosmological parameters are free parameters which can be fitted for. - The interface of this class is the same as the astropy `FlatLambdaCDM` class, it simply overwrites the - __init__ method and inherits from it in a way that ensures **PyAutoFit** can compose a model from it - without issue. + This is a custom implementation designed to work with both NumPy and JAX backends, enabling automatic + differentiation and GPU acceleration when needed. The interface and parameter naming conventions follow + the astropy `FlatLambdaCDM` class for compatibility. - The class also inherits from `LensingCosmology`, which is a class that provides additional functionality - for calculating lensing specific quantities in the cosmology. + The class inherits from `LensingCosmology`, which provides additional functionality for calculating + lensing specific quantities in the cosmology. Parameters ---------- From 1042a7193d18b6d911b5c5ddd101cf9299849a6a Mon Sep 17 00:00:00 2001 From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com> Date: Fri, 30 Jan 2026 14:56:26 +0000 Subject: [PATCH 3/3] Remove trailing whitespace from docstring Co-authored-by: Jammy2211 <23455639+Jammy2211@users.noreply.github.com> --- autogalaxy/cosmology/model.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/autogalaxy/cosmology/model.py b/autogalaxy/cosmology/model.py index 0c452892e..bfbce142b 100644 --- a/autogalaxy/cosmology/model.py +++ b/autogalaxy/cosmology/model.py @@ -399,14 +399,14 @@ def __init__( Ob0: float = 0.04897, ): """ - A JAX-compatible implementation of the FlatLambdaCDM cosmology model, which allows it to be used for + A JAX-compatible implementation of the FlatLambdaCDM cosmology model, which allows it to be used for modeling such that the cosmological parameters are free parameters which can be fitted for. - This is a custom implementation designed to work with both NumPy and JAX backends, enabling automatic - differentiation and GPU acceleration when needed. The interface and parameter naming conventions follow + This is a custom implementation designed to work with both NumPy and JAX backends, enabling automatic + differentiation and GPU acceleration when needed. The interface and parameter naming conventions follow the astropy `FlatLambdaCDM` class for compatibility. - The class inherits from `LensingCosmology`, which provides additional functionality for calculating + The class inherits from `LensingCosmology`, which provides additional functionality for calculating lensing specific quantities in the cosmology. Parameters