From 53e539875a579b6f1d5b995c92e6f20a7603ff08 Mon Sep 17 00:00:00 2001
From: Carlos Vega <carlos.vega@uni.lu>
Date: Tue, 19 Nov 2019 12:23:53 +0100
Subject: [PATCH] added limit to get cooccurrences
---
biokb/biokb.py | 7 +++----
1 file changed, 3 insertions(+), 4 deletions(-)
diff --git a/biokb/biokb.py b/biokb/biokb.py
index 31a7429..0f2ac0e 100644
--- a/biokb/biokb.py
+++ b/biokb/biokb.py
@@ -60,7 +60,7 @@ class BioKBservice(TextMiningService):
return values
- def get_co_occurrences(self, entity: str) -> List[CoOccurrence]:
+ def get_co_occurrences(self, entity: str, limit: int = 20) -> List[CoOccurrence]:
entity = standarise_underscored_entity_code(entity)
query = """
select * where {
@@ -89,9 +89,8 @@ class BioKBservice(TextMiningService):
GROUP BY ?other_entity
- } ORDER BY DESC(?count)
- """.replace('%ENTITY', entity)
-
+ } ORDER BY DESC(?count) LIMIT %LIMIT%
+ """.replace('%ENTITY%', entity).replace('%LIMIT%', str(limit))
results = self._run_sparql_query(query)
values = []
values = []
--
GitLab