From a04a501028136c30879d355d073971e22e102e3d Mon Sep 17 00:00:00 2001 From: Jammy2211 Date: Thu, 14 May 2026 12:18:11 +0100 Subject: [PATCH] fix: add **kwargs passthrough to AnalysisPoint.__init__ AnalysisPoint's explicit __init__ omitted **kwargs, so any kwarg not in its named-parameter list (notably use_jax_for_visualization from the autofit base Analysis) raised TypeError instead of being forwarded to super. This left use_jax_for_visualization=True silently unusable on al.AnalysisPoint despite the point visualizer dispatching via fit_for_visualization (autolens/point/model/visualizer.py:67). Mirrors the AnalysisInterferometer fix in PR #500. The autolens_workspace_test point_source scripts (visualization.py, visualization_jax.py, modeling_visualization_jit.py) follow in PyAutoLabs/autolens_workspace_test#90 sibling PR. ag.AnalysisInterferometer has the same **kwargs gap on the autogalaxy side; deferred to Phase 1C of the JAX visualization roadmap. --- autolens/point/model/analysis.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/autolens/point/model/analysis.py b/autolens/point/model/analysis.py index bc613767b..9341fb040 100644 --- a/autolens/point/model/analysis.py +++ b/autolens/point/model/analysis.py @@ -43,6 +43,7 @@ def __init__( cosmology: ag.cosmo.LensingCosmology = None, title_prefix: str = None, use_jax: bool = True, + **kwargs, ): """ Fits a lens model to a point source dataset (e.g. positions, fluxes, time delays) via a non-linear search. @@ -77,7 +78,7 @@ def __init__( A string that is added before the title of all figures output by visualization, for example to put the name of the dataset and galaxy in the title. """ - super().__init__(cosmology=cosmology, use_jax=use_jax) + super().__init__(cosmology=cosmology, use_jax=use_jax, **kwargs) AnalysisLens.__init__(self=self, cosmology=cosmology, use_jax=use_jax)