Skip to content

Commit 39b5df2

Browse files
committed
fix
1 parent 1b73388 commit 39b5df2

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

camb/results.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1634,7 +1634,7 @@ def luminosity_distance(self, z: float) -> float: ...
16341634
@overload
16351635
def luminosity_distance(self, z: Array1D) -> np.ndarray: ...
16361636

1637-
def luminosity_distance(self, z):
1637+
def luminosity_distance(self, z: NumberOrArray1D):
16381638
"""
16391639
Get luminosity distance from to redshift z.
16401640
@@ -1644,8 +1644,9 @@ def luminosity_distance(self, z):
16441644
:param z: redshift or array of redshifts
16451645
:return: luminosity distance (matches rank of z)
16461646
"""
1647-
1648-
return self.angular_diameter_distance(z) * (1.0 + z) ** 2
1647+
if not np.isscalar(z):
1648+
z = np.ascontiguousarray(z, dtype=np.float64)
1649+
return self.angular_diameter_distance(z) * (1.0 + z) ** 2 # type: ignore
16491650

16501651
@overload
16511652
def h_of_z(self, z: float) -> float: ...

0 commit comments

Comments
 (0)