Merge pull request #46 from OxfordIonTrapGroup/positive_hfs_const

common: fix assignment of F in setB
OxfordIonTrapGroup · Aug 4, 2022 · b32491c · b32491c
2 parents f734498 + 683ca5b
commit b32491c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/atomic_physics/common.py b/atomic_physics/common.py
@@ -406,7 +406,7 @@ def setB(self, B: float):
 
  for M in set(self.M[lev]):
  for Fidx, idx in np.ndenumerate(np.where(M == self.M[lev])):
- self.F[lev][idx] = F_list[M <= F_list][Fidx[1]]
+ self.F[lev][idx] = F_list[abs(M) <= F_list][Fidx[1]]
 
  if self.M1 is not None:
  self.calc_M1()