Skip to content

Feature/workflow extend #373

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Jammy2211 merged 2 commits into from
Nov 19, 2025
Merged

Feature/workflow extend #373

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
6140e05
FITSModelGalaxyImages added to autolens
Nov 19, 2025
4ce57ea
shortern titles on subplot
Nov 19, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions autolens/aggregator/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,5 +27,6 @@
from autolens.aggregator.subplot import SubplotFitX1Plane as subplot_fit_x1_plane
from autolens.aggregator.subplot import SubplotFit as subplot_fit
from autolens.aggregator.subplot import SubplotFitLog10 as subplot_fit_log10
from autolens.aggregator.subplot import FITSModelGalaxyImages as fits_model_galaxy_images
from autolens.aggregator.subplot import FITSTracer as fits_tracer
from autolens.aggregator.subplot import FITSFit as fits_fit
9 changes: 9 additions & 0 deletions autolens/aggregator/subplot.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,15 @@
from enum import Enum


class FITSModelGalaxyImages(Enum):
"""
The HDUs that can be extracted from the fit.fits file.
"""

lens_light_image = "GALAXY_0"
lensed_source_image = "GALAXY_1"


class FITSTracer(Enum):
"""
The HDUs that can be extracted from the fit.fits file.
Expand Down
1 change: 0 additions & 1 deletion autolens/imaging/model/analysis.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,4 @@
import logging
import numpy as np

import autofit as af
import autogalaxy as ag
Expand Down
16 changes: 8 additions & 8 deletions autolens/imaging/plot/fit_imaging_plotters.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -428,7 +428,7 @@ def subplot_fit_x1_plane(self):
self.mat_plot_2d.cmap.kwargs.pop("vmax")

self.residuals_symmetric_cmap = False
self.set_title(label="Lens Light Subtracted Image")
self.set_title(label="Lens Light Subtracted")
self.figures_2d(normalized_residual_map=True)

self.mat_plot_2d.cmap.kwargs["vmin"] = 0.0
Expand Down Expand Up @@ -468,7 +468,7 @@ def subplot_fit_log10_x1_plane(self):
self.mat_plot_2d.cmap.kwargs.pop("vmax")

self.residuals_symmetric_cmap = False
self.set_title(label="Lens Light Subtracted Image")
self.set_title(label="Lens Light Subtracted")
self.figures_2d(normalized_residual_map=True)

self.residuals_symmetric_cmap = True
Expand Down Expand Up @@ -519,15 +519,15 @@ def subplot_fit(self, plane_index: Optional[int] = None):

self.mat_plot_2d.cmap.kwargs["vmin"] = 0.0

self.set_title(label="Lens Light Subtracted Image")
self.set_title(label="Lens Light Subtracted")
self.figures_2d_of_planes(
plane_index=plane_index,
subtracted_image=True,
use_source_vmax=True,
remove_critical_caustic=True,
)

self.set_title(label="Source Model Image (Image Plane)")
self.set_title(label="Source Model Image")
self.figures_2d_of_planes(
plane_index=plane_index,
model_image=True,
Expand All @@ -551,7 +551,7 @@ def subplot_fit(self, plane_index: Optional[int] = None):
self.mat_plot_2d.cmap.kwargs["vmin"] = -1.0
self.mat_plot_2d.cmap.kwargs["vmax"] = 1.0

self.set_title(label="Normalized Residual Map (1 sigma)")
self.set_title(label=r"Normalized Residual Map $1\sigma$")
self.figures_2d(normalized_residual_map=True)
self.set_title(label=None)

Expand Down Expand Up @@ -624,12 +624,12 @@ def subplot_fit_log10(self, plane_index: Optional[int] = None):

self.mat_plot_2d.cmap.kwargs["vmin"] = 0.0

self.set_title(label="Lens Light Subtracted Image")
self.set_title(label="Lens Light Subtracted")
self.figures_2d_of_planes(
plane_index=plane_index, subtracted_image=True, use_source_vmax=True, remove_critical_caustic=True
)

self.set_title(label="Source Model Image (Image Plane)")
self.set_title(label="Source Model Image")
self.figures_2d_of_planes(
plane_index=plane_index, model_image=True, use_source_vmax=True, remove_critical_caustic=True
)
Expand All @@ -652,7 +652,7 @@ def subplot_fit_log10(self, plane_index: Optional[int] = None):
self.mat_plot_2d.cmap.kwargs["vmin"] = -1.0
self.mat_plot_2d.cmap.kwargs["vmax"] = 1.0

self.set_title(label="Normalized Residual Map (1 sigma)")
self.set_title(label=r"Normalized Residual Map $1\sigma$")
self.figures_2d(normalized_residual_map=True)
self.set_title(label=None)

Expand Down
2 changes: 1 addition & 1 deletion autolens/interferometer/plot/fit_interferometer_plotters.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -420,7 +420,7 @@ def subplot_fit(self):
self.mat_plot_2d.cmap.kwargs["vmin"] = -1.0
self.mat_plot_2d.cmap.kwargs["vmax"] = 1.0

self.set_title(label="Normalized Residual Map (1 sigma)")
self.set_title(label=r"Normalized Residual Map $1\sigma$")
self.figures_2d(dirty_normalized_residual_map=True)
self.set_title(label=None)

Expand Down
Loading