add apply default sql_to_gcs operator

yimingl17 · Sep 18, 2019 · 06aa3d4 · 06aa3d4
1 parent fa8e18a
commit 06aa3d4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/airflow/operators/sql_to_gcs.py b/airflow/operators/sql_to_gcs.py
@@ -29,6 +29,7 @@
 
 from airflow.gcp.hooks.gcs import GoogleCloudStorageHook
 from airflow.models import BaseOperator
+from airflow.utils.decorators import apply_defaults
 
 
 class BaseSQLToGoogleCloudStorageOperator(BaseOperator, metaclass=abc.ABCMeta):
@@ -77,7 +78,7 @@ class BaseSQLToGoogleCloudStorageOperator(BaseOperator, metaclass=abc.ABCMeta):
     template_fields = ('sql', 'bucket', 'filename', 'schema_filename', 'schema', 'parameters')
     template_ext = ('.sql',)
     ui_color = '#a0e08c'
-
+    @apply_defaults
     def __init__(self,  # pylint: disable=too-many-arguments
                  sql,
                  bucket,