From b52e4bb05eba28f90e824efebad7d5d391a2081b Mon Sep 17 00:00:00 2001 From: BO ZHANG Date: Fri, 14 Jul 2023 21:55:52 +0800 Subject: [PATCH] set option for dfs and crds --- csst_common/pipeline.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/csst_common/pipeline.py b/csst_common/pipeline.py index aec9333..bc17d47 100644 --- a/csst_common/pipeline.py +++ b/csst_common/pipeline.py @@ -24,6 +24,8 @@ class Pipeline: n_jobs_gpu=9, device="CPU", filter_warnings=False, + dfs=True, + crds=False, ): self.dir_in = dir_in self.dir_out = dir_out @@ -53,8 +55,14 @@ class Pipeline: self.time_stamp = TimeStamp(os.path.join(self.dir_out, "time_stamp.txt")) self.exit_code = ExitCode(os.path.join(self.dir_out, "exit_code")) - self.dfs = DFS(n_try=5) - self.crds = CRDS() + if dfs: + self.dfs = DFS(n_try=5) + else: + self.dfs = None + if crds: + self.crds = CRDS() + else: + self.crds = None if filter_warnings: self.filter_warnings() -- GitLab