update setup.py

csst_ifs_sim/csst_ifs_sim.py

@@ -1267,7 +1267,7 @@ class IFSsimulator():
         # self.log.info('Reading in cosmic ray information from %s and %s' % (self.information['cosmicraylengths'],
         #                                                                     self.information['cosmicraydistance']))
         print(self.information)
-        print(self.dir_path)
+        
         
         crLengths = np.loadtxt(
             self.information['dir_path']+self.information['cosmicraylengths']) 
@@ -1396,10 +1396,7 @@ class IFSsimulator():
         # else:
         #     self.result_path='/data/ifspip/CCD_ima/IFS_simData_'+self.source+ss+result_day
 
-        dir_path = os.path.join(os.environ['UNIT_TEST_DATA_ROOT'],'ifs_sim/')
-        print(dir_path)
-        
-        self.result_path = dir_path+'/ifs_sim_result/'+self.source+ss+result_day
+        self.result_path = self.information['dir_path']+'ifs_sim_result/'+self.source+ss+result_day
 
         if os.path.isdir(self.result_path) == False:
             os.mkdir(self.result_path)
@@ -4730,7 +4727,7 @@ class IFSsimulator():
 
         hdu2 = fits.ImageHDU(Wave)
 
-        hdu2.header.append(('WaveUnit', 'nm'))
+        hdu2.header.append(('WaveUnit', 'nm'))  
 
         newd = fits.HDUList([hdu1, hdu2])