chore: run linter checks

Author: aianch

lib/idp_common_pkg/idp_common/assessment/service.py

@@ -678,18 +678,20 @@ def process_document_section(self, document: Document, section_id: str) -> Docum
             logger.info(f"Time taken to read text content: {t2 - t1:.2f} seconds")
 
             # Read page images with configurable dimensions
-            image_config = self.config.get('image', {})
-            target_width = image_config.get('target_width', 951)   # Default fallback
-            target_height = image_config.get('target_height', 1268)
-            
+            image_config = self.config.get("image", {})
+            target_width = image_config.get("target_width", 951)  # Default fallback
+            target_height = image_config.get("target_height", 1268)
+
             page_images = []
             for page_id in sorted_page_ids:
                 if page_id not in document.pages:
                     continue
 
                 page = document.pages[page_id]
                 image_uri = page.image_uri
-                image_content = image.prepare_image(image_uri, target_width, target_height)
+                image_content = image.prepare_image(
+                    image_uri, target_width, target_height
+                )
                 page_images.append(image_content)
 
             t3 = time.time()

lib/idp_common_pkg/idp_common/classification/service.py

@@ -608,11 +608,13 @@ def classify_page_bedrock(
         # Load image content from URI with configurable dimensions
         if image_uri:
             try:
-                image_config = self.config.get('image', {})
-                target_width = image_config.get('target_width', 951)   # Default fallback
-                target_height = image_config.get('target_height', 1268)
-                
-                image_content = image.prepare_image(image_uri, target_width, target_height)
+                image_config = self.config.get("image", {})
+                target_width = image_config.get("target_width", 951)  # Default fallback
+                target_height = image_config.get("target_height", 1268)
+
+                image_content = image.prepare_image(
+                    image_uri, target_width, target_height
+                )
             except Exception as e:
                 logger.warning(f"Failed to load image content from {image_uri}: {e}")
                 # Continue without image content

lib/idp_common_pkg/idp_common/extraction/service.py

@@ -618,18 +618,20 @@ def process_document_section(self, document: Document, section_id: str) -> Docum
             logger.info(f"Time taken to read text content: {t1 - t0:.2f} seconds")
 
             # Read page images with configurable dimensions
-            image_config = self.config.get('image', {})
-            target_width = image_config.get('target_width', 951)   # Default fallback
-            target_height = image_config.get('target_height', 1268)
-            
+            image_config = self.config.get("image", {})
+            target_width = image_config.get("target_width", 951)  # Default fallback
+            target_height = image_config.get("target_height", 1268)
+
             page_images = []
             for page_id in sorted_page_ids:
                 if page_id not in document.pages:
                     continue
 
                 page = document.pages[page_id]
                 image_uri = page.image_uri
-                image_content = image.prepare_image(image_uri, target_width, target_height)
+                image_content = image.prepare_image(
+                    image_uri, target_width, target_height
+                )
                 page_images.append(image_content)
 
             t2 = time.time()

lib/idp_common_pkg/idp_common/ocr/service.py

@@ -56,13 +56,15 @@ def __init__(
         self.max_workers = max_workers
         self.dpi = dpi
         self.resize_config = resize_config
-        
+
         # Log DPI setting for debugging
         logger.info(f"OCR Service initialized with DPI: {self.dpi}")
-        
+
         # Log resize config if provided
         if self.resize_config:
-            logger.info(f"OCR Service initialized with resize config: {self.resize_config}")
+            logger.info(
+                f"OCR Service initialized with resize config: {self.resize_config}"
+            )
 
         # Define valid Textract feature types
         VALID_FEATURES = ["TABLES", "FORMS", "SIGNATURES", "LAYOUT"]
@@ -291,13 +293,18 @@ def _process_single_page(
         # Resize image for OCR processing if configured
         ocr_img_bytes = img_bytes  # Default to original image
         if self.resize_config:
-            target_width = self.resize_config.get('target_width')
-            target_height = self.resize_config.get('target_height')
-            
+            target_width = self.resize_config.get("target_width")
+            target_height = self.resize_config.get("target_height")
+
             if target_width and target_height:
                 from idp_common import image
-                ocr_img_bytes = image.resize_image(img_bytes, target_width, target_height)
-                logger.debug(f"Resized image for OCR processing (page {page_id}) to {target_width}x{target_height}")
+
+                ocr_img_bytes = image.resize_image(
+                    img_bytes, target_width, target_height
+                )
+                logger.debug(
+                    f"Resized image for OCR processing (page {page_id}) to {target_width}x{target_height}"
+                )
 
         # Process with OCR using potentially resized image
         if isinstance(self.enhanced_features, list) and self.enhanced_features: