From 06c774cb5c9607ab6d593d7df2279298a420a634 Mon Sep 17 00:00:00 2001 From: Pablo Estrada Date: Mon, 30 Nov 2020 06:56:36 -0800 Subject: [PATCH] Fixup --- sdks/python/apache_beam/io/gcp/bigquery_read_internal.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/sdks/python/apache_beam/io/gcp/bigquery_read_internal.py b/sdks/python/apache_beam/io/gcp/bigquery_read_internal.py index f1ad38112fc1..394e435a3ace 100644 --- a/sdks/python/apache_beam/io/gcp/bigquery_read_internal.py +++ b/sdks/python/apache_beam/io/gcp/bigquery_read_internal.py @@ -176,10 +176,11 @@ def _get_temp_dataset(self): else: return self.temp_dataset - def process(self, element: 'ReadFromBigQueryRequest', *args, - **kwargs) -> Iterable[BoundedSource]: + def process(self, + element: 'ReadFromBigQueryRequest') -> Iterable[BoundedSource]: bq = bigquery_tools.BigQueryWrapper( temp_dataset_id=self._get_temp_dataset().datasetId) + # TODO(BEAM-11359): Clean up temp dataset at pipeline completion. if element.query is not None: self._setup_temporary_dataset(bq, element)