From f1466dac28558b17688949067bdd64d7b331ca00 Mon Sep 17 00:00:00 2001
From: bay9355 <mia.le@studium.uni-hamburg.de>
Date: Thu, 16 Mar 2023 22:51:24 +0100
Subject: [PATCH] redundant lines while saving the output in create_consensus

---
 cami_src/cami_suite.py | 17 ++++++-----------
 1 file changed, 6 insertions(+), 11 deletions(-)

diff --git a/cami_src/cami_suite.py b/cami_src/cami_suite.py
index a7df6f7..4825d62 100644
--- a/cami_src/cami_suite.py
+++ b/cami_src/cami_suite.py
@@ -342,7 +342,7 @@ class cami():
         # transform all vertex indices to their corresponding gene names in a result set
         for tool in result_sets:
             self.result_gene_sets[tool.name] = set([gene_name_map[vertex] for vertex in result_sets[tool]])
-                
+        
         # create integer codes for cami_versions (needed for predicted_by vertex property)
         recursion_limit = sys.getrecursionlimit()
         for cami_method_name, cami_params in camis.items():
@@ -374,20 +374,15 @@ class cami():
             # for visualization with nvenn
             self.result_gene_sets[cami_method_name] = set(cami_genes)
             
-            
-            # transform all vertex indices to their corresponding gene names in a result set
-            for tool in result_sets:
-                self.result_gene_sets[tool.name] = set([gene_name_map[vertex] for vertex in result_sets[tool]])
-                
-            # add seeds to result sets for drugstone and digest
-            for toolname in self.result_gene_sets:
-                self.result_module_sets[toolname] = self.result_gene_sets[toolname].union(set([gene_name_map[svertex] for svertex in self.seed_lst]))
-                print(f'With the {len(seed_genes)} seed genes the module predicted by {toolname} contains {len(self.result_module_sets[toolname])} genes')
-
             sys.setrecursionlimit(recursion_limit)
             # save the results in outputfiles
             self.generate_output(cami_method_name, seed_genes, cami_vlist, cami_vertices, putative_vertices, cami_genes,
                                  gene_name_map, codes2tools, cami_scores)
+            
+         # add seeds to result sets for drugstone and digest
+        for toolname in self.result_gene_sets:
+            self.result_module_sets[toolname] = self.result_gene_sets[toolname].union(set([gene_name_map[svertex] for svertex in self.seed_lst]))
+            print(f'With the {len(seed_genes)} seed genes the module predicted by {toolname} contains {len(self.result_module_sets[toolname])} genes')
 
     def generate_output(self, cami_method, seed_genes, cami_vlist, cami_vertices, putative_vertices, cami_genes,
                         gene_name_map, codes2tools, cami_scores):
-- 
GitLab