Merge_Pred_Dicts.py

import argparse
import os
import Pipeline_Functions as pf

def main():
    parser = argparse.ArgumentParser(description="Merge replicate prediction dictionaries.")
    parser.add_argument("--pred_dict_folder", default="./data/raw_networks/", help="Folder containing raw networks (.dat)")
    parser.add_argument("--save_folder", default="./data/merged_networks/", help="Folder to save merged networks")
    parser.add_argument("--min_reps", type=int, default=1, help="Minimum number of replicates a pair must appear in to be kept")
    parser.add_argument("--save_name_add", default="_merged.dat", help="Suffix to add to the saved merged files")
    args = parser.parse_args()

    # Create save folder if it doesn't exist
    if not os.path.exists(args.save_folder):
        print(f"Creating missing output directory: {args.save_folder}")
        os.makedirs(args.save_folder, exist_ok=True)

    pf.merge_pred_dicts(
        pred_dict_folder=args.pred_dict_folder,
        save_folder=args.save_folder,
        min_reps=args.min_reps,
        save_name_add=args.save_name_add
    )

if __name__ == '__main__':
    main()