Loading csst_common/pipeline.py +15 −0 Original line number Diff line number Diff line import json import os.path import subprocess import warnings from astropy import time Loading @@ -22,6 +23,7 @@ class Pipeline: n_jobs_cpu=18, n_jobs_gpu=9, device="CPU", filter_warnings=False, ): self.dir_in = dir_in self.dir_out = dir_out Loading Loading @@ -54,6 +56,9 @@ class Pipeline: self.dfs = DFS(n_try=5) self.crds = CRDS() if filter_warnings: self.filter_warnings() def set_test_env(self): os.environ["CSST_DFS_API_MODE"] = "cluster" os.environ["CSST_DFS_GATEWAY"] = "172.24.27.2:30880" Loading @@ -71,6 +76,16 @@ class Pipeline: def now(self): return time.Time.now().isot @staticmethod def filter_warnings(): # Suppress all warnings warnings.filterwarnings("ignore") @staticmethod def reset_warnings(self): # Reset warning filters warnings.resetwarnings() class Message: def __init__(self, file_path=""): Loading Loading
csst_common/pipeline.py +15 −0 Original line number Diff line number Diff line import json import os.path import subprocess import warnings from astropy import time Loading @@ -22,6 +23,7 @@ class Pipeline: n_jobs_cpu=18, n_jobs_gpu=9, device="CPU", filter_warnings=False, ): self.dir_in = dir_in self.dir_out = dir_out Loading Loading @@ -54,6 +56,9 @@ class Pipeline: self.dfs = DFS(n_try=5) self.crds = CRDS() if filter_warnings: self.filter_warnings() def set_test_env(self): os.environ["CSST_DFS_API_MODE"] = "cluster" os.environ["CSST_DFS_GATEWAY"] = "172.24.27.2:30880" Loading @@ -71,6 +76,16 @@ class Pipeline: def now(self): return time.Time.now().isot @staticmethod def filter_warnings(): # Suppress all warnings warnings.filterwarnings("ignore") @staticmethod def reset_warnings(self): # Reset warning filters warnings.resetwarnings() class Message: def __init__(self, file_path=""): Loading