Fix deprecation error message rather than silencing it (#18126)

(cherry picked from commit c9d2946)
apache · Sep 10, 2021 · f5f70e0 · f5f70e0
1 parent 247382f
commit f5f70e0
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 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
@@ -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
@@ -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
@@ -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'"),
         ),