remove nan from power

mgpro/client.py

@@ -228,7 +228,6 @@ Syntax: powspec [w1, w2]
                 print('w1 should be less than w2')
             self.mg.power_fit(w1, w2)
             self.mg.plotpower()
-            self.mg.pcot = None
         else:
             print('Only two arguments can be accepted')
 

mgpro/mgpro.py

@@ -191,6 +191,8 @@ class mgmat(object):
         power = np.zeros_like(radfreq)
         for i, frq in enumerate(radfreq):
             power[i] = np.mean(sf_amp[np.where(omega == frq)])
+        power = power[np.where(np.logical_not(np.isnan(power)))]
+        radfreq = radfreq[np.where(np.logical_not(np.isnan(power)))]
         self.power = np.vstack((radfreq, np.log(power)))
     
     def power_fit(self, freqmin, freqmax):