Skip to content

Commit fd83da4

Browse files
Jammy2211Jammy2211
authored andcommitted
black
1 parent 30ed351 commit fd83da4

3 files changed

Lines changed: 8 additions & 8 deletions

File tree

autolens/interferometer/fit_interferometer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -173,7 +173,9 @@ def galaxy_image_dict(self) -> Dict[ag.Galaxy, np.ndarray]:
173173
For modeling, this dictionary is used to set up the `adapt_images` that adapt certain pixelizations to the
174174
data being fitted.
175175
"""
176-
galaxy_image_dict = self.tracer.galaxy_image_2d_dict_from(grid=self.grids.lp, xp=self._xp)
176+
galaxy_image_dict = self.tracer.galaxy_image_2d_dict_from(
177+
grid=self.grids.lp, xp=self._xp
178+
)
177179

178180
galaxy_linear_obj_image_dict = self.galaxy_linear_obj_data_dict_from(
179181
use_operated=False

autolens/lens/tracer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -610,7 +610,9 @@ def galaxy_image_2d_dict_from(
610610
for plane_index, galaxies in enumerate(self.planes):
611611
image_2d_list = [
612612
galaxy.image_2d_from(
613-
grid=traced_grid_list[plane_index], operated_only=operated_only, xp=xp
613+
grid=traced_grid_list[plane_index],
614+
operated_only=operated_only,
615+
xp=xp,
614616
)
615617
for galaxy in galaxies
616618
]

test_autolens/interferometer/test_fit_interferometer.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -180,12 +180,8 @@ def test___galaxy_image_dict(interferometer_7, interferometer_7_grid):
180180
tracer=tracer,
181181
)
182182

183-
assert fit.galaxy_image_dict[g0_linear][4] == pytest.approx(
184-
1.00018622848, 1.0e-2
185-
)
186-
assert fit.galaxy_image_dict[g1_linear][3] == pytest.approx(
187-
-0.017435532289, 1.0e-2
188-
)
183+
assert fit.galaxy_image_dict[g0_linear][4] == pytest.approx(1.00018622848, 1.0e-2)
184+
assert fit.galaxy_image_dict[g1_linear][3] == pytest.approx(-0.017435532289, 1.0e-2)
189185

190186
pixelization = al.Pixelization(
191187
mesh=al.mesh.RectangularUniform(shape=(3, 3)),

0 commit comments

Comments
 (0)