Move source_objects datatype check out of GCSToBigQueryOperator.__init__ (#20347)

josh-fell · web-flow · commit 17404f1f10ef · 2021-12-16T21:58:20.000+01:00
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()
