update main method to allow for merge subcommand

raphael-group · Jan 7, 2025 · 5fcb75c · 5fcb75c
1 parent d016544
commit 5fcb75c
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/dialect/__main__.py b/src/dialect/__main__.py
@@ -7,6 +7,7 @@
     identify_pairwise_interactions,
     read_cbase_results_file,
     run_comparison_methods,
+    merge_pairwise_interaction_results,
 )
 
 
@@ -43,6 +44,14 @@ def main():
             k=args.top_k,
         )
 
+    elif args.command == "merge":
+        os.makedirs(args.out, exist_ok=True)
+        merge_pairwise_interaction_results(
+            dialect_results=args.dialect,
+            alt_results=args.alt,
+            out=args.out,
+        )
+
 
 if __name__ == "__main__":
     main()