diff --git a/cob_datapipeline/boundwith_move_alma_sftp_to_s3_dag.py b/cob_datapipeline/boundwith_move_alma_sftp_to_s3_dag.py index 7e6c793d..33f0739f 100644 --- a/cob_datapipeline/boundwith_move_alma_sftp_to_s3_dag.py +++ b/cob_datapipeline/boundwith_move_alma_sftp_to_s3_dag.py @@ -2,7 +2,6 @@ from datetime import datetime, timedelta import logging import pendulum -from tulflow import tasks import airflow from airflow.models import Variable from airflow.providers.sftp.hooks.sftp import SFTPHook diff --git a/cob_datapipeline/catalog_move_alma_sftp_to_s3_dag.py b/cob_datapipeline/catalog_move_alma_sftp_to_s3_dag.py index 7ac792f5..e6f87b73 100644 --- a/cob_datapipeline/catalog_move_alma_sftp_to_s3_dag.py +++ b/cob_datapipeline/catalog_move_alma_sftp_to_s3_dag.py @@ -2,7 +2,6 @@ from datetime import datetime, timedelta import logging import pendulum -from tulflow import tasks import airflow from airflow.models import Variable from airflow.providers.sftp.hooks.sftp import SFTPHook @@ -27,7 +26,7 @@ "on_failure_callback": [slackpostonfail], "on_success_callback": [slackpostonsuccess], "email_on_retry": False, - "on_failure_callback": tasks.execute_slackpostonfail, + "on_failure_callback": [slackpostonfail], "retries": 5, "retry_delay": timedelta(minutes=5), } diff --git a/cob_datapipeline/dspace_harvest_dag.py b/cob_datapipeline/dspace_harvest_dag.py index 6f4c2243..5f038c6d 100644 --- a/cob_datapipeline/dspace_harvest_dag.py +++ b/cob_datapipeline/dspace_harvest_dag.py @@ -2,7 +2,7 @@ from datetime import datetime, timedelta import os import pendulum -from tulflow import harvest, tasks +from tulflow import harvest from cob_datapipeline import helpers from airflow.hooks.base import BaseHook from airflow.models import Variable