From 5bb3a2163fd32eb5e4f6626c0dbb45c3d3616b0c Mon Sep 17 00:00:00 2001 From: Deepak Date: Tue, 24 Nov 2020 12:51:03 -0600 Subject: [PATCH] adds more type hinting --- pysuperconductor/os_gap_calc.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/pysuperconductor/os_gap_calc.py b/pysuperconductor/os_gap_calc.py index bf886ec..944ef96 100644 --- a/pysuperconductor/os_gap_calc.py +++ b/pysuperconductor/os_gap_calc.py @@ -32,11 +32,16 @@ def gap_integral( 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) -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 sol = scipy.optimize.root( lambda d: gap_integral(temp, d, mu_star, debye_frequency) - nv_inv,