add noise

linetools · Mar 11, 2016 · 70ad923 · 70ad923
1 parent 1a22dbc
commit 70ad923
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/linetools/spectra/xspectrum1d.py b/linetools/spectra/xspectrum1d.py
@@ -362,12 +362,13 @@ def add_noise(self, seed=None, s2n=None):
 
         # Modify the flux
         if s2n is not None:
-            sig = 1. / s2n
+            sig = np.ones_like(self.flux) / s2n
         else:
             sig = self.sig.value
         # Copy
         newspec = self.copy()
         newspec.data['flux'][self.select] = self.flux.value + (rand * sig)
+        newspec.data['sig'][self.select] = sig
         #
         return newspec