[AIRFLOW-3138] Use current data type for migrations (apache#3985)

* Use timestamp instead of timestamp with timezone for migration.
aliceabe · Jan 3, 2019 · d7b10eb · d7b10eb
1 parent 5347289
commit d7b10eb
Showing 1 changed file with 1 addition and 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
@@ -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)