Skip to content

Commit

Permalink
Fix typos
Browse files Browse the repository at this point in the history
  • Loading branch information
gmatteo committed Jun 25, 2024
1 parent e137906 commit e5c95b2
Showing 1 changed file with 9 additions and 3 deletions.
12 changes: 9 additions & 3 deletions abipy/eph/gstore.py
Original file line number Diff line number Diff line change
Expand Up @@ -292,7 +292,10 @@ def get_g2q_interpolator_kpoint(self, kpoint, method="linear", check_mesh=1):

def get_g_qpt_kpt(self, qpoint, kpoint, what) -> np.ndarray:
"""
Return numpy array with the |g(k,q)|^2 for the given (qpoint, kpoint) pair.
Return numpy array with e-ph matrix elements the for the given (qpoint, kpoint) pair.
Args:
what="g2" for |g(k,q)|^2, "g" for g(k,q)
"""
# Find the internal indices of (qpoint, kpoint)
iq_g, qpoint = self.gstore.r.find_iq_glob_qpoint(qpoint, self.spin)
Expand All @@ -310,9 +313,12 @@ def get_g_qpt_kpt(self, qpoint, kpoint, what) -> np.ndarray:
def get_gdf_at_qpt_kpt(self, qpoint, kpoint, what="g2") -> pd.DataFrame:
"""
Build and return a dataframe with the |g(k,q)|^2 for the given (qpoint, kpoint) pair.
Args:
what="g2" for |g(k,q)|^2, "g" for g(k,q)
"""
g2_slice = self.get_g2_qpt_kpt(qpoint, kpoint, )
df = nparr_to_df("g2", g2_slice, ["imode", "m_kq", "n_k"])
g2_slice = self.get_g_qpt_kpt(qpoint, kpoint, what)
df = nparr_to_df(what, g2_slice, ["imode", "m_kq", "n_k"])
#df["m_kq"] += bstart_mkq
#df["n_k"] += bstart_nk

Expand Down

0 comments on commit e5c95b2

Please sign in to comment.