add n_jobs and backend to dm.quickstart (eef7fc6b) · Commits · csst-pipeline / csst_common

csst_common/data_manager.py

+7 −3

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

	@staticmethod		@staticmethod
	def quickstart(ver_sim="C5.2", datatype="mbi", dir_l1=".", exposure_id=100,		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		Quick dataset generator for tests on dandelion or PML

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

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

	return CsstMsDataManager.from_dir(		return CsstMsDataManager.from_dir(
	ver_sim=ver_sim, datatype=datatype, dir_l0=dir_l0, dir_l1=dir_l1, path_aux=path_aux,		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):		def __repr__(self):
	lines = ""		lines = ""