fix lint errors

Author: Bob Strahan

lib/idp_common_pkg/idp_common/classification/service.py

@@ -16,7 +16,6 @@
 import time
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from datetime import datetime, timedelta, timezone
-from decimal import Decimal
 from typing import Any, Dict, List, Optional, Set, Union
 
 import boto3

lib/idp_common_pkg/tests/unit/classification/test_classification_service.py

@@ -274,9 +274,7 @@ def test_classify_page_bedrock_model_error(
 
         # Call the method and expect exception to be raised
         with pytest.raises(Exception, match="Model error"):
-            service.classify_page_bedrock(
-                page_id="1", text_uri="s3://bucket/text.txt"
-            )
+            service.classify_page_bedrock(page_id="1", text_uri="s3://bucket/text.txt")
 
     @patch("boto3.client")
     def test_classify_page_sagemaker_success(self, mock_boto_client, mock_config):

lib/idp_common_pkg/tests/unit/extraction/test_extraction_service.py

@@ -331,13 +331,7 @@ def test_process_document_section_missing_page(
         mock_invoke_model.return_value = {
             "response": {
                 "output": {
-                    "message": {
-                        "content": [
-                            {
-                                "text": '{"invoice_number": "INV-123"}'
-                            }
-                        ]
-                    }
+                    "message": {"content": [{"text": '{"invoice_number": "INV-123"}'}]}
                 }
             },
             "metering": {"input_tokens": 100, "output_tokens": 50},