From 4d47543b6640592827410aac6175425b6d382b07 Mon Sep 17 00:00:00 2001 From: Michelle Noronha Date: Mon, 1 Oct 2018 18:02:02 -0400 Subject: [PATCH] [AIRFLOW-3138] Use current data type for migrations * Use timestamp instead of timestamp with timezone for migration. --- .../cc1e65623dc7_add_max_tries_column_to_task_instance.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/airflow/migrations/versions/cc1e65623dc7_add_max_tries_column_to_task_instance.py b/airflow/migrations/versions/cc1e65623dc7_add_max_tries_column_to_task_instance.py index 6155a40c81934..03195a7536d3d 100644 --- a/airflow/migrations/versions/cc1e65623dc7_add_max_tries_column_to_task_instance.py +++ b/airflow/migrations/versions/cc1e65623dc7_add_max_tries_column_to_task_instance.py @@ -29,7 +29,6 @@ import sqlalchemy as sa from airflow import settings from airflow.models import DagBag -from airflow.utils.sqlalchemy import UtcDateTime from sqlalchemy import Column, Integer, String from sqlalchemy.engine.reflection import Inspector @@ -51,7 +50,7 @@ class TaskInstance(Base): task_id = Column(String(ID_LEN), primary_key=True) dag_id = Column(String(ID_LEN), primary_key=True) - execution_date = Column(UtcDateTime, primary_key=True) + execution_date = Column(sa.DateTime, primary_key=True) max_tries = Column(Integer) try_number = Column(Integer, default=0)