types

daisybio · Oct 21, 2024 · bea20b7 · bea20b7
1 parent 92cbbe4
commit bea20b7
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/drevalpy/datasets/dataset.py b/drevalpy/datasets/dataset.py
@@ -250,12 +250,12 @@ def reduce_to(self, cell_line_ids: Optional[ArrayLike], drug_ids: Optional[Array
 
     def split_dataset(
         self,
-        n_cv_splits,
-        mode,
-        split_validation=True,
-        split_early_stopping=True,
-        validation_ratio=0.1,
-        random_state=42,
+        n_cv_splits: int,
+        mode: str,
+        split_validation: bool = True,
+        split_early_stopping: bool = True,
+        validation_ratio: float = 0.1,
+        random_state: int = 42,
     ) -> list[dict]:
         """
         Splits the dataset into training, validation and test sets for cross-validation.