diff --git a/target_s3/sinks.py b/target_s3/sinks.py index c9f2acc..f84ecb4 100644 --- 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() \ No newline at end of file + format_type_client.run()