Merge pull request #79 from ebotiab/fix-analyzer-wrapper

Fix Presidio wrapper to call predict with def lang
microsoft · Sep 21, 2023 · 6745717 · 6745717
2 parents 66e2433 + 9b9f890
commit 6745717
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/presidio_evaluator/models/presidio_analyzer_wrapper.py b/presidio_evaluator/models/presidio_analyzer_wrapper.py
@@ -40,7 +40,7 @@ def predict(self, sample: InputSample) -> List[str]:
         results = self.analyzer_engine.analyze(
             text=sample.full_text,
             entities=self.entities,
-            language="en",
+            language=self.language,
             score_threshold=self.score_threshold,
         )
         starts = []