Skip to content

Commit

Permalink
fixed flake8 issues
Browse files Browse the repository at this point in the history
  • Loading branch information
arthurmloureiro committed Jul 4, 2024
1 parent d9c33c5 commit 16ef71f
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion augur/tracers/two_point.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,6 +45,7 @@ def gaus_kernel(z, sig_z):
z_samp = np.linspace(-10*np.max(sig_z), 10*np.max(sig_z), len(z))
return 1./np.sqrt(2*np.pi*sig_z**2)*np.exp(-0.5*z_samp**2/sig_z**2)


def equal_density_zbins(z, nz, nbins):
"""
This function takes a redshift distribution and returns the redshift bins
Expand All @@ -67,6 +68,7 @@ def equal_density_zbins(z, nz, nbins):
zbinedges = np.interp(np.linspace(0, 1, nbins+1), cuml_nz, z)
return list(zip(zbinedges, zbinedges[1:]))


class ZDist(object):
"""
Base class for redshift distribution
Expand Down Expand Up @@ -122,6 +124,7 @@ def __init__(self, z, Nz_center, Nz_width,
self.Nz = convolve(dndz_bin, gaus_kernel(self.z, Nz_sigmaz*(1+self.z)))
self.zav = np.average(self.z, weights=self.Nz/np.sum(self.Nz))


class SourceSRD2018(ZDist):
"""
Source from 2018 SRD, benchmarked against Paul Rogozenski's notebook
Expand Down Expand Up @@ -154,7 +157,7 @@ def __init__(self, z, Nz_nbins, Nz_sigmaz, Nz_ibin,
# vectorises the erf function:

erf_vec = np.vectorize(math.erf, otypes=(float,))
sz = 2**0.5*Nz_sigmaz*(1+ self.z)
sz = 2 ** 0.5 * Nz_sigmaz * (1 + self.z)
binned_nz = erf_vec((z - z_low)/sz)
binned_nz -= erf_vec((z - z_high)/sz)
binned_nz /= 1 + erf_vec(z / sz)
Expand All @@ -164,6 +167,7 @@ def __init__(self, z, Nz_nbins, Nz_sigmaz, Nz_ibin,
self.Nz = binned_nz[Nz_ibin]
self.zav = np.average(self.z, weights=self.Nz/np.sum(self.Nz))


class TopHat(ZDist):
"""
Top Hat redshift bin
Expand Down

0 comments on commit 16ef71f

Please sign in to comment.