fig bug: large mermory in the model FlatLED

observation_sim/mock_objects/FlatLED.py

@@ -75,7 +75,7 @@ class FlatLED(MockObject):
         ledflats = bandtoLed[self.chip.filter_type]
         iFlat = np.zeros([self.chip.npix_y, self.chip.npix_x])
         for nled in ledflats:
-            iFlat = iFlat + self.getLEDImage(led_type=nled, LED_Img_flag=False)
+            iFlat = iFlat + self.getLEDImage1(led_type=nled, LED_Img_flag=False)
         iFlat = iFlat/len(ledflats)
         return iFlat
 
@@ -123,7 +123,7 @@ class FlatLED(MockObject):
     # return LED flat, e/s
     ###
 
-    def getLEDImage1(self, led_type='LED1'):
+    def getLEDImage1(self, led_type='LED1', LED_Img_flag =True):
         # cwave = cwaves[led_type]
         flat = fits.open(os.path.join(self.flatDir, 'model_' +
                          cwaves_name[led_type] + 'nm.fits'))
@@ -155,7 +155,9 @@ class FlatLED(MockObject):
             N[0:self.chip.npix_y, 0:self.chip.npix_x]),
             method='linear')
         U = U/np.mean(U)
-        flatImage = U*fluxLED[led_type]*1000
+        flatImage = U
+        if LED_Img_flag:
+            flatImage = U*fluxLED[led_type]*1000
         gc.collect()
         return flatImage