From b025922493a2e45fa48e6a42f959cdb1a5a54086 Mon Sep 17 00:00:00 2001 From: Giovanni Pizzi Date: Wed, 29 May 2024 09:17:18 +0200 Subject: [PATCH] Apply suggestions from code review Co-authored-by: Jamal Mustafa --- src/wannier90io/_u.py | 4 ++-- src/wannier90io/_unk.py | 1 - 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/src/wannier90io/_u.py b/src/wannier90io/_u.py index f92fa8d..76a3afb 100644 --- a/src/wannier90io/_u.py +++ b/src/wannier90io/_u.py @@ -12,7 +12,8 @@ def read_u(stream: typing.TextIO) -> tuple[np.ndarray, np.ndarray]: Read unitary matrix file (seedname_u.mat) or the rectangular U_dis matrix file (seedname_u_dis.mat). - Note: for the _u.mat file, num_bands == num_wann. + Note: + for the _u.mat file, num_bands == num_wann. Arguments: stream: a file-like stream @@ -21,7 +22,6 @@ def read_u(stream: typing.TextIO) -> tuple[np.ndarray, np.ndarray]: kpoint coordinates in fractional coordinates (num_kpts, 3) U matrix U(k) or U_dis(k) (num_kpts, num_bands, num_wann) - """ stream.readline() # header diff --git a/src/wannier90io/_unk.py b/src/wannier90io/_unk.py index ee02360..48da968 100644 --- a/src/wannier90io/_unk.py +++ b/src/wannier90io/_unk.py @@ -27,7 +27,6 @@ def read_unk_formatted(stream: typing.TextIO) -> tuple[int, np.ndarray]: k-point index ik (integer) complex wavefunction (ngx, ngy, ngz, Nb) - """ [ngx, ngy, ngz, ik, nbnd] = np.fromstring(stream.readline(), sep=' ', dtype=int)