remove positional args

MannLabs · May 22, 2024 · 29a8212 · 29a8212
1 parent 1ebcc44
commit 29a8212
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sparcscore/ml/utils.py b/src/sparcscore/ml/utils.py
@@ -105,7 +105,7 @@ def combine_datasets_balanced(list_of_datasets, class_labels, train_per_class, v
     return train_dataset, val_dataset, test_dataset
 
 
-def split_dataset_regression(list_of_datasets, train_size, test_size, val_size, fractions=None, seed=None):
+def split_dataset_regression(list_of_datasets, train_size=None, test_size=None, val_size=None, fractions=None, seed=None):
     """
     Split a dataset into train, test, and validation set.