Merge pull request #116 from microsoft/omri/make_azure_sdk_optional

Author: omri374

presidio_evaluator/models/text_analytics_wrapper.py

@@ -1,9 +1,15 @@
+import warnings
 from typing import List, Optional, Dict
 
 from presidio_evaluator import InputSample, span_to_tag
 from presidio_evaluator.models import BaseModel
-from azure.ai.textanalytics import TextAnalyticsClient
-from azure.core.credentials import AzureKeyCredential
+
+try:
+    from azure.ai.textanalytics import TextAnalyticsClient
+    from azure.core.credentials import AzureKeyCredential
+except ImportError:
+    TextAnalyticsClient = None
+    AzureKeyCredential = None
 
 
 class TextAnalyticsWrapper(BaseModel):
@@ -31,11 +37,22 @@ def __init__(
             labeling_scheme=labeling_scheme,
             entity_mapping=entity_mapping,
         )
+
+        warnings.warn(
+            f"{self.__class__.__name__} is deprecated and will be removed in a future version."
+            f"Please use the TextAnalyticsRecognizer within Presidio Analyzer instead.",
+            DeprecationWarning,
+            stacklevel=2,
+        )
+
         self.score_threshold = score_threshold
         self.language = language
         self.ta_key = ta_key
         self.ta_endpoint = ta_endpoint
 
+        if not TextAnalyticsClient:
+            raise ImportError("azure.ai.textanalytics is not installed")
+
         if not ta_client:
             ta_client = self.__authenticate_client(ta_key, ta_endpoint)
         self.ta_client = ta_client

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "presidio_evaluator"
-version = "0.2.2"
+version = "0.2.3"
 description = ""
 authors = ["Microsoft"]
 readme = "README.md"
@@ -21,7 +21,6 @@ requests = "^2.25"
 xmltodict = "^0.12.0"
 python-dotenv = "^1.0.0"
 plotly= "^5.24.0"
-azure-ai-textanalytics = "^5.3.0"
 
 # optional dependencies for the different NLP approaches
 [tool.poetry.group.ner]
@@ -31,6 +30,7 @@ optional=true
 flair = "^0.14.0"
 spacy_stanza = "^1.0.0"
 spacy_huggingface_pipelines = "^0.0.4"
+azure-ai-textanalytics = "^5.3.0"
 
 
 [tool.poetry.group.dev.dependencies]