multiplying with prefactor before vstack

Author: Niek Wielders

autogalaxy/profiles/mass/dark/nfw.py

@@ -81,13 +81,15 @@ def deflections_2d_via_analytic_from(
         deflection_x = (x1 * ((x1**2 - e_hk24**2) * (1 - e_hk24**2) + x2**2 * (1 + e_hk24**2)) * nfw_hk24_util.small_f_1(x1, x2, e_hk24, xp=xp)
                         + x1 * (x1**2 + x2**2 - e_hk24**2) * nfw_hk24_util.small_f_2(x1, x2, e_hk24, xp=xp)
                         - x2 * (x1**2 + x2**2 + e_hk24**2) * nfw_hk24_util.small_f_3(x1, x2, e_hk24, xp=xp))
+        deflection_x *= prefactor
 
         deflection_y = (x2 * (x1**2 * (1 - 2 * e_hk24**2) + x2**2 + e_hk24**2) * nfw_hk24_util.small_f_1(x1, x2, e_hk24, xp=xp)
                         + x2 * (x1**2 + x2**2 + e_hk24**2) * nfw_hk24_util.small_f_2(x1, x2, e_hk24, xp=xp)
                         + x1 * (x1**2 + x2**2 - e_hk24**2) * nfw_hk24_util.small_f_3(x1, x2, e_hk24, xp=xp))
+        deflection_y *= prefactor
 
         return self.rotated_grid_from_reference_frame_from(
-            grid=xp.multiply(self.scale_radius, xp.multiply(prefactor, xp.vstack((deflection_y, deflection_x)).T)),
+            xp.multiply(self.scale_radius, xp.vstack((deflection_y, deflection_x)).T),
             xp=xp,
             **kwargs,
         )