added poetry + fixed failing tests

.gitignore

@@ -189,4 +189,5 @@ datasets/
 /data
 
 *.spacy
-*.pickle
+*.pickle
+/poetry.lock

README.md

@@ -36,8 +36,10 @@ conda create --name presidio python=3.9
 conda activate presidio
 
 # Install package+dependencies
-pip install -r requirements.txt
-python setup.py install
+pip install poetry
+poetry install
+# To install with all additional NER dependencies (e.g. Flair, Stanza, CRF), run:
+# poetry install --with ner 
 
 # Download a spaCy model used by presidio-analyzer
 python -m spacy download en_core_web_lg

azure-pipelines.yml

@@ -0,0 +1,35 @@
+pr:
+  branches:
+    include:
+    - master
+    - feature/*
+
+pool:
+  vmImage: 'ubuntu-latest'
+strategy:
+  matrix:
+    Python38:
+      python.version: '3.8'
+    Python39:
+      python.version: '3.9'
+    Python310:
+      python.version: '3.10'
+    Python311:
+      python.version: '3.11'
+steps:
+- task: UsePythonVersion@0
+  inputs:
+    versionSpec: '$(python.version)'
+  displayName: 'Use Python $(python.version)'
+
+- script: |
+    python -m pip install --upgrade pip
+    pip install poetry
+    poetry install --with dev,ner
+
+  displayName: 'Install dependencies'
+
+- script: |
+    poetry add pytest-azurepipelines
+    poetry run pytest --runslow
+  displayName: 'pytest'

presidio_evaluator/models/presidio_recognizer_wrapper.py

@@ -41,12 +41,15 @@ def __init__(
         self.recognizer = recognizer
         self.nlp_engine = nlp_engine
 
+        if not self.nlp_engine.is_loaded():
+            self.nlp_engine.load()
+
     #
     def __make_nlp_artifacts(self, text: str):
         return self.nlp_engine.process_text(text, "en")
 
     #
-    def predict(self, sample: InputSample) -> List[str]:
+    def predict(self, sample: InputSample, **kwargs) -> List[str]:
         nlp_artifacts = None
         if self.with_nlp_artifacts:
             nlp_artifacts = self.__make_nlp_artifacts(sample.full_text)

pyproject.toml

@@ -0,0 +1,48 @@
+[tool.poetry]
+name = "presidio_evaluator"
+version = "0.1.0"
+description = ""
+authors = ["Omri Mendels <[email protected]>"]
+readme = "README.md"
+
+[tool.poetry.dependencies]
+python = "^3.9"
+spacy = ">=3.2.0, <4.0.0"
+numpy = ">=1.20.2,<2.0.0"
+jupyter = ">=1"
+pandas = ">=1.2.4,<2.0.0"
+tqdm = ">=4.60.0,<5.0.0"
+haikunator = ">=2.1.0,<3.0.0"
+schwifty = ">=2023.11.2,<2024.0.0"
+faker = ">=9.6.0,<10.0.0"
+scikit-learn = ">1.3.2,<2.0.0"
+pytest = ">=6.2.3"
+presidio-analyzer = "^2.2.351"
+presidio-anonymizer = "^2.2.351"
+requests = ">=2.25.1"
+xmltodict = ">=0.12.0"
+python-dotenv = "^1.0.0"
+plotly = "^5.18.0"
+azure-ai-textanalytics = ">=5.3.0"
+en_core_web_sm = {url = "https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.7.1/en_core_web_sm-3.7.1.tar.gz"}
+en_core_web_lg = {url = "https://github.com/explosion/spacy-models/releases/download/en_core_web_lg-3.7.1/en_core_web_lg-3.7.1.tar.gz"}
+
+# optional dependencies for the different NLP approaches
+[tool.poetry.group.ner]
+optional=true
+
+[tool.poetry.group.ner.dependencies]
+flair = "^0.13.0"
+spacy_stanza = "^1.0.0"
+sklearn_crfsuite = "^0.3.6"
+spacy_huggingface_pipelines = "^0.0.4"
+
+
+[tool.poetry.group.dev.dependencies]
+pytest = ">=6.*"
+flake8 = ">=3.*"
+pytest-azurepipelines = "^1.0.5"
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"