Make enricher algorithm configurable

opensanctions · Oct 6, 2023 · ca3663d · ca3663d
1 parent 3a39fd4
commit ca3663d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/nomenklatura/enrich/yente.py b/nomenklatura/enrich/yente.py
@@ -25,6 +25,7 @@ def __init__(self, dataset: DS, cache: Cache, config: EnricherConfig):
         self._api: str = config.pop("api")
         self._dataset: str = config.pop("dataset", "default")
         self._threshold: Optional[float] = config.pop("threshold", None)
+        self._algorithm: Optional[float] = config.pop("algorithm", "best")
         self._nested: bool = config.pop("expand_nested", True)
         self._fuzzy: bool = config.pop("fuzzy", False)
         self._ns: Optional[Namespace] = None
@@ -48,7 +49,7 @@ def match(self, entity: CE) -> Generator[CE, None, None]:
         if not entity.schema.matchable:
             return
         url = urljoin(self._api, f"match/{self._dataset}")
-        params: Dict[str, Any] = {"fuzzy": self._fuzzy}
+        params: Dict[str, Any] = {"fuzzy": self._fuzzy, "algorithm": self._algorithm}
         if self._threshold is not None:
             params["threshold"] = self._threshold
         url = normalize_url(url, params)