Fix pylint complaint

SMTG-Bham · Jan 11, 2024 · 19b55ac · 19b55ac
1 parent a64a07d
commit 19b55ac
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/easyunfold/effective_mass.py b/easyunfold/effective_mass.py
@@ -144,8 +144,8 @@ def get_band_extrema(self, mode: str = 'cbm', extrema_tol: float = None, ispin=0
         cbm_indices = []
         for ik, wset in enumerate(weights):
             for isubset in range(wset.shape[1]):
-                etmp = weights[ik][ispin, isubset, :, 0]
-                wtmp = weights[ik][ispin, isubset, :, 1]
+                etmp = wset[ispin, isubset, :, 0]
+                wtmp = wset[ispin, isubset, :, 1]
                 # Filter by intensity
                 mask = wtmp > intensity_tol
                 midx = np.where(mask)[0]