From 86307759c70e78506f1011623f4a084c229d99c1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Valentin=20Grou=C3=A8s?= <valentin.groues@uni.lu>
Date: Tue, 19 Nov 2019 14:47:59 +0100
Subject: [PATCH] add missing limit parameter
---
dummy-service/dummyService.py | 3 ++-
jensenlab-service/jensenLabService.py | 2 +-
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/dummy-service/dummyService.py b/dummy-service/dummyService.py
index 7459b53..842baec 100644
--- a/dummy-service/dummyService.py
+++ b/dummy-service/dummyService.py
@@ -2,6 +2,7 @@ import logging
from typing import List
from interface.TextMiningService import TextMiningService, Publication
+from models.coocurrence import CoOccurrence
logger = logging.getLogger(__name__)
@@ -13,7 +14,7 @@ class DummyService(TextMiningService):
def get_mentions(self, entities: List, limit: int = 20) -> List[Publication]:
return [Publication(pm_id="00000" + str(i)) for i in range(20)]
- def get_co_occurrences(self, entity: str) -> List[str]:
+ def get_co_occurrences(self, entity: str, limit: int = 20) -> List[CoOccurrence]:
pass
diff --git a/jensenlab-service/jensenLabService.py b/jensenlab-service/jensenLabService.py
index c2b9b98..a3a77e0 100644
--- a/jensenlab-service/jensenLabService.py
+++ b/jensenlab-service/jensenLabService.py
@@ -29,7 +29,7 @@ class JensenLabService(TextMiningService):
publications_ids_intersection = set.intersection(*publications_ids)
return [Publication(pm_id=pid) for pid in publications_ids_intersection]
- def get_co_occurrences(self, entity: str) -> List[str]:
+ def get_co_occurrences(self, entity: str, limit: int = 20) -> List[str]:
pass
@staticmethod
--
GitLab