debug

csst_ifs_sim/csst_ifs_sim.py

@@ -3289,9 +3289,10 @@ class IFSsimulator():
         hdu_b.header['NAXIS1'] = (np.int32(1344), 'length of first data axis')
 
         hdu_b.header['NAXIS2'] = (np.int32(9672), 'length of second data axis')
-
-        hdu_b.header['NAXIS3'] = (
-            np.int32(self.information['exposuretimes']), 'length of third data axis')
+        
+        if self.information['exposuretimes']>1:
+            hdu_b.header['NAXIS3'] = (
+                np.int32(self.information['exposuretimes']), 'length of third data axis')
 
         if self.source == 'SCI' or self.source == 'COMP':
             hdu_b.header['EXTNAME'] = ('SCI', '')
@@ -3668,9 +3669,10 @@ class IFSsimulator():
 
         hdu_r.header['NAXIS2'] = (
             np.int32(13768), 'length of second data axis')
-
-        hdu_r.header['NAXIS3'] = (
-            np.int32(self.information['exposuretimes']), 'length of third data axis')
+        
+        if self.information['exposuretimes']>1:
+            hdu_r.header['NAXIS3'] = (
+                np.int32(self.information['exposuretimes']), 'length of third data axis')
 
         if self.source == 'SCI' or self.source == 'COMP':
             hdu_r.header['EXTNAME'] = ('SCI', '')