diff --git a/csst_common/pipeline.py b/csst_common/pipeline.py index aec9333d07217176ffe951ac425700100e2f5775..bc17d47c028881286fbad3c1dc7c339b113f988a 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()