Configure back_populates between DagScheduleDatasetReference.dag …

…and `DagModel.schedule_dataset_references` (apache#39392)
RodrigoGanancia · May 10, 2024 · 8e794b3 · 8e794b3
1 parent baa7f40
commit 8e794b3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/airflow/models/dag.py b/airflow/models/dag.py
@@ -3668,6 +3668,7 @@ class DagModel(Base):
     )
     schedule_dataset_references = relationship(
         "DagScheduleDatasetReference",
+        back_populates="dag",
         cascade="all, delete, delete-orphan",
     )
     schedule_datasets = association_proxy("schedule_dataset_references", "dataset")

diff --git a/airflow/models/dataset.py b/airflow/models/dataset.py
@@ -112,7 +112,7 @@ class DagScheduleDatasetReference(Base):
     updated_at = Column(UtcDateTime, default=timezone.utcnow, onupdate=timezone.utcnow, nullable=False)
 
     dataset = relationship("DatasetModel", back_populates="consuming_dags")
-    dag = relationship("DagModel")
+    dag = relationship("DagModel", back_populates="schedule_dataset_references")
 
     queue_records = relationship(
         "DatasetDagRunQueue",