Skip to content

Commit 3db51dd

Browse files
Jammy2211claude
authored andcommitted
update to LensCalc API (rename from OperateDeflections)
- Update import in autolens/__init__.py - tracer_util.py uses ag.LensCalc.from_mass_obj for fermat_potential_from - Update point/fit/abstract.py and point/solver/shape_solver.py - Update tests and docs Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
1 parent 3a65af2 commit 3db51dd

7 files changed

Lines changed: 12 additions & 8 deletions

File tree

autolens/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@
8181
)
8282
from autogalaxy.profiles.light.linear import LightProfileLinearObjFuncList
8383
from autogalaxy.operate.image import OperateImage
84-
from autogalaxy.operate.deflections import OperateDeflections
84+
from autogalaxy.operate.lens_calc import LensCalc
8585
from autogalaxy.quantity.dataset_quantity import DatasetQuantity
8686
from autogalaxy import convert
8787

autolens/lens/tracer_util.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -354,7 +354,11 @@ def time_delays_from(
354354
D_dt_m = (1.0 + z_l) * (Dd_kpc * Ds_kpc / Dds_kpc) * kpc_in_m
355355

356356
# Fermat potential (should be in arcsec^2 for this formula)
357-
fermat_potential = galaxies.fermat_potential_from(grid=grid, xp=xp)
357+
import autogalaxy as ag
358+
359+
fermat_potential = ag.LensCalc.from_mass_obj(galaxies).fermat_potential_from(
360+
grid=grid, xp=xp
361+
)
358362

359363
# Final time delay in days
360364
return (D_dt_m / c) * fermat_potential * factor

autolens/point/fit/abstract.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ def magnifications_at_positions(self) -> aa.ArrayIrregular:
110110
if use_multi_plane
111111
else -1
112112
)
113-
od = ag.OperateDeflections.from_tracer(
113+
od = ag.LensCalc.from_tracer(
114114
tracer=self.tracer,
115115
use_multi_plane=use_multi_plane,
116116
plane_j=plane_j,

autolens/point/solver/shape_solver.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -274,7 +274,7 @@ def _filter_low_magnification(
274274
The points with an absolute magnification above the threshold.
275275
"""
276276
points = self._xp.array(points)
277-
magnifications = ag.OperateDeflections.from_mass_obj(
277+
magnifications = ag.LensCalc.from_mass_obj(
278278
tracer
279279
).magnification_2d_via_hessian_from(
280280
grid=aa.Grid2DIrregular(points).array, xp=self._xp

docs/api/source.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,4 +31,4 @@ Operators
3131
:recursive:
3232

3333
OperateImage
34-
OperateDeflections
34+
LensCalc

test_autolens/lens/test_operate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -194,7 +194,7 @@ def test__operate_lens__sums_individual_quantities():
194194
cosmology=al.cosmo.Planck15(),
195195
)
196196

197-
einstein_mass = ag.OperateDeflections.from_mass_obj(
197+
einstein_mass = ag.LensCalc.from_mass_obj(
198198
tracer
199199
).einstein_mass_angular_from(grid=grid)
200200

test_autolens/point/fit/test_abstract.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def test__magnifications_at_positions__multi_plane_calculation(gal_x1_mp):
2323
tracer=tracer,
2424
)
2525

26-
od_0 = ag.OperateDeflections.from_tracer(tracer, use_multi_plane=True, plane_j=1)
26+
od_0 = ag.LensCalc.from_tracer(tracer, use_multi_plane=True, plane_j=1)
2727
magnification_0 = abs(od_0.magnification_2d_via_hessian_from(grid=positions))
2828

2929
assert fit_0.magnifications_at_positions[0] == magnification_0
@@ -36,7 +36,7 @@ def test__magnifications_at_positions__multi_plane_calculation(gal_x1_mp):
3636
tracer=tracer,
3737
)
3838

39-
od_1 = ag.OperateDeflections.from_tracer(tracer, use_multi_plane=True, plane_j=2)
39+
od_1 = ag.LensCalc.from_tracer(tracer, use_multi_plane=True, plane_j=2)
4040
magnification_1 = abs(od_1.magnification_2d_via_hessian_from(grid=positions))
4141

4242
assert fit_1.magnifications_at_positions[0] == magnification_1

0 commit comments

Comments
 (0)