Merge pull request #33 from dmnpignaud/patch-1

Fix the max_batch_size parameter usage
crowemi · Jan 9, 2024 · 24d451a · 24d451a
2 parents 4649b26 + 9817fce
commit 24d451a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/target_s3/sinks.py b/target_s3/sinks.py
@@ -44,7 +44,7 @@ def max_size(self) -> int:
         Returns:
             Maximum batch size
         """
-        return self.config.get("batch_size", 10000)
+        return self.config.get("max_batch_size", 10000)
 
     def process_batch(self, context: dict) -> None:
         """Write out any prepped records and return once fully written."""
@@ -61,4 +61,4 @@ def process_batch(self, context: dict) -> None:
             isinstance(format_type_client, FormatBase) is True
         ), f"format_type_client must be of type Base; Type: {type(self.format_type_client)}."
 
-        format_type_client.run()
+        format_type_client.run()