add n_jobs and backend to dm.quickstart (eef7fc6b) · Commits · Wu Maochun / csst_common

csst_common/data_manager.py

+7 −3

Original line number	Diff line number	Diff line
		@@ -509,7 +509,7 @@ class CsstMsDataManager:

		@staticmethod
		def quickstart(ver_sim="C5.2", datatype="mbi", dir_l1=".", exposure_id=100,
		use_dfs=False, dfs_node="kmust", clear_dir=False):
		use_dfs=False, dfs_node="kmust", clear_l1=False, n_jobs=18, backend="multiprocessing"):
		"""
		Quick dataset generator for tests on dandelion or PML

		@@ -527,8 +527,12 @@ class CsstMsDataManager:
		If True, use DFS.
		dfs_node : str
		The DFS node. Defaults to "kmust", could be "pml".
		clear_dir : bool
		clear_l1 : bool
		If True, clear dir_l1.
		n_jobs : int
		The number of jobs.
		backend : str
		The joblib backend.

		Returns
		-------
		@@ -564,7 +568,7 @@ class CsstMsDataManager:

		return CsstMsDataManager.from_dir(
		ver_sim=ver_sim, datatype=datatype, dir_l0=dir_l0, dir_l1=dir_l1, path_aux=path_aux,
		use_dfs=use_dfs, dfs_node=dfs_node)
		use_dfs=use_dfs, dfs_node=dfs_node, n_jobs=n_jobs, backend=backend)

		def __repr__(self):
		lines = ""