Add pathway id info in universe

PathwayMerger · May 14, 2019 · ef96f15 · ef96f15
1 parent 56f3491
commit ef96f15
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/pathme/export_utils.py b/src/pathme/export_utils.py
@@ -9,9 +9,6 @@
 import click
 import networkx as nx
 import pybel
-from pathme.constants import KEGG, PATHME_DIR, REACTOME, WIKIPATHWAYS
-from pathme.normalize_names import normalize_graph_names
-from pathme.pybel_utils import flatten_complex_nodes
 from pybel import BELGraph, union
 from pybel import from_pickle
 from pybel.constants import ANNOTATIONS, RELATION
@@ -20,6 +17,10 @@
 from pybel_tools.analysis.spia import bel_to_spia_matrices, spia_matrices_to_excel
 from tqdm import tqdm
 
+from pathme.constants import KEGG, PATHME_DIR, REACTOME, WIKIPATHWAYS
+from pathme.normalize_names import normalize_graph_names
+from pathme.pybel_utils import flatten_complex_nodes
+
 logger = logging.getLogger(__name__)
 
 
@@ -191,6 +192,9 @@ def _iterate_universe_graphs(
             logger.warning(f'Unknown pickle file: {file}')
             continue
 
+        graph.annotation_pattern['PathwayID'] = '.*'
+        add_annotation_value(graph, 'PathwayID', file.strip(".pickle"))
+
         yield graph