Skip to content

Commit e43b63b

Browse files
author
Niek Wielders
committed
corrected w5 and w6
1 parent a5744ba commit e43b63b

1 file changed

Lines changed: 7 additions & 11 deletions

File tree

autogalaxy/profiles/mass/stellar/gaussian.py

Lines changed: 7 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -264,15 +264,13 @@ def wofz(self, z, xp=np):
264264
9022.228, 24322.84, 32066.6], dtype=xp.float64)
265265

266266
# Horner form in z^2
267-
num5 = U5[0]
268-
for k in range(1, 6):
267+
num5 = sqrt_pi
268+
for k in range(0, 6):
269269
num5 = num5 * z2 + U5[k]
270-
num5 = num5 * z2 + sqrt_pi
271270

272-
den5 = V5[0]
273-
for k in range(1, 7):
271+
den5 = 1.0
272+
for k in range(0, 7):
274273
den5 = den5 * z2 + V5[k]
275-
den5 = den5 * z2 + z2
276274

277275
w5 = xp.exp(-z2) + 1j * z * num5 / den5
278276

@@ -282,15 +280,13 @@ def wofz(self, z, xp=np):
282280
V6 = xp.array([10.479857, 53.992907, 170.35400,
283281
348.70392, 457.33448, 352.73063, 122.60793], dtype=xp.float64)
284282

285-
num6 = U6[0]
286-
for k in range(1, 6):
283+
num6 = sqrt_pi
284+
for k in range(0, 6):
287285
num6 = num6 * (-1j * z) + U6[k]
288-
num6 = num6 * (-1j * z) + sqrt_pi
289286

290-
den6 = V6[0]
287+
den6 = 1
291288
for k in range(1, 7):
292289
den6 = den6 * (-1j * z) + V6[k]
293-
den6 = den6 * (-1j * z) + (-1j * z)
294290

295291
w6 = num6 / den6
296292

0 commit comments

Comments
 (0)