diff --git a/ea_airflow_util/dags/s3_to_snowflake_dag.py b/ea_airflow_util/dags/s3_to_snowflake_dag.py index a9e8c71..af9f48a 100644 --- a/ea_airflow_util/dags/s3_to_snowflake_dag.py +++ b/ea_airflow_util/dags/s3_to_snowflake_dag.py @@ -1,6 +1,6 @@ import os -from airflow.operators.python_operator import PythonOperator +from airflow.operators.python import PythonOperator from airflow.providers.amazon.aws.operators.s3 import S3ListOperator from airflow.utils.helpers import chain diff --git a/ea_airflow_util/dags/sftp_to_snowflake_dag.py b/ea_airflow_util/dags/sftp_to_snowflake_dag.py index 599908d..b37d3c8 100644 --- a/ea_airflow_util/dags/sftp_to_snowflake_dag.py +++ b/ea_airflow_util/dags/sftp_to_snowflake_dag.py @@ -6,8 +6,8 @@ from airflow.exceptions import AirflowSkipException from airflow.providers.amazon.aws.hooks.s3 import S3Hook -from airflow.operators.bash_operator import BashOperator -from airflow.operators.python_operator import PythonOperator +from airflow.operators.bash import BashOperator +from airflow.operators.python import PythonOperator from airflow.providers.sftp.hooks.sftp import SFTPHook from airflow.utils.task_group import TaskGroup diff --git a/ea_airflow_util/providers/aws/operators/s3.py b/ea_airflow_util/providers/aws/operators/s3.py index 795617e..c7e87d6 100644 --- a/ea_airflow_util/providers/aws/operators/s3.py +++ b/ea_airflow_util/providers/aws/operators/s3.py @@ -8,7 +8,6 @@ from airflow.models import BaseOperator from airflow.providers.amazon.aws.operators.s3 import S3FileTransformOperator from airflow.providers.snowflake.hooks.snowflake import SnowflakeHook -from airflow.utils.decorators import apply_defaults class LoopS3FileTransformOperator(S3FileTransformOperator): @@ -105,7 +104,6 @@ class S3ToSnowflakeOperator(BaseOperator): """ template_fields = ('s3_destination_key', 's3_destination_dir', 's3_destination_filename',) - @apply_defaults def __init__(self, *, snowflake_conn_id: str,