def load_all_windows()

in src/suggest_cls_streamlit.py [0:0]

• 15 lines of code
• 5 McCabe index (conditional complexity)

def load_all_windows():
    BASE_DIR = os.path.dirname(os.path.realpath(__file__))
    df_path_default = os.path.join(BASE_DIR, "..", "data", "suggest_datasets", "processed.csv")
    df_synthetic_path = os.path.join(BASE_DIR, "..", "data", "suggest_datasets", "synthetic_processed.csv")

    if os.path.exists(df_path_default):
        df = pd.read_csv(df_path_default)
    elif os.path.exists(df_synthetic_path):
        df = pd.read_csv(df_synthetic_path)
    else:
        # If neither CSV is found, return an empty list or handle it differently
        return []

    windows = []
    for test_set_id in df['test_set_id'].unique():
        subset = df[df['test_set_id'] == test_set_id]
        windows.append(generate_window_from_df(subset))
    return windows