Skip to content

Commit

Permalink
fix pandas changes?
Browse files Browse the repository at this point in the history
  • Loading branch information
AWehrhahn committed Jul 23, 2021
1 parent fd06d44 commit dfeee05
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion src/pysme/nlte.py
Original file line number Diff line number Diff line change
Expand Up @@ -531,6 +531,8 @@ def select_levels(self, conf, term, species, rotnum):
low = self.linelist["term_lower"][lineindices]
upp = self.linelist["term_upper"][lineindices]
# Remove quotation marks (if any are there)
low = low.astype(str)
upp = upp.astype(str)
low = np.char.replace(low, "'", "")
upp = np.char.replace(upp, "'", "")
# Get only the relevant part
Expand Down Expand Up @@ -984,7 +986,7 @@ def update_coefficients(self, sme, dll, lfs_nlte):
# loop through the list of relevant _lines_, substitute both their levels into the main b matrix
# Make sure both levels have corrections available
if lr[0] != -1 and lr[1] != -1:
dll.InputNLTE(bmat[:, lr], li)
dll.InputNLTE(bmat[:, lr].T, li)

# flags = sme_synth.GetNLTEflags(sme.linelist)

Expand Down

0 comments on commit dfeee05

Please sign in to comment.