set option for dfs and crds (b52e4bb0) · Commits · csst-pipeline / csst_common

csst_common/pipeline.py

+10 −2

Original line number	Diff line number	Diff line
		@@ -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"))

		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()