Skip to content

Commit

Permalink
Removing Levels Index Variable Within Atomic Data Preperation (#1957)
Browse files Browse the repository at this point in the history
* Removed Levels Index Filter

* Test
  • Loading branch information
bartnikm authored Apr 21, 2022
1 parent d8ab5f0 commit 507c50f
Showing 1 changed file with 0 additions and 16 deletions.
16 changes: 0 additions & 16 deletions tardis/io/atom_data/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -330,10 +330,6 @@ def prepare_atom_data(

self.nlte_species = nlte_species

self.levels_index = pd.Series(
np.arange(len(self.levels), dtype=int), index=self.levels.index
)

# cutting levels_lines
self.lines = self.lines[
self.lines.index.isin(
Expand All @@ -352,22 +348,10 @@ def prepare_atom_data(
"level_number_upper"
)

self.lines_lower2level_idx = (
self.levels_index.loc[tmp_lines_lower2level_idx]
.astype(np.int64)
.values
)

tmp_lines_upper2level_idx = self.lines.index.droplevel(
"level_number_lower"
)

self.lines_upper2level_idx = (
self.levels_index.loc[tmp_lines_upper2level_idx]
.astype(np.int64)
.values
)

if (
self.macro_atom_data_all is not None
and not line_interaction_type == "scatter"
Expand Down

0 comments on commit 507c50f

Please sign in to comment.