From ce58b871edf20079ab9958c2b97c4e15cc93f64b Mon Sep 17 00:00:00 2001 From: aecelaya Date: Fri, 13 Sep 2024 17:21:24 -0500 Subject: [PATCH] Add types to inputs of add_folds_to_df. --- mist/runtime/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mist/runtime/utils.py b/mist/runtime/utils.py index 92f2536..300af0c 100755 --- a/mist/runtime/utils.py +++ b/mist/runtime/utils.py @@ -253,7 +253,7 @@ def get_files_df(path_to_dataset_json: str, train_or_test: str) -> pd.DataFrame: return df -def add_folds_to_df(df, n_splits=5): +def add_folds_to_df(df: pd.DataFrame, n_splits: int=5): """Add folds to the dataframe for k-fold cross-validation. Args: