From 9817fce4c2df255b2cfb07b4c7461869a240b661 Mon Sep 17 00:00:00 2001 From: dmnpignaud Date: Mon, 8 Jan 2024 18:42:14 +0100 Subject: [PATCH] Fix the max_batch_size parameter usage --- target_s3/sinks.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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()