Fixup

apache · Nov 30, 2020 · 06c774c · 06c774c
1 parent f328870
commit 06c774c
Showing 1 changed file with 3 additions and 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
@@ -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)