fix: Add missing argument

PostHog · Dec 13, 2024 · beb597e · beb597e
1 parent 2bc6960
commit beb597e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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(