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

add n_jobs and backend to dm.quickstart

parent e66ed791
......@@ -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 = ""
......
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