Merge pull request #52 from chenyangkang/debug

njobs of fit function defaults to njob attribute of object

stemflow/model/AdaSTEM.py

@@ -568,14 +568,17 @@ def fit(
         y_train: Union[pd.core.frame.DataFrame, np.ndarray],
         verbosity: Union[None, int] = None,
         ax=None,
-        njobs: int = 1,
+        njobs: Union[None, int] = None,
     ):
         """Fitting method
 
         Args:
             X_train: Training variables
             y_train: Training target
             ax: matplotlib Axes to add to
+            verbosty: whether to show progress bar. 0 for no and 1 for yes.
+            ax: matplotlib ax for adding grid plot on that.
+            njobs: multiprocessing thread count. Default the njob of model object.
 
         Raises:
             TypeError: X_train is not a type of pd.core.frame.DataFrame