diff --git a/data/QUGS/convert.py b/data/QUGS/convert.py index 17af789..a04c6b0 100644 --- a/data/QUGS/convert.py +++ b/data/QUGS/convert.py @@ -61,9 +61,9 @@ def generate_df_tuples(total_dfs=30, group_size=5, prefix="zzzAD", extension="TX group = [] for i in range(1, 6): # TODO: shouldnt be hardcoded n = g * 5 + i - bottom = i # 1, 2, 3, 4, 5 - b = a + 25 # 26, 27, 28, 29, 30 # TODO: shouldnt be hardcoded - group.append((f"{prefix}{n}.{extension}", [a, b])) + bottom_end = i # 1, 2, 3, 4, 5 + top_end = bottom_end + 25 # 26, 27, 28, 29, 30 # TODO: shouldnt be hardcoded + group.append((f"{prefix}{n}.{extension}", [bottom_end, top_end])) result.append(group) # Add special groups at specified positions (other than beginning)