Move source_objects datatype check out of GCSToBigQueryOperator.__ini…

…t__ (#20347)
apache · Dec 16, 2021 · 17404f1 · 17404f1
1 parent 6e51608
commit 17404f1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/airflow/providers/google/cloud/transfers/gcs_to_bigquery.py b/airflow/providers/google/cloud/transfers/gcs_to_bigquery.py
@@ -216,7 +216,7 @@ def __init__(
  if time_partitioning is None:
  time_partitioning = {}
  self.bucket = bucket
- self.source_objects = source_objects if isinstance(source_objects, list) else [source_objects]
+ self.source_objects = source_objects
  self.schema_object = schema_object
 
  # BQ config
@@ -279,6 +279,9 @@ def execute(self, context):
  else:
  schema_fields = self.schema_fields
 
+ self.source_objects = (
+ self.source_objects if isinstance(self.source_objects, list) else [self.source_objects]
+ )
  source_uris = [f'gs://{self.bucket}/{source_object}' for source_object in self.source_objects]
  conn = bq_hook.get_conn()
  cursor = conn.cursor()