Composer db cleanup dag (GoogleCloudPlatform#11091)

keshav9316 · web-flow · commit d1d77c5a323d · 2024-01-03T10:01:53.000-05:00
* The execution_date attribute is no longer supported by the TaskFail model as the column was dropped in migration 48925b2719cb[1] which belongs to Airflow 2.3.0 [1] https://airflow.apache.org/docs/apache-airflow/stable/migrations-ref.html * The execution_date attribute is no longer supported by the TaskFail model as the column was dropped in migration 48925b2719cb [1] which belongs to Airflow 2.3.0 [1] https://airflow.apache.org/docs/apache-airflow/stable/migrations-ref.html
diff --git a/composer/workflows/airflow_db_cleanup.py b/composer/workflows/airflow_db_cleanup.py
@@ -109,7 +109,7 @@
     },
     {
         "airflow_db_model": TaskInstance,
-        "age_check_column": TaskInstance.execution_date
+        "age_check_column": TaskInstance.start_date
         if AIRFLOW_VERSION < ["2", "2", "0"]
         else TaskInstance.start_date,
         "keep_last": False,
@@ -174,7 +174,7 @@
     DATABASE_OBJECTS.append(
         {
             "airflow_db_model": TaskFail,
-            "age_check_column": TaskFail.execution_date,
+            "age_check_column": TaskFail.start_date,
             "keep_last": False,
             "keep_last_filters": None,
             "keep_last_group_by": None,
