Apply suggestions from code review

Co-authored-by: Jamal Mustafa <[email protected]>
wannier-developers · May 30, 2024 · b025922 · b025922
1 parent 394cfe1
commit b025922
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git 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
@@ -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)