Skip to content

Commit a7ca9bb

Browse files
author
Maranville
committed
use type instead of string for dtype, making numba happy
1 parent 50a0b76 commit a7ca9bb

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

refl1d/reflectivity.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -574,19 +574,19 @@ def magnetic_amplitude_py(d, sigma, rho, irho,
574574
if (np.fabs(rhoM[0]) <= MINIMAL_RHO_M and np.fabs(rhoM[layers-1]) <= MINIMAL_RHO_M):
575575
# calculations for I+ and I- are the same in the fronting and backing.
576576
for i in prange(points):
577-
Y = np.empty(4, dtype='complex128')
577+
Y = np.empty(4, dtype=np.complex128)
578578
Cr4xa(layers, d, sigma, 1.0, rho, irho, rhoM, u1, u3, KZ[i], Y)
579579
Ra[i], Rb[i], Rc[i], Rd[i] = Y[0], Y[1], Y[2], Y[3]
580580
else:
581581
# plus polarization
582582
for i in prange(points):
583-
Y = np.empty(4, dtype='complex128')
583+
Y = np.empty(4, dtype=np.complex128)
584584
Cr4xa(layers, d, sigma, 1.0, rho, irho, rhoM, u1, u3, KZ[i], Y)
585585
Ra[i], Rb[i] = Y[0], Y[1]
586586

587587
# minus polarization
588588
for i in prange(points):
589-
Y = np.empty(4, dtype='complex128')
589+
Y = np.empty(4, dtype=np.complex128)
590590
Cr4xa(layers, d, sigma, -1.0, rho, irho, rhoM, u1, u3, KZ[i], Y)
591591
Rc[i], Rd[i] = Y[2], Y[3]
592592

0 commit comments

Comments
 (0)