From 8b0960278e62bef80d1899db90752fce56df4809 Mon Sep 17 00:00:00 2001 From: Ash Berlin-Taylor Date: Fri, 10 Sep 2021 00:07:19 +0100 Subject: [PATCH] Fix deprecation error message rather than silencing it (#18126) (cherry picked from commit c9d29467f71060f14863ca3508cb1055572479b5) --- .../092435bf5d12_add_max_active_runs_column_to_dagmodel_.py | 2 +- .../ccde3e26fe78_add_index_on_state_dag_id_for_queued_.py | 2 +- airflow/models/dagrun.py | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/airflow/migrations/versions/092435bf5d12_add_max_active_runs_column_to_dagmodel_.py b/airflow/migrations/versions/092435bf5d12_add_max_active_runs_column_to_dagmodel_.py index 82bb4c295a7451..a9f612d591ef84 100644 --- a/airflow/migrations/versions/092435bf5d12_add_max_active_runs_column_to_dagmodel_.py +++ b/airflow/migrations/versions/092435bf5d12_add_max_active_runs_column_to_dagmodel_.py @@ -44,7 +44,7 @@ def upgrade(): batch_op.create_index( 'idx_dag_run_running_dags', ["state", "dag_id"], - postgres_where=text("state='running'"), + postgresql_where=text("state='running'"), mssql_where=text("state='running'"), sqlite_where=text("state='running'"), ) diff --git a/airflow/migrations/versions/ccde3e26fe78_add_index_on_state_dag_id_for_queued_.py b/airflow/migrations/versions/ccde3e26fe78_add_index_on_state_dag_id_for_queued_.py index 7326d73abff07b..6a1cbe65329da7 100644 --- a/airflow/migrations/versions/ccde3e26fe78_add_index_on_state_dag_id_for_queued_.py +++ b/airflow/migrations/versions/ccde3e26fe78_add_index_on_state_dag_id_for_queued_.py @@ -40,7 +40,7 @@ def upgrade(): batch_op.create_index( 'idx_dag_run_queued_dags', ["state", "dag_id"], - postgres_where=text("state='queued'"), + postgresql_where=text("state='queued'"), mssql_where=text("state='queued'"), sqlite_where=text("state='queued'"), ) diff --git a/airflow/models/dagrun.py b/airflow/models/dagrun.py index 1e5c2c1676cd66..ec4bdfb0697da9 100644 --- a/airflow/models/dagrun.py +++ b/airflow/models/dagrun.py @@ -103,7 +103,7 @@ class DagRun(Base, LoggingMixin): 'idx_dag_run_running_dags', 'state', 'dag_id', - postgres_where=text("state='running'"), + postgresql_where=text("state='running'"), mssql_where=text("state='running'"), sqlite_where=text("state='running'"), ), @@ -113,7 +113,7 @@ class DagRun(Base, LoggingMixin): 'idx_dag_run_queued_dags', 'state', 'dag_id', - postgres_where=text("state='queued'"), + postgresql_where=text("state='queued'"), mssql_where=text("state='queued'"), sqlite_where=text("state='queued'"), ),