added gravity to plot save

BDNYC · Jul 28, 2017 · 055044a · 055044a
1 parent 5593f71
commit 055044a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/TypeFinder.py b/TypeFinder.py
@@ -227,7 +227,7 @@ def bracketed_plot(self, gravity_type): #This is for bracketed plots that are no
                 axes2[jj, 3].plot(self.wavelength, self.norm_flux, c='k')
                 axes2[jj, 3].axis('off')
 
-        save_name = 'new_types/' + self.file_name[12:-5] + r'_L{}'.format(self.type_number)
+        save_name = 'new_types/' + self.file_name[12:-5] + r'_L{}{}'.format(self.type_number, gravity_type)
 
         def save_figure(event):
             if event.key == 'w':
@@ -334,7 +334,7 @@ def first_bracketed_plot(self, gravity_type):
                 axes2[jj, 3].plot(self.wavelength, self.norm_flux, c='k')
                 axes2[jj, 3].axis('off')
 
-        save_name = 'new_types/' + self.file_name[12:-5] + r'_L{}'.format(self.type_number)
+        save_name = 'new_types/' + self.file_name[12:-5] + r'_L{}{}'.format(self.type_number, gravity_type)
 
         def save_figure(event):
             if event.key == 'w':
@@ -441,7 +441,7 @@ def last_bracketed_plot(self, gravity_type):
         axes2[2, 3].plot(self.wavelength, self.norm_flux, c='k')
         axes2[2, 3].axis('off')
 
-        save_name = 'new_types/' + self.file_name[12:-5] + r'_L{}'.format(self.type_number)
+        save_name = 'new_types/' + self.file_name[12:-5] + r'_L{}{}'.format(self.type_number, gravity_type)
 
         def save_figure(event):
             if event.key == 'w':