run black on code (#18887)

Azure · May 24, 2021 · 79f4c97 · 79f4c97
1 parent ff63cbf
commit 79f4c97
Show file tree

Hide file tree

Showing 4 changed files with 85 additions and 46 deletions.
diff --git a/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/_form_base_client.py b/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/_form_base_client.py
@@ -20,9 +20,7 @@ def __init__(self, endpoint, credential, **kwargs):
         # type: (str, Union[AzureKeyCredential, TokenCredential], Any) -> None
         self._endpoint = endpoint
         self._credential = credential
-        self._api_version = kwargs.pop(
-            "api_version", FormRecognizerApiVersion.V2_1
-        )
+        self._api_version = kwargs.pop("api_version", FormRecognizerApiVersion.V2_1)
         if self._api_version.startswith("v"):  # v2.0 released with this option
             self._api_version = self._api_version[1:]
         validate_api_version(self._api_version)
@@ -44,7 +42,15 @@ def __init__(self, endpoint, credential, **kwargs):
             }
         )
         http_logging_policy.allowed_query_params.update(
-            {"includeTextDetails", "locale", "language", "includeKeys", "op", "pages", "readingOrder"}
+            {
+                "includeTextDetails",
+                "locale",
+                "language",
+                "includeKeys",
+                "op",
+                "pages",
+                "readingOrder",
+            }
         )
 
         self._client = FormRecognizer(

diff --git a/...formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/_form_recognizer_client.py b/...formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/_form_recognizer_client.py
@@ -401,7 +401,9 @@ def begin_recognize_identity_documents(self, identity_document, **kwargs):
             raise e
 
     @distributed_trace
-    def begin_recognize_identity_documents_from_url(self, identity_document_url, **kwargs):
+    def begin_recognize_identity_documents_from_url(
+        self, identity_document_url, **kwargs
+    ):
         # type: (str, Any) -> LROPoller[List[RecognizedForm]]
         """Extract field text and semantic values from a given identity document.
         The input document must be the location (URL) of the identity document to be analyzed.

diff --git a/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/_models.py b/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/_models.py
@@ -242,7 +242,9 @@ def to_dict(self):
         """
         return {
             "text": self.text,
-            "bounding_box": [f.to_dict() for f in self.bounding_box] if self.bounding_box else [],
+            "bounding_box": [f.to_dict() for f in self.bounding_box]
+            if self.bounding_box
+            else [],
             "page_number": self.page_number,
             "kind": self.kind,
         }
@@ -346,7 +348,8 @@ def from_dict(cls, data):
         """
         return cls(
             fields={k: FormField.from_dict(v) for k, v in data.get("fields").items()}  # type: ignore
-            if data.get("fields") else {},
+            if data.get("fields")
+            else {},
             form_type=data.get("form_type", None),
             pages=[FormPage.from_dict(v) for v in data.get("pages")]  # type: ignore
             if len(data.get("pages", [])) > 0
@@ -560,9 +563,13 @@ def to_dict(self):
         """
         return {
             "text": self.text,
-            "bounding_box": [f.to_dict() for f in self.bounding_box] if self.bounding_box else [],
+            "bounding_box": [f.to_dict() for f in self.bounding_box]
+            if self.bounding_box
+            else [],
             "page_number": self.page_number,
-            "field_elements": [f.to_dict() for f in self.field_elements] if self.field_elements else []
+            "field_elements": [f.to_dict() for f in self.field_elements]
+            if self.field_elements
+            else [],
         }
 
     @classmethod
@@ -668,7 +675,9 @@ def to_dict(self):
             "unit": self.unit,
             "tables": [table.to_dict() for table in self.tables] if self.tables else [],
             "lines": [line.to_dict() for line in self.lines] if self.lines else [],
-            "selection_marks": [mark.to_dict() for mark in self.selection_marks] if self.selection_marks else []
+            "selection_marks": [mark.to_dict() for mark in self.selection_marks]
+            if self.selection_marks
+            else [],
         }
 
     @classmethod
@@ -764,11 +773,13 @@ def to_dict(self):
         """
         return {
             "text": self.text,
-            "bounding_box": [f.to_dict() for f in self.bounding_box] if self.bounding_box else [],
+            "bounding_box": [f.to_dict() for f in self.bounding_box]
+            if self.bounding_box
+            else [],
             "words": [f.to_dict() for f in self.words] if self.words else [],
             "page_number": self.page_number,
             "kind": self.kind,
-            "appearance": self.appearance.to_dict() if self.appearance else None
+            "appearance": self.appearance.to_dict() if self.appearance else None,
         }
 
     @classmethod
@@ -843,7 +854,9 @@ def to_dict(self):
         """
         return {
             "text": self.text,
-            "bounding_box": [f.to_dict() for f in self.bounding_box] if self.bounding_box else [],
+            "bounding_box": [f.to_dict() for f in self.bounding_box]
+            if self.bounding_box
+            else [],
             "confidence": self.confidence,
             "page_number": self.page_number,
             "kind": self.kind,
@@ -905,7 +918,12 @@ def _from_generated(cls, mark, page):
 
     def __repr__(self):
         return "FormSelectionMark(text={}, bounding_box={}, confidence={}, page_number={}, state={}, kind={})".format(
-            self.text, self.bounding_box, self.confidence, self.page_number, self.state, self.kind
+            self.text,
+            self.bounding_box,
+            self.confidence,
+            self.page_number,
+            self.state,
+            self.kind,
         )[
             :1024
         ]
@@ -919,7 +937,9 @@ def to_dict(self):
         """
         return {
             "text": self.text,
-            "bounding_box": [f.to_dict() for f in self.bounding_box] if self.bounding_box else [],
+            "bounding_box": [f.to_dict() for f in self.bounding_box]
+            if self.bounding_box
+            else [],
             "confidence": self.confidence,
             "state": self.state,
             "page_number": self.page_number,
@@ -997,7 +1017,9 @@ def to_dict(self):
             "row_count": self.row_count,
             "column_count": self.column_count,
             "cells": [cell.to_dict() for cell in self.cells],
-            "bounding_box": [box.to_dict() for box in self.bounding_box] if self.bounding_box else []
+            "bounding_box": [box.to_dict() for box in self.bounding_box]
+            if self.bounding_box
+            else [],
         }
 
     @classmethod
@@ -1124,9 +1146,12 @@ def to_dict(self):
             "is_header": self.is_header,
             "is_footer": self.is_footer,
             "page_number": self.page_number,
-            "bounding_box": [box.to_dict() for box in self.bounding_box] if self.bounding_box else [],
+            "bounding_box": [box.to_dict() for box in self.bounding_box]
+            if self.bounding_box
+            else [],
             "field_elements": [element.to_dict() for element in self.field_elements]
-            if self.field_elements else []
+            if self.field_elements
+            else [],
         }
 
     @classmethod
@@ -1279,11 +1304,15 @@ def to_dict(self):
             "status": self.status,
             "training_started_on": self.training_started_on,
             "training_completed_on": self.training_completed_on,
-            "submodels": [submodel.to_dict() for submodel in self.submodels] if self.submodels else [],
+            "submodels": [submodel.to_dict() for submodel in self.submodels]
+            if self.submodels
+            else [],
             "errors": [err.to_dict() for err in self.errors] if self.errors else [],
-            "training_documents": [doc.to_dict() for doc in self.training_documents] if self.training_documents else [],
+            "training_documents": [doc.to_dict() for doc in self.training_documents]
+            if self.training_documents
+            else [],
             "model_name": self.model_name,
-            "properties": self.properties.to_dict() if self.properties else None
+            "properties": self.properties.to_dict() if self.properties else None,
         }
 
     @classmethod
@@ -1421,8 +1450,10 @@ def to_dict(self):
         return {
             "model_id": self.model_id,
             "accuracy": self.accuracy,
-            "fields": {k: v.to_dict() for k, v in self.fields.items()} if self.fields else {},
-            "form_type": self.form_type
+            "fields": {k: v.to_dict() for k, v in self.fields.items()}
+            if self.fields
+            else {},
+            "form_type": self.form_type,
         }
 
     @classmethod
@@ -1438,7 +1469,8 @@ def from_dict(cls, data):
             model_id=data.get("model_id", None),
             accuracy=data.get("accuracy", None),
             fields={k: CustomFormModelField.from_dict(v) for k, v in data.get("fields").items()}  # type: ignore
-            if data.get("fields") else {},
+            if data.get("fields")
+            else {},
             form_type=data.get("form_type", None),
         )
 
@@ -1485,11 +1517,7 @@ def to_dict(self):
         :return: dict
         :rtype: dict
         """
-        return {
-            "label": self.label,
-            "accuracy": self.accuracy,
-            "name": self.name
-        }
+        return {"label": self.label, "accuracy": self.accuracy, "name": self.name}
 
     @classmethod
     def from_dict(cls, data):
@@ -1589,7 +1617,7 @@ def to_dict(self):
             "status": self.status,
             "page_count": self.page_count,
             "errors": [err.to_dict() for err in self.errors] if self.errors else [],
-            "model_id": self.model_id
+            "model_id": self.model_id,
         }
 
     @classmethod
@@ -1646,10 +1674,7 @@ def to_dict(self):
         :return: dict
         :rtype: dict
         """
-        return {
-            "code": self.code,
-            "message": self.message
-        }
+        return {"code": self.code, "message": self.message}
 
     @classmethod
     def from_dict(cls, data):
@@ -1743,7 +1768,7 @@ def to_dict(self):
             "training_started_on": self.training_started_on,
             "training_completed_on": self.training_completed_on,
             "model_name": self.model_name,
-            "properties": self.properties.to_dict() if self.properties else None
+            "properties": self.properties.to_dict() if self.properties else None,
         }
 
     @classmethod
@@ -1802,7 +1827,7 @@ def to_dict(self):
         """
         return {
             "custom_model_count": self.custom_model_count,
-            "custom_model_limit": self.custom_model_limit
+            "custom_model_limit": self.custom_model_limit,
         }
 
     @classmethod
@@ -1850,9 +1875,7 @@ def to_dict(self):
         :return: dict
         :rtype: dict
         """
-        return {
-            "is_composed_model": self.is_composed_model
-        }
+        return {"is_composed_model": self.is_composed_model}
 
     @classmethod
     def from_dict(cls, data):
@@ -1889,11 +1912,13 @@ def _from_generated(cls, appearance):
             return appearance
         return cls(
             style_name=appearance.style.name,
-            style_confidence=appearance.style.confidence
+            style_confidence=appearance.style.confidence,
         )
 
     def __repr__(self):
-        return "TextAppearance(style_name={}, style_confidence={})".format(self.style_name, self.style_confidence)
+        return "TextAppearance(style_name={}, style_confidence={})".format(
+            self.style_name, self.style_confidence
+        )
 
     def to_dict(self):
         # type: () -> dict
@@ -1904,7 +1929,7 @@ def to_dict(self):
         """
         return {
             "style_name": self.style_name,
-            "style_confidence": self.style_confidence
+            "style_confidence": self.style_confidence,
         }
 
     @classmethod

diff --git a/...recognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/aio/_form_base_client_async.py b/...recognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/aio/_form_base_client_async.py
@@ -31,9 +31,7 @@ def __init__(
     ) -> None:
         self._endpoint = endpoint
         self._credential = credential
-        self._api_version = kwargs.pop(
-            "api_version", FormRecognizerApiVersion.V2_1
-        )
+        self._api_version = kwargs.pop("api_version", FormRecognizerApiVersion.V2_1)
         if self._api_version.startswith("v"):  # v2.0 released with this option
             self._api_version = self._api_version[1:]
         validate_api_version(self._api_version)
@@ -55,7 +53,15 @@ def __init__(
             }
         )
         http_logging_policy.allowed_query_params.update(
-            {"includeTextDetails", "locale", "language", "includeKeys", "op", "pages", "readingOrder"}
+            {
+                "includeTextDetails",
+                "locale",
+                "language",
+                "includeKeys",
+                "op",
+                "pages",
+                "readingOrder",
+            }
         )
         self._client = FormRecognizer(
             endpoint=endpoint,