From 274cd60d27478070a3c2064ca7f980f912ad3620 Mon Sep 17 00:00:00 2001 From: nuluh Date: Mon, 11 Aug 2025 18:49:41 +0700 Subject: [PATCH] refactor(src): update generate_df_tuples function signature to include type hints for better clarity --- code/src/data_preprocessing.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/code/src/data_preprocessing.py b/code/src/data_preprocessing.py index 59db93e..340c706 100644 --- a/code/src/data_preprocessing.py +++ b/code/src/data_preprocessing.py @@ -35,8 +35,8 @@ def complement_pairs(n, prefix, extension): if a != orig_a: # skip original a yield (filename, [a, a + 25]) # use yield instead of return to return a generator of tuples -def generate_df_tuples(total_dfs, prefix, extension, first_col_start, last_col_offset, - group_size=5, special_groups=None, group=True): +def generate_df_tuples(prefix: str, total_dfs: int=30, extension: str="TXT", first_col_start: int=1, last_col_offset: int=25, + group_size: int=5, special_groups: list=None, group: bool=True): """ Generate a structured list of tuples containing DataFrame references and column indices.