From 852648d9c8223b53b85116e3b8a7dc8061eaa52f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tom=C3=A1s=20Far=C3=ADas=20Santana?= Date: Wed, 20 Mar 2024 10:53:36 +0100 Subject: [PATCH] fix: Return inside context manager Co-authored-by: Brett Hoerner --- posthog/temporal/batch_exports/postgres_batch_export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/posthog/temporal/batch_exports/postgres_batch_export.py b/posthog/temporal/batch_exports/postgres_batch_export.py index ab46cadc6941f..5dbfc6faa4acf 100644 --- a/posthog/temporal/batch_exports/postgres_batch_export.py +++ b/posthog/temporal/batch_exports/postgres_batch_export.py @@ -359,7 +359,7 @@ async def flush_to_postgres(): if pg_file.tell() > 0: await flush_to_postgres() - return pg_file.records_total + return pg_file.records_total @workflow.defn(name="postgres-export")