Merge pull request #273 from Jammy2211/jax_hot_fix

NiekWielders · web-flow · commit 06449828aa92 · 2026-02-06T12:33:14.000Z
Jax hot fix
diff --git a/autogalaxy/abstract_fit.py b/autogalaxy/abstract_fit.py
@@ -110,7 +110,7 @@ def linear_light_profile_intensity_dict(
             reconstruction = self.inversion.reconstruction_dict[linear_obj_func]
 
             for i, light_profile in enumerate(linear_obj_func.light_profile_list):
-                linear_light_profile_intensity_dict[light_profile] = reconstruction[i]
+                linear_light_profile_intensity_dict[light_profile] = float(reconstruction[i])
 
         return linear_light_profile_intensity_dict
 
diff --git a/autogalaxy/profiles/basis.py b/autogalaxy/profiles/basis.py
@@ -215,7 +215,7 @@ def deflections_yx_2d_from(
         """
         if len(self.mass_profile_list) > 0:
             return sum(
-                [mass.deflections_yx_2d_from(grid=grid) for mass in self.profile_list]
+                [mass.deflections_yx_2d_from(grid=grid, xp=xp) for mass in self.profile_list]
             )
         return xp.zeros((grid.shape[0], 2))
 

Original file line number	Diff line number	Diff line change
`@@ -215,7 +215,7 @@ def deflections_yx_2d_from(`
`215`	`215`	`"""`
`216`	`216`	`if len(self.mass_profile_list) > 0:`
`217`	`217`	`return sum(`
`218`		`- [mass.deflections_yx_2d_from(grid=grid) for mass in self.profile_list]`
	`218`	`+ [mass.deflections_yx_2d_from(grid=grid, xp=xp) for mass in self.profile_list]`
`219`	`219`	`)`
`220`	`220`	`return xp.zeros((grid.shape[0], 2))`
`221`	`221`