From 0af813aa361bb3aa563be606bc33b062b025f81d Mon Sep 17 00:00:00 2001 From: Jammy2211 Date: Wed, 29 Oct 2025 12:08:19 +0000 Subject: [PATCH] preloads added to interferometer stuff --- autolens/interferometer/fit_interferometer.py | 4 ++++ autolens/interferometer/model/analysis.py | 3 +++ 2 files changed, 7 insertions(+) diff --git a/autolens/interferometer/fit_interferometer.py b/autolens/interferometer/fit_interferometer.py index 4e59c287e..85506a818 100644 --- a/autolens/interferometer/fit_interferometer.py +++ b/autolens/interferometer/fit_interferometer.py @@ -20,6 +20,7 @@ def __init__( dataset_model: Optional[aa.DatasetModel] = None, adapt_images: Optional[ag.AdaptImages] = None, settings_inversion: aa.SettingsInversion = aa.SettingsInversion(), + preloads: aa.Preloads = None, ): """ Fits an interferometer dataset using a `Tracer` object. @@ -80,6 +81,8 @@ def __init__( self=self, model_obj=tracer, settings_inversion=settings_inversion ) + self.preloads = preloads + @property def profile_visibilities(self) -> aa.Visibilities: """ @@ -113,6 +116,7 @@ def tracer_to_inversion(self) -> TracerToInversion: tracer=self.tracer, adapt_images=self.adapt_images, settings_inversion=self.settings_inversion, + preloads=self.preloads ) @cached_property diff --git a/autolens/interferometer/model/analysis.py b/autolens/interferometer/model/analysis.py index aa9d11133..9de9fe530 100644 --- a/autolens/interferometer/model/analysis.py +++ b/autolens/interferometer/model/analysis.py @@ -32,6 +32,7 @@ def __init__( adapt_image_maker: Optional[ag.AdaptImageMaker] = None, cosmology: ag.cosmo.LensingCosmology = None, settings_inversion: aa.SettingsInversion = None, + preloads: aa.Preloads = None, raise_inversion_positions_likelihood_exception: bool = True, title_prefix: str = None, ): @@ -82,6 +83,7 @@ def __init__( adapt_image_maker=adapt_image_maker, cosmology=cosmology, settings_inversion=settings_inversion, + preloads=preloads, raise_inversion_positions_likelihood_exception=raise_inversion_positions_likelihood_exception, title_prefix=title_prefix, ) @@ -193,6 +195,7 @@ def fit_from( tracer=tracer, adapt_images=adapt_images, settings_inversion=self.settings_inversion, + preloads=self.preloads ) def save_attributes(self, paths: af.DirectoryPaths):