import airflow.operators.python

lsst-uk · Jul 1, 2024 · f04ad26 · f04ad26
1 parent bae2b19
commit f04ad26
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/echo-side/dags/monitor.py b/echo-side/dags/monitor.py
@@ -1,10 +1,9 @@
 from airflow import DAG
-# from airflow.operators.docker_operator import DockerOperator
 from airflow.providers.cncf.kubernetes.operators.pod import KubernetesPodOperator
 from airflow.utils.dates import days_ago
 from airflow.models import Variable
 from datetime import timedelta
-from airflow.operators.python_operator import PythonOperator
+from airflow.operators.python import PythonOperator
 
 from kubernetes.client import models
 from datetime import datetime
@@ -68,6 +67,7 @@ def compare_csv_file_lists(log_folder):
     task_id='compare_csv_file_lists',
     python_callable=compare_csv_file_lists,
     dag=dag,
+    op_args=['/lsst-backup-logs'],
 )
 
 # Set the task sequence