Merge pull request #796 from lukeshingles/fix_pandas_warning

Prevent pandas SettingWithCopyWarning by copying DataFrames when slicing
tardis-sn · Oct 6, 2017 · 0a1a812 · 0a1a812
2 parents f7c77d1 + 65029ff
commit 0a1a812
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tardis/io/atomic.py b/tardis/io/atomic.py
@@ -324,13 +324,13 @@ def prepare_atom_data(
 
             self.macro_atom_data = self.macro_atom_data_all.loc[
                 self.macro_atom_data_all['atomic_number'].isin(self.selected_atomic_numbers)
-            ]
+            ].copy()
 
             self.macro_atom_references = self.macro_atom_references_all[
                 self.macro_atom_references_all.index.isin(
                     self.selected_atomic_numbers,
                     level='atomic_number')
-            ]
+            ].copy()
 
             if line_interaction_type == 'downbranch':
                 self.macro_atom_data = self.macro_atom_data.loc[