try to fix black errors

datafog/main.py

@@ -11,12 +11,13 @@
 logger = logging.getLogger("datafog_logger")
 logger.setLevel(logging.INFO)
 
+
 class DataFog:
     def __init__(
         self,
-        image_service = ImageService(),
-        text_service = TextService(),
-        spark_service = None,
+        image_service=ImageService(),
+        text_service=TextService(),
+        spark_service=None,
         operations: List[OperationType] = [OperationType.ANNOTATE_PII],
     ):
         self.image_service = image_service
@@ -44,7 +45,7 @@
             )
 
             if OperationType.ANNOTATE_PII in self.operations:
-                annotated_text = await self.text_service.batch_annotate_text_async( 
+                annotated_text = await self.text_service.batch_annotate_text_async(
                     extracted_text
                 )
                 self.logger.info(
@@ -62,14 +63,16 @@
         try:
             self.logger.info(f"Starting text pipeline with {len(str_list)} texts.")
             if OperationType.ANNOTATE_PII in self.operations:
-                annotated_text = await self.text_service.batch_annotate_text_async(str_list)
+                annotated_text = await self.text_service.batch_annotate_text_async(
+                    str_list
+                )
                 self.logger.info(
                     f"Text annotation completed with {len(annotated_text)} annotations."
                 )
                 return annotated_text
 
             self.logger.info("No annotation operation found; returning original texts.")
             return str_list
         except Exception as e:
             self.logger.error(f"Error in run_text_pipeline: {str(e)}")
             raise
@@ -85,17 +88,18 @@
                 )
                 return annotated_text
 
             self.logger.info("No annotation operation found; returning original texts.")
             return str_list
         except Exception as e:
             self.logger.error(f"Error in run_text_pipeline: {str(e)}")
             raise
 
     def _add_attributes(self, attributes: dict):
         """Add multiple attributes."""
         for key, value in attributes.items():
             pass
 
+
 class TextPIIAnnotator:
     def __init__(self):
         self.text_annotator = SpacyPIIAnnotator.create()

tests/test_main.py

@@ -35,6 +35,7 @@ def test_textpii_annotator():
 #     annotated_text = await ocr_annotator.run([image_url])
 #     assert "Satya Nadella" in annotated_text[0].get("PER", []), "PII not annotated correctly."
 
+
 def test_datafog_text_annotation_sync():
     """Test DataFog class for synchronous text annotation."""
     text = ["Joe Biden is the President of the United States."]
@@ -49,6 +50,7 @@ def test_datafog_text_annotation_sync():
         annotated_text, "the United States"
     ), "United States not found in annotated results."
 
+
 @pytest.mark.asyncio
 async def test_datafog_text_annotation():
     """Test DataFog class for text annotation."""