adds more type hinting

This commit is contained in:
Deepak Mallubhotla 2020-11-24 12:51:03 -06:00
parent 7744f040f6
commit 5bb3a2163f
Signed by: deepak
GPG Key ID: 64BF53A3369104E7

View File

@ -32,11 +32,16 @@ def gap_integral(
return integrate.quad(integrand, 0, debye_frequency)[0] return integrate.quad(integrand, 0, debye_frequency)[0]
def equilibrium_gap(temp, debye_frequency, nv): def equilibrium_gap(temp: float, debye_frequency: float, nv: float) -> float:
return find_gap(temp, 0, debye_frequency, nv) return find_gap(temp, 0, debye_frequency, nv)
def find_gap(temp, mu_star, debye_frequency, nv): def find_gap(
temp: float,
mu_star: float,
debye_frequency: float,
nv: float
) -> float:
nv_inv = 1 / nv nv_inv = 1 / nv
sol = scipy.optimize.root( sol = scipy.optimize.root(
lambda d: gap_integral(temp, d, mu_star, debye_frequency) - nv_inv, lambda d: gap_integral(temp, d, mu_star, debye_frequency) - nv_inv,