From 4751c0ed46773d2ac1d1487d133a9f287ff38256 Mon Sep 17 00:00:00 2001 From: Ross Gray Date: Thu, 28 Nov 2024 14:26:17 +0000 Subject: [PATCH] Cleanup --- posthog/temporal/batch_exports/bigquery_batch_export.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/posthog/temporal/batch_exports/bigquery_batch_export.py b/posthog/temporal/batch_exports/bigquery_batch_export.py index b5862ed977d0d..e99ba77f3c1bf 100644 --- a/posthog/temporal/batch_exports/bigquery_batch_export.py +++ b/posthog/temporal/batch_exports/bigquery_batch_export.py @@ -199,15 +199,14 @@ async def managed_table( project_id: str, dataset_id: str, table_id: str, - table_schema: list[bigquery.SchemaField] | None = None, + table_schema: list[bigquery.SchemaField], exists_ok: bool = True, not_found_ok: bool = True, delete: bool = True, create: bool = True, ) -> collections.abc.AsyncGenerator[bigquery.Table, None]: - """Manage a table in BigQuery by ensure it exists while in context.""" + """Manage a table in BigQuery by ensuring it exists while in context.""" if create is True: - assert table_schema is not None, "Table schema is required when creating a table" table = await self.acreate_table(project_id, dataset_id, table_id, table_schema, exists_ok) else: table = await self.aget_table(project_id, dataset_id, table_id)