Fix for remove_nans

Author: Kyle Kaplan

src/muler/echelle.py

@@ -599,7 +599,8 @@ def remove_nans(self):
         finite_spec : (KeckNIRSPECSpectrum)
             Spectrum with NaNs removed
         """
-        keep_indices = (self.mask == False) & (self.flux == self.flux)
+        #keep_indices = (self.mask == False) & (self.flux == self.flux)
+        keep_indicies = ~np.isnan(self.flux.value) & np.isinifite(self.flux.value)
         return self.apply_boolean_mask(keep_indices)
 
     def smooth_spectrum(self, size=50):

src/muler/igrins.py

@@ -748,8 +748,8 @@ def fullfitTellurics(self, verbose=True, plot=False, pdfobj=None, name=''):
                 #corrected_flux[order] = flux1d[order] / total_trans[order]
                 corrected_flux[order] = flux1d[order] / final_trans[order]
                 if order == 0:
-                    plt.plot(wave1d[order][100:1950], flux1d[order][100:1950], color='silver', label='Telluric Corrected Orders')
-                    plt.plot(wave1d[order][100:1950], corrected_flux[order][100:1950], color='black', label='Uncorrected Orders')
+                    plt.plot(wave1d[order][100:1950], flux1d[order][100:1950], color='silver', label='Uncorrected Orders')
+                    plt.plot(wave1d[order][100:1950], corrected_flux[order][100:1950], color='black', label='Telluric Corrected Orders')
                 else:
                     plt.plot(wave1d[order][100:1950], flux1d[order][100:1950], color='silver')
                     plt.plot(wave1d[order][100:1950], corrected_flux[order][100:1950], color='black')
@@ -981,8 +981,8 @@ def fitTellurics(self, verbose=True, plot=False, pdfobj=None, name=''):
                 corrected_flux[order] = flux1d[order] / final_trans[order]
                 smoothed_corrected_flux[order] = convolve(corrected_flux[order],g, normalize_kernel=False)
                 if order == 0:
-                    plt.plot(wave1d[order][100:1950], flux1d[order][100:1950], color='silver', label='Telluric Corrected Orders')
-                    plt.plot(wave1d[order][100:1950], corrected_flux[order][100:1950], color='black', label='Uncorrected Orders')
+                    plt.plot(wave1d[order][100:1950], flux1d[order][100:1950], color='silver', label='Uncorrected Orders')
+                    plt.plot(wave1d[order][100:1950], corrected_flux[order][100:1950], color='black', label='Telluric Corrected Orders')
                 else:
                     plt.plot(wave1d[order][100:1950], flux1d[order][100:1950], color='silver')
                     plt.plot(wave1d[order][100:1950], corrected_flux[order][100:1950], color='black')