Merge pull request #58 from antelmor/master

Normalization correction for merging algorithm
mailhexu · Jul 4, 2024 · 7ecb5c6 · 7ecb5c6
2 parents 7b51ae7 + f2e9074
commit 7ecb5c6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/TB2J/io_merge.py b/TB2J/io_merge.py
@@ -48,7 +48,8 @@ def __init__(self, *args, **kwargs):
 
     def _set_projection_vectors(self):
 
-        spinat = self.spinat
+        norm = np.linalg.norm(self.spinat, axis=-1).reshape(-1, 1)
+        spinat = self.spinat / norm
         idx = [self.ind_atoms[i] for i in self.index_spin if i >= 0]
         projv = {}
         for i, j in combinations_with_replacement(range(self.nspin), 2):