update docstring

herberteuler · Nov 28, 2022 · 32ff81e · 32ff81e
1 parent 9b46389
commit 32ff81e
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/presidio_evaluator/models/text_analytics_wrapper.py b/presidio_evaluator/models/text_analytics_wrapper.py
@@ -20,8 +20,12 @@ def __init__(
         entity_mapping: Optional[Dict[str, str]] = None,
     ):
         """
-        Evaluation wrapper for the Presidio Analyzer
-        :param ta_client: object of type TextAnalyticsClient (from presidio-analyzer)
+        Evaluation wrapper for the Azure Text Analytics
+        :param ta_client: object of type TextAnalyticsClient
+        :param ta_key: Azure cognitive Services for Language key
+        :param ta_endpoint: Azure cognitive Services for Language endpoint
+        :param entities_to_keep: List of entities to predict on
+        :param entity_mapping: Mapping between input dataset entities and entities expected by Azure cognitive Services for Language 
         """
         super().__init__(
             entities_to_keep=entities_to_keep,