fix for airflow 2.3

datahub-project · Dec 12, 2024 · 0c35da9 · 0c35da9
1 parent 8ac7246
commit 0c35da9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...-modules/airflow-plugin/src/datahub_airflow_plugin/example_dags/graph_usage_sample_dag.py b/...-modules/airflow-plugin/src/datahub_airflow_plugin/example_dags/graph_usage_sample_dag.py
@@ -1,5 +1,7 @@
 """This example DAG demonstrates how to create and use a DataHubGraph client."""
 
+from datetime import timedelta
+
 import pendulum
 from airflow.decorators import dag, task
 from datahub.ingestion.graph.client import DataHubGraph, RemovedStatusFilter
@@ -8,7 +10,7 @@
 
 
 @dag(
-    schedule=None,
+    schedule_interval=timedelta(days=1),
     start_date=pendulum.datetime(2021, 1, 1, tz="UTC"),
     catchup=False,
 )