fix: Typing issues

PostHog · Nov 9, 2023 · 638d5d1 · 638d5d1
1 parent dc1afc9
commit 638d5d1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/posthog/temporal/workflows/postgres_batch_export.py b/posthog/temporal/workflows/postgres_batch_export.py
@@ -121,7 +121,7 @@ async def create_table_in_postgres(
                     # This is safe as we are hardcoding the type values anyways.
                     sql.SQL("{field} {type}").format(
                         field=sql.Identifier(field),
-                        type=sql.SQL(field_type),  # type: ignore
+                        type=sql.SQL(field_type),
                     )
                     for field, field_type in fields
                 ),

diff --git a/posthog/temporal/workflows/redshift_batch_export.py b/posthog/temporal/workflows/redshift_batch_export.py
@@ -72,11 +72,11 @@ async def insert_records_to_redshift(
             if len(batch) < batch_size:
                 continue
 
-            await cursor.execute_many(cursor, query, batch, template)
+            await cursor.executemany(cursor, query, batch, template)
             batch = []
 
         if len(batch) > 0:
-            await cursor.execute_many(cursor, query, batch, template)
+            await cursor.executemany(cursor, query, batch, template)
 
 
 @dataclass
@@ -148,7 +148,7 @@ async def insert_into_redshift_activity(inputs: RedshiftInsertInputs):
         properties_type = "VARCHAR(65535)" if inputs.properties_data_type == "varchar" else "SUPER"
 
         async with postgres_connection(inputs) as connection:
-            create_table_in_postgres(
+            await create_table_in_postgres(
                 connection,
                 schema=inputs.schema,
                 table_name=inputs.table_name,