Skip to content

Commit d04b958

Browse files
Jammy2211Jammy2211
authored andcommitted
feat: dispatch autogalaxy visualizers via fit_for_visualization
Closes #389. - imaging/model/visualizer.py — both single-analysis visualize() and visualize_combined() multi-analysis path - interferometer/model/visualizer.py — single-analysis visualize() PyAutoLens made the same swap in #443 (2026-04-19) when use_jax_for_visualization landed in autofit. PyAutoGalaxy got matching pytree registration for FitImaging (#364) and FitInterferometer (#376) but the dispatch call sites were never swapped, leaving use_jax_for_visualization=True a silent no-op on autogalaxy. This PR fixes that. Safe for the NumPy default path: fit_for_visualization falls back to fit_from whenever use_jax_for_visualization=False.
1 parent 07432c2 commit d04b958

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

autogalaxy/imaging/model/visualizer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ def visualize(
7676
An instance of the model that is being fitted to the data by this analysis (whose parameters have been set
7777
via a non-linear search).
7878
"""
79-
fit = analysis.fit_from(instance=instance)
79+
fit = analysis.fit_for_visualization(instance=instance)
8080

8181
plotter = PlotterImaging(
8282
image_path=paths.image_path, title_prefix=analysis.title_prefix
@@ -173,7 +173,7 @@ def visualize_combined(
173173
)
174174

175175
fit_list = [
176-
analysis.fit_from(instance=single_instance)
176+
analysis.fit_for_visualization(instance=single_instance)
177177
for analysis, single_instance in zip(analyses, instance)
178178
]
179179

autogalaxy/interferometer/model/visualizer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def visualize(
7878
An instance of the model that is being fitted to the data by this analysis (whose parameters have been set
7979
via a non-linear search).
8080
"""
81-
fit = analysis.fit_from(instance=instance)
81+
fit = analysis.fit_for_visualization(instance=instance)
8282

8383
plotter = PlotterInterferometer(
8484
image_path=paths.image_path, title_prefix=analysis.title_prefix

0 commit comments

Comments
 (0)