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

set option for dfs and crds

parent 7ec00c4f
......@@ -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()
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment