Skip to content

Commit 0740467

Browse files
Jammy2211Jammy2211
authored andcommitted
refactor complete
1 parent 334250d commit 0740467

14 files changed

Lines changed: 66 additions & 69 deletions

File tree

autolens/__init__.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -20,11 +20,9 @@
2020
from autoarray.inversion.pixelization.mesh.abstract import AbstractMesh
2121
from autoarray.inversion.regularization.abstract import AbstractRegularization
2222
from autoarray.inversion.pixelization.pixelization import Pixelization
23-
from autoarray.inversion.inversion.settings import SettingsInversion
23+
from autoarray.settings import Settings
2424
from autoarray.inversion.inversion.factory import inversion_from as Inversion
2525
from autoarray.inversion.pixelization.mappers.abstract import AbstractMapper
26-
from autoarray.inversion.pixelization.mappers.mapper_grids import MapperGrids
27-
from autoarray.inversion.pixelization.mappers.factory import mapper_from as Mapper
2826
from autoarray.inversion.pixelization.border_relocator import BorderRelocator
2927
from autoarray.operators.transformer import TransformerDFT
3028
from autoarray.operators.transformer import TransformerNUFFT
@@ -37,8 +35,8 @@
3735
from autoarray.structures.grids.uniform_1d import Grid1D
3836
from autoarray.structures.grids.uniform_2d import Grid2D
3937
from autoarray.structures.grids.irregular_2d import Grid2DIrregular
40-
from autoarray.inversion.pixelization.mesh_grid.rectangular_2d import Mesh2DRectangular
41-
from autoarray.inversion.pixelization.mesh_grid.delaunay_2d import Mesh2DDelaunay
38+
from autoarray.inversion.pixelization.mesh_grid.rectangular import Mesh2DRectangular
39+
from autoarray.inversion.pixelization.mesh_grid.delaunay import Mesh2DDelaunay
4240
from autoarray.structures.triangles.shape import Circle
4341
from autoarray.structures.triangles.shape import Triangle
4442
from autoarray.structures.triangles.shape import Square

autolens/aggregator/fit_imaging.py

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
def _fit_imaging_from(
1616
fit: af.Fit,
1717
instance: Optional[af.ModelInstance] = None,
18-
settings_inversion: aa.SettingsInversion = None,
18+
settings: aa.Settings = None,
1919
) -> List[FitImaging]:
2020
"""
2121
Returns a list of `FitImaging` object from a `PyAutoFit` loaded directory `Fit` or sqlite database `Fit` object.
@@ -25,7 +25,7 @@ def _fit_imaging_from(
2525
2626
- The imaging data, noise-map, PSF and settings as .fits files (e.g. `dataset/data.fits`).
2727
- The mask used to mask the `Imaging` data structure in the fit (`dataset.fits[hdu=0]`).
28-
- The settings of inversions used by the fit (`dataset/settings_inversion.json`).
28+
- The settings of inversions used by the fit (`dataset/settings.json`).
2929
3030
Each individual attribute can be loaded from the database via the `fit.value()` method.
3131
@@ -36,7 +36,7 @@ def _fit_imaging_from(
3636
is instead used to load lists of the data, noise-map, PSF and mask and combine them into a list of
3737
`FitImaging` objects.
3838
39-
The settings of an inversion can be overwritten by inputting a `settings_inversion` object, for example
39+
The settings of an inversion can be overwritten by inputting a `settings` object, for example
4040
if you want to use a grid with a different inversion solver.
4141
4242
Parameters
@@ -47,8 +47,8 @@ def _fit_imaging_from(
4747
instance
4848
A manual instance that overwrites the max log likelihood instance in fit (e.g. for drawing the instance
4949
randomly from the PDF).
50-
settings_inversion
51-
Optionally overwrite the `SettingsInversion` of the `Inversion` object that is created from the fit.
50+
settings
51+
Optionally overwrite the `Settings` of the `Inversion` object that is created from the fit.
5252
"""
5353

5454
dataset_list = _imaging_from(fit=fit)
@@ -59,7 +59,7 @@ def _fit_imaging_from(
5959

6060
adapt_images_list = agg_util.adapt_images_from(fit=fit)
6161

62-
settings_inversion = settings_inversion or fit.value(name="settings_inversion")
62+
settings = settings or fit.value(name="settings")
6363

6464
fit_dataset_list = []
6565

@@ -75,7 +75,7 @@ def _fit_imaging_from(
7575
tracer=tracer,
7676
dataset_model=dataset_model,
7777
adapt_images=adapt_images,
78-
settings_inversion=settings_inversion,
78+
settings=settings,
7979
)
8080
)
8181

@@ -86,7 +86,7 @@ class FitImagingAgg(af.AggBase):
8686
def __init__(
8787
self,
8888
aggregator: af.Aggregator,
89-
settings_inversion: Optional[aa.SettingsInversion] = None,
89+
settings: Optional[aa.Settings] = None,
9090
):
9191
"""
9292
Interfaces with an `PyAutoFit` aggregator object to create instances of `FitImaging` objects from the results
@@ -97,7 +97,7 @@ def __init__(
9797
9898
- The imaging data, noise-map, PSF and settings as .fits files (e.g. `dataset/data.fits`).
9999
- The mask used to mask the `Imaging` data structure in the fit (`dataset.fits[hdu=0]`).
100-
- The settings of inversions used by the fit (`dataset/settings_inversion.json`).
100+
- The settings of inversions used by the fit (`dataset/settings.json`).
101101
102102
The `aggregator` contains the path to each of these files, and they can be loaded individually. This class
103103
can load them all at once and create an `FitImaging` object via the `_fit_imaging_from` method.
@@ -119,16 +119,16 @@ def __init__(
119119
----------
120120
aggregator
121121
A `PyAutoFit` aggregator object which can load the results of model-fits.
122-
settings_inversion
123-
Optionally overwrite the `SettingsInversion` of the `Inversion` object that is created from the fit.
122+
settings
123+
Optionally overwrite the `Settings` of the `Inversion` object that is created from the fit.
124124
use_preloaded_grid
125125
Certain pixelization's construct their mesh in the source-plane from a stochastic KMeans algorithm. This
126126
grid may be output to hard-disk after the model-fit and loaded via the database to ensure the same grid is
127127
used as the fit.
128128
"""
129129
super().__init__(aggregator=aggregator)
130130

131-
self.settings_inversion = settings_inversion
131+
self.settings = settings
132132

133133
def object_via_gen_from(
134134
self, fit, instance: Optional[af.ModelInstance] = None
@@ -150,5 +150,5 @@ def object_via_gen_from(
150150
return _fit_imaging_from(
151151
fit=fit,
152152
instance=instance,
153-
settings_inversion=self.settings_inversion,
153+
settings=self.settings,
154154
)

autolens/aggregator/fit_interferometer.py

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
def _fit_interferometer_from(
1616
fit: af.Fit,
1717
instance: Optional[af.ModelInstance] = None,
18-
settings_inversion: aa.SettingsInversion = None,
18+
settings: aa.Settings = None,
1919
) -> List[FitInterferometer]:
2020
"""
2121
Returns a list of `FitInterferometer` objects from a `PyAutoFit` loaded directory `Fit` or sqlite database `Fit` object.
@@ -25,7 +25,7 @@ def _fit_interferometer_from(
2525
2626
- The interferometer data, noise-map, uv-wavelengths and settings as .fits files (e.g. `dataset/data.fits`).
2727
- The real space mask defining the grid of the interferometer for the FFT (`dataset/real_space_mask.fits`).
28-
- The settings of inversions used by the fit (`dataset/settings_inversion.json`).
28+
- The settings of inversions used by the fit (`dataset/settings.json`).
2929
3030
Each individual attribute can be loaded from the database via the `fit.value()` method.
3131
@@ -37,7 +37,7 @@ def _fit_interferometer_from(
3737
method is instead used to load lists of the data, noise-map, PSF and mask and combine them into a list of
3838
`FitInterferometer` objects.
3939
40-
The settings of an inversion can be overwritten by inputting a `settings_inversion` object, for
40+
The settings of an inversion can be overwritten by inputting a `settings` object, for
4141
example if you want to use a grid with a different inversion solver.
4242
4343
Parameters
@@ -48,8 +48,8 @@ def _fit_interferometer_from(
4848
instance
4949
A manual instance that overwrites the max log likelihood instance in fit (e.g. for drawing the instance
5050
randomly from the PDF).
51-
settings_inversion
52-
Optionally overwrite the `SettingsInversion` of the `Inversion` object that is created from the fit.
51+
settings
52+
Optionally overwrite the `Settings` of the `Inversion` object that is created from the fit.
5353
"""
5454
dataset_list = _interferometer_from(
5555
fit=fit,
@@ -59,7 +59,7 @@ def _fit_interferometer_from(
5959

6060
adapt_images_list = agg_util.adapt_images_from(fit=fit)
6161

62-
settings_inversion = settings_inversion or fit.value(name="settings_inversion")
62+
settings = settings or fit.value(name="settings")
6363

6464
fit_dataset_list = []
6565

@@ -75,7 +75,7 @@ def _fit_interferometer_from(
7575
tracer=tracer,
7676
dataset_model=dataset_model,
7777
adapt_images=adapt_images,
78-
settings_inversion=settings_inversion,
78+
settings=settings,
7979
)
8080
)
8181

@@ -86,7 +86,7 @@ class FitInterferometerAgg(af.AggBase):
8686
def __init__(
8787
self,
8888
aggregator: af.Aggregator,
89-
settings_inversion: Optional[aa.SettingsInversion] = None,
89+
settings: Optional[aa.Settings] = None,
9090
):
9191
"""
9292
Interfaces with an `PyAutoFit` aggregator object to create instances of `FitInterferometer` objects from the
@@ -97,7 +97,7 @@ def __init__(
9797
9898
- The interferometer data, noise-map, uv-wavelengths and settings as .fits files (e.g. `dataset/data.fits`).
9999
- The real space mask defining the grid of the interferometer for the FFT (`dataset/real_space_mask.fits`).
100-
- The settings of inversions used by the fit (`dataset/settings_inversion.json`).
100+
- The settings of inversions used by the fit (`dataset/settings.json`).
101101
102102
The `aggregator` contains the path to each of these files, and they can be loaded individually. This class
103103
can load them all at once and create an `FitInterferometer` object via the `_fit_interferometer_from` method.
@@ -115,16 +115,16 @@ def __init__(
115115
----------
116116
aggregator
117117
A `PyAutoFit` aggregator object which can load the results of model-fits.
118-
settings_inversion
119-
Optionally overwrite the `SettingsInversion` of the `Inversion` object that is created from the fit.
118+
settings
119+
Optionally overwrite the `Settings` of the `Inversion` object that is created from the fit.
120120
use_preloaded_grid
121121
Certain pixelization's construct their mesh in the source-plane from a stochastic KMeans algorithm. This
122122
grid may be output to hard-disk after the model-fit and loaded via the database to ensure the same grid is
123123
used as the fit.
124124
"""
125125
super().__init__(aggregator=aggregator)
126126

127-
self.settings_inversion = settings_inversion
127+
self.settings = settings
128128

129129
def object_via_gen_from(
130130
self, fit, instance: Optional[af.ModelInstance] = None
@@ -146,5 +146,5 @@ def object_via_gen_from(
146146
return _fit_interferometer_from(
147147
fit=fit,
148148
instance=instance,
149-
settings_inversion=self.settings_inversion,
149+
settings=self.settings,
150150
)

autolens/aggregator/subhalo.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,15 +10,15 @@ class SubhaloAgg:
1010
def __init__(
1111
self,
1212
aggregator_grid_search: af.GridSearchAggregator,
13-
settings_inversion: Optional[aa.SettingsInversion] = None,
13+
settings: Optional[aa.Settings] = None,
1414
):
1515
"""
1616
Wraps a PyAutoFit aggregator in order to create generators of fits to imaging data, corresponding to the
1717
results of a non-linear search model-fit.
1818
"""
1919

2020
self.aggregator_grid_search = aggregator_grid_search
21-
self.settings_inversion = settings_inversion
21+
self.settings = settings
2222

2323
if len(aggregator_grid_search) == 0:
2424
raise exc.AggregatorException(

autolens/analysis/analysis/dataset.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ def __init__(
3030
positions_likelihood_list: Optional[List[PositionsLH]] = None,
3131
adapt_images: Optional[ag.AdaptImages] = None,
3232
cosmology: ag.cosmo.LensingCosmology = None,
33-
settings_inversion: aa.SettingsInversion = None,
33+
settings: aa.Settings = None,
3434
preloads: aa.Preloads = None,
3535
raise_inversion_positions_likelihood_exception: bool = True,
3636
title_prefix: str = None,
@@ -60,7 +60,7 @@ def __init__(
6060
reconstructed galaxy's morphology.
6161
cosmology
6262
The Cosmology assumed for this analysis.
63-
settings_inversion
63+
settings
6464
Settings controlling how an inversion is fitted during the model-fit, for example which linear algebra
6565
formalism is used.
6666
raise_inversion_positions_likelihood_exception
@@ -74,7 +74,7 @@ def __init__(
7474
dataset=dataset,
7575
adapt_images=adapt_images,
7676
cosmology=cosmology,
77-
settings_inversion=settings_inversion,
77+
settings=settings,
7878
preloads=preloads,
7979
title_prefix=title_prefix,
8080
use_jax=use_jax,

autolens/imaging/fit_imaging.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ def __init__(
2222
tracer: Tracer,
2323
dataset_model : Optional[aa.DatasetModel] = None,
2424
adapt_images: Optional[ag.AdaptImages] = None,
25-
settings_inversion: aa.SettingsInversion = aa.SettingsInversion(),
25+
settings: aa.Settings = None,
2626
preloads: aa.Preloads = None,
2727
xp=np
2828
):
@@ -61,19 +61,19 @@ def __init__(
6161
adapt_images
6262
Contains the adapt-images which are used to make a pixelization's mesh and regularization adapt to the
6363
reconstructed galaxy's morphology.
64-
settings_inversion
64+
settings
6565
Settings controlling how an inversion is fitted for example which linear algebra formalism is used.
6666
"""
6767

6868
super().__init__(dataset=dataset, dataset_model=dataset_model, xp=xp)
6969
AbstractFitInversion.__init__(
70-
self=self, model_obj=tracer, settings_inversion=settings_inversion, xp=xp
70+
self=self, model_obj=tracer, settings=settings, xp=xp
7171
)
7272

7373
self.tracer = tracer
7474

7575
self.adapt_images = adapt_images
76-
self.settings_inversion = settings_inversion
76+
self.settings = settings or aa.Settings()
7777
self.preloads = preloads
7878

7979
@property
@@ -119,7 +119,7 @@ def tracer_to_inversion(self) -> TracerToInversion:
119119
dataset=dataset,
120120
tracer=self.tracer,
121121
adapt_images=self.adapt_images,
122-
settings_inversion=self.settings_inversion,
122+
settings=self.settings,
123123
preloads=self.preloads,
124124
xp=self._xp,
125125
)

autolens/imaging/model/analysis.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ def fit_from(
107107
tracer=tracer,
108108
dataset_model=dataset_model,
109109
adapt_images=adapt_images,
110-
settings_inversion=self.settings_inversion,
110+
settings=self.settings,
111111
preloads=self.preloads,
112112
xp=self._xp
113113
)

autolens/interferometer/fit_interferometer.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ def __init__(
1919
tracer: Tracer,
2020
dataset_model: Optional[aa.DatasetModel] = None,
2121
adapt_images: Optional[ag.AdaptImages] = None,
22-
settings_inversion: aa.SettingsInversion = aa.SettingsInversion(),
22+
settings: aa.Settings = None,
2323
preloads: aa.Preloads = None,
2424
xp=np,
2525
):
@@ -59,23 +59,23 @@ def __init__(
5959
adapt_images
6060
Contains the adapt-images which are used to make a pixelization's mesh and regularization adapt to the
6161
reconstructed galaxy's morphology.
62-
settings_inversion
62+
settings
6363
Settings controlling how an inversion is fitted for example which linear algebra formalism is used.
6464
"""
6565

6666
self.tracer = tracer
6767

6868
self.adapt_images = adapt_images
6969

70-
self.settings_inversion = settings_inversion
70+
self.settings = settings
7171

7272
super().__init__(
7373
dataset=dataset,
7474
dataset_model=dataset_model,
7575
xp=xp,
7676
)
7777
AbstractFitInversion.__init__(
78-
self=self, model_obj=tracer, settings_inversion=settings_inversion, xp=xp
78+
self=self, model_obj=tracer, settings=settings, xp=xp
7979
)
8080

8181
self.preloads = preloads
@@ -122,7 +122,7 @@ def tracer_to_inversion(self) -> TracerToInversion:
122122
dataset=dataset,
123123
tracer=self.tracer,
124124
adapt_images=self.adapt_images,
125-
settings_inversion=self.settings_inversion,
125+
settings=self.settings,
126126
preloads=self.preloads,
127127
xp=self._xp,
128128
)

0 commit comments

Comments
 (0)