Skip to content

Commit a04a501

Browse files
Jammy2211Jammy2211
authored andcommitted
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.
1 parent 3c4fcc0 commit a04a501

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

autolens/point/model/analysis.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ def __init__(
4343
cosmology: ag.cosmo.LensingCosmology = None,
4444
title_prefix: str = None,
4545
use_jax: bool = True,
46+
**kwargs,
4647
):
4748
"""
4849
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__(
7778
A string that is added before the title of all figures output by visualization, for example to
7879
put the name of the dataset and galaxy in the title.
7980
"""
80-
super().__init__(cosmology=cosmology, use_jax=use_jax)
81+
super().__init__(cosmology=cosmology, use_jax=use_jax, **kwargs)
8182

8283
AnalysisLens.__init__(self=self, cosmology=cosmology, use_jax=use_jax)
8384

0 commit comments

Comments
 (0)