correct float param constant value format

GCS-ZHN · Oct 7, 2022 · 5deb279 · 5deb279
1 parent 65a43cb
commit 5deb279
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/socube/task/doublet/data.py b/src/socube/task/doublet/data.py
@@ -246,8 +246,8 @@ def _check(x: pd.Series):
 @parallel
 def createTrainData(samples: pd.DataFrame,
                     output_path: str,
-                    ratio: float = 1,
-                    adj: float = 1,
+                    ratio: float = 1.0,
+                    adj: float = 1.0,
                     seed: Optional[int] = None,
                     mode: Optional[int] = "balance") -> Tuple[NDFrame]:
     """

diff --git a/src/socube/task/doublet/train.py b/src/socube/task/doublet/train.py
@@ -66,7 +66,7 @@ def fit(home_dir: str,
         step: int = 5,
         model_id: str = None,
         pretrain_model_path: str = None,
-        max_acc_limit: float = 1,
+        max_acc_limit: float = 1.0,
         multi_process: bool = False,
         **kwargs) -> str:
     """