From beb597e7c31ab9fceaee30e4a3bc7aedc37a825d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tom=C3=A1s=20Far=C3=ADas=20Santana?= Date: Fri, 13 Dec 2024 15:25:11 +0100 Subject: [PATCH] fix: Add missing argument --- posthog/temporal/batch_exports/snowflake_batch_export.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/posthog/temporal/batch_exports/snowflake_batch_export.py b/posthog/temporal/batch_exports/snowflake_batch_export.py index 51663102e6d41..aef1f217090b3 100644 --- a/posthog/temporal/batch_exports/snowflake_batch_export.py +++ b/posthog/temporal/batch_exports/snowflake_batch_export.py @@ -501,11 +501,12 @@ def __init__( heartbeater: Heartbeater, heartbeat_details: SnowflakeHeartbeatDetails, data_interval_start: dt.datetime | str | None, + writer_format: WriterFormat, snowflake_client: SnowflakeClient, snowflake_table: str, snowflake_table_stage_prefix: str, ): - super().__init__(heartbeater, heartbeat_details, data_interval_start) + super().__init__(heartbeater, heartbeat_details, data_interval_start, writer_format) self.heartbeat_details: SnowflakeHeartbeatDetails = heartbeat_details self.snowflake_table = snowflake_table self.snowflake_client = snowflake_client @@ -729,6 +730,7 @@ async def insert_into_snowflake_activity(inputs: SnowflakeInsertInputs) -> Recor snowflake_client=snow_client, snowflake_table=snow_stage_table if requires_merge else snow_table, snowflake_table_stage_prefix=data_interval_end_str, + multiple_files=True, ) await snow_client.copy_loaded_files_to_snowflake_table(