♻️ misc fixes for newest additions

Author: sebastianMindee

.pylintrc

@@ -418,7 +418,7 @@ attr-naming-style=snake_case
 # Regular expression matching correct attribute names. Overrides attr-naming-
 # style. If left empty, attribute names will be checked with the set naming
 # style.
-attr-rgx=[a-z_][a-z0-9_]{1,30}$
+attr-rgx=[a-z_][a-z0-9_]{1,40}$
 
 # Bad variable names which should always be refused, separated by a comma.
 bad-names=foo,

mindee/cli.py

@@ -79,6 +79,10 @@ class CommandConfig(Generic[TypeDoc]):
         help="FR Bank Account Details",
         doc_class=documents.fr.TypeBankAccountDetailsV1,
     ),
+    "multi-receipts-detector": CommandConfig(
+        help="Multi Receipts Detector",
+        doc_class=documents.TypeMultiReceiptsDetectorV1,
+    ),
     "invoice-splitter": CommandConfig(
         help="Invoice Splitter",
         doc_class=documents.TypeInvoiceSplitterV1,

mindee/documents/multi_receipts_detector/__init__.py

@@ -1 +1,4 @@
-from .multi_receipts_detector_v1 import MultiReceiptsDetectorV1, TypeMultiReceiptsDetectorV1
+from .multi_receipts_detector_v1 import (
+    MultiReceiptsDetectorV1,
+    TypeMultiReceiptsDetectorV1,
+)

mindee/documents/us/w9/w9_v1.py

@@ -28,7 +28,7 @@ class W9V1(Document):
     """The federal tax classification, which can vary depending on the revision date."""
     tax_classification_llc: TextField
     """Depending on revision year, among S, C, P or D for Limited Liability Company Classification."""
-    tax_classif_other_details: TextField
+    tax_classification_other_details: TextField
     """Tax Classification Other Details."""
     w9_revision_date: TextField
     """The Revision month and year of the W9 form."""
@@ -103,7 +103,7 @@ def _build_from_api_prediction(
             api_prediction.get("tax_classification_llc", {}),
             page_id=page_n,
         )
-        self.tax_classif_other_details = TextField(
+        self.tax_classification_other_details = TextField(
             api_prediction.get("tax_classification_other_details", {}),
             page_id=page_n,
         )
@@ -124,7 +124,7 @@ def __str__(self) -> str:
             f":Business Name: {self.business_name}\n"
             f":EIN: {self.ein}\n"
             f":Tax Classification: {self.tax_classification}\n"
-            f":Tax Classification Other Details: {self.tax_classif_other_details}\n"
+            f":Tax Classification Other Details: {self.tax_classification_other_details}\n"
             f":W9 Revision Date: {self.w9_revision_date}\n"
             f":Signature Position: {self.signature_position}\n"
             f":Signature Date Position: {self.signature_date_position}\n"

tests/documents/test_multi_receipts_detector_v1.py

@@ -5,26 +5,38 @@
 from mindee.documents import MultiReceiptsDetectorV1
 
 MULTI_RECEIPTS_DETECTOR_DATA_DIR = "./tests/data/products/multi_receipts_detector"
-FILE_PATH_MULTI_RECEIPTS_DETECTOR_V1_COMPLETE = f"{ MULTI_RECEIPTS_DETECTOR_DATA_DIR }/response_v1/complete.json"
-FILE_PATH_MULTI_RECEIPTS_DETECTOR_V1_EMPTY = f"{ MULTI_RECEIPTS_DETECTOR_DATA_DIR }/response_v1/empty.json"
+FILE_PATH_MULTI_RECEIPTS_DETECTOR_V1_COMPLETE = (
+    f"{ MULTI_RECEIPTS_DETECTOR_DATA_DIR }/response_v1/complete.json"
+)
+FILE_PATH_MULTI_RECEIPTS_DETECTOR_V1_EMPTY = (
+    f"{ MULTI_RECEIPTS_DETECTOR_DATA_DIR }/response_v1/empty.json"
+)
 
 
 @pytest.fixture
 def multi_receipts_detector_v1_doc() -> MultiReceiptsDetectorV1:
-    json_data = json.load(open(FILE_PATH_MULTI_RECEIPTS_DETECTOR_V1_COMPLETE, encoding="utf-8"))
+    json_data = json.load(
+        open(FILE_PATH_MULTI_RECEIPTS_DETECTOR_V1_COMPLETE, encoding="utf-8")
+    )
     return MultiReceiptsDetectorV1(json_data["document"]["inference"], page_n=None)
 
 
 @pytest.fixture
 def multi_receipts_detector_v1_doc_empty() -> MultiReceiptsDetectorV1:
-    json_data = json.load(open(FILE_PATH_MULTI_RECEIPTS_DETECTOR_V1_EMPTY, encoding="utf-8"))
+    json_data = json.load(
+        open(FILE_PATH_MULTI_RECEIPTS_DETECTOR_V1_EMPTY, encoding="utf-8")
+    )
     return MultiReceiptsDetectorV1(json_data["document"]["inference"], page_n=None)
 
 
 @pytest.fixture
 def multi_receipts_detector_v1_page0():
-    json_data = json.load(open(FILE_PATH_MULTI_RECEIPTS_DETECTOR_V1_COMPLETE, encoding="utf-8"))
-    return MultiReceiptsDetectorV1(json_data["document"]["inference"]["pages"][0], page_n=0)
+    json_data = json.load(
+        open(FILE_PATH_MULTI_RECEIPTS_DETECTOR_V1_COMPLETE, encoding="utf-8")
+    )
+    return MultiReceiptsDetectorV1(
+        json_data["document"]["inference"]["pages"][0], page_n=0
+    )
 
 
 def test_empty_doc_constructor(multi_receipts_detector_v1_doc_empty):

tests/documents/us/test_w9_v1.py

@@ -35,7 +35,7 @@ def test_empty_doc_constructor(w9_v1_doc_empty):
     assert w9_v1_doc_empty.business_name.value is None
     assert w9_v1_doc_empty.ein.value is None
     assert w9_v1_doc_empty.tax_classification.value is None
-    assert w9_v1_doc_empty.tax_classif_other_details.value is None
+    assert w9_v1_doc_empty.tax_classification_other_details.value is None
     assert w9_v1_doc_empty.w9_revision_date.value is None
     assert w9_v1_doc_empty.signature_position.value is None
     assert w9_v1_doc_empty.signature_date_position.value is None