From 0c35da9df789af79bcb193320fdbc814195ee940 Mon Sep 17 00:00:00 2001
From: Harshal Sheth <hsheth2@gmail.com>
Date: Thu, 12 Dec 2024 16:48:58 -0500
Subject: [PATCH] fix for airflow 2.3

---
 .../example_dags/graph_usage_sample_dag.py                    | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/metadata-ingestion-modules/airflow-plugin/src/datahub_airflow_plugin/example_dags/graph_usage_sample_dag.py b/metadata-ingestion-modules/airflow-plugin/src/datahub_airflow_plugin/example_dags/graph_usage_sample_dag.py
index adf92d30a106b..d72ba67c23cd7 100644
--- a/metadata-ingestion-modules/airflow-plugin/src/datahub_airflow_plugin/example_dags/graph_usage_sample_dag.py
+++ b/metadata-ingestion-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,
 )