exclude singletons, go not in term graph

BackofenLab · Oct 31, 2023 · d5e8ae6 · d5e8ae6
1 parent fb57357
commit d5e8ae6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/backend/src/main.py b/backend/src/main.py
@@ -100,9 +100,9 @@ def proteins_subgraph_api():
     stopwatch.round("Setup")
 
     if len(protein_ids) > 1:
-        _, source, target, score = queries.get_protein_associations(driver, protein_ids, threshold, species_id)
+        proteins, source, target, score = queries.get_protein_associations(driver, protein_ids, threshold, species_id)
     else:
-        _, source, target, score = queries.get_protein_neighbours(driver, protein_ids, threshold, species_id)
+        proteins, source, target, score = queries.get_protein_neighbours(driver, protein_ids, threshold, species_id)
 
     stopwatch.round("Neo4j")
 

diff --git a/backend/src/queries.py b/backend/src/queries.py
@@ -18,6 +18,8 @@ def get_terms_connected_by_overlap(driver: neo4j.Driver, term_ids: list[str], sp
         MATCH (source:FT:{species})-[association:OVERLAP]->(target:FT:{species})
         WHERE source.Term IN {term_ids}
             AND target.Term IN {term_ids}
+            AND NOT source.Category IN ["GOCC", "GOMF", "GOBP"]
+            AND NOT target.Category IN ["GOCC", "GOMF", "GOBP"]
         RETURN source, target, association.Score AS score;
         """
     with driver.session() as session: