Minor fix, only normalise proj_xyz if normalise = True

SMTG-Bham · Jun 22, 2024 · 67181d9 · 67181d9
1 parent 63a9e52
commit 67181d9
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/easyunfold/procar.py b/easyunfold/procar.py
@@ -154,10 +154,10 @@ def _read(self, fobj, parsed_kpoints=None):
         if self.normalise:
             self.normalise_projs(proj_data)
 
-        if proj_xyz is not None:
-            proj_sum = np.sum(proj_xyz, axis=(-3, -2, -1), keepdims=True)
-            proj_sum[proj_sum == 0] = 1
-            proj_xyz /= proj_sum
+            if proj_xyz is not None:
+                proj_sum = np.sum(proj_xyz, axis=(-3, -2, -1), keepdims=True)
+                proj_sum[proj_sum == 0] = 1
+                proj_xyz /= proj_sum
 
         # Update the parsed kpoints
         parsed_kpoints.update({kvec_section_counter_tuple[0] for kvec_section_counter_tuple in this_procar_parsed_kpoints})