Commit dfafed3b authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

shut down parallel in instrumental correction

parent ba060b49

csst/msc/inst_corr.py

+2 −1
Original line number Original line Diff line number Diff line
@@ -3,6 +3,7 @@ from pathlib import Path 
import numpy as np

import numpy as np

from ccdproc import cosmicray_lacosmic

from ccdproc import cosmicray_lacosmic

from deepCR import deepCR

from deepCR import deepCR

from collections import OrderedDict





from ..core.processor import CsstProcessor, CsstProcStatus

from ..core.processor import CsstProcessor, CsstProcStatus

from ..msc import CsstMscImgData

from ..msc import CsstMscImgData
@@ -94,7 +95,7 @@ class CsstMscInstrumentProc(CsstProcessor): 
            inpaint_model = 'ACS-WFC-F606W-2-32'

            inpaint_model = 'ACS-WFC-F606W-2-32'

            model = deepCR(clean_model, inpaint_model, device='CPU', hidden=50)

            model = deepCR(clean_model, inpaint_model, device='CPU', hidden=50)

            masked, cleaned = model.clean(

            masked, cleaned = model.clean(

                self.__img, threshold=0.5, inpaint=True, segment=True, patch=256, parallel=True, n_jobs=self.n_jobs)

                self.__img, threshold=0.5, inpaint=True, segment=True, patch=256, parallel=False, n_jobs=self.n_jobs)

        else:

        else:

            cleaned, masked = cosmicray_lacosmic(ccd=self.__img,

            cleaned, masked = cosmicray_lacosmic(ccd=self.__img,

                                                 sigclip=3.,  # cr_threshold

                                                 sigclip=3.,  # cr_threshold