Merge pull request #1443 from sul-dlss/t1441-polling-dag-bug

Create a DAG object for dag_run in Poll 979 Sensor
sul-dlss · Nov 18, 2024 · db6c4b0 · db6c4b0
2 parents b76bf62 + ebb5944
commit db6c4b0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/libsys_airflow/plugins/digital_bookplates/dag_979_sensor.py b/libsys_airflow/plugins/digital_bookplates/dag_979_sensor.py
@@ -1,6 +1,6 @@
 import logging
 
-from airflow.models import DagRun
+from airflow.models import DAG, DagRun
 from airflow.sensors.base_sensor_operator import BaseSensorOperator
 
 from libsys_airflow.plugins.shared.utils import dag_run_url
@@ -25,6 +25,7 @@ def poke(self, context) -> bool:
             dag_run = dag_runs[0]
             state = dag_run.get_state()
             self.dag_runs[dag_run_id]['state'] = state
+            dag_run.dag = DAG(dag_id='digital_bookplate_979')
             self.dag_runs[dag_run_id]['url'] = dag_run_url(dag_run=dag_run)
             if state in ['success', 'failed']:
                 instances = []