Fix the max_batch_size parameter usage

crowemi · Jan 8, 2024 · 9817fce · 9817fce
1 parent 4649b26
commit 9817fce
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()