fix(ingest): clickhouse - fix types changes in clickhouse sqlalchemy …

…0.2.3 (#6572)
datahub-project · Nov 29, 2022 · fb2ffe4 · fb2ffe4
1 parent 1bb8be5
commit fb2ffe4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/sql/clickhouse.py b/metadata-ingestion/src/datahub/ingestion/source/sql/clickhouse.py
@@ -73,6 +73,9 @@
 base.ischema_names["Int256"] = INTEGER
 base.ischema_names["UInt128"] = INTEGER
 base.ischema_names["UInt256"] = INTEGER
+# This is needed for clickhouse-sqlalchemy 0.2.3
+base.ischema_names["DateTime"] = DATETIME
+base.ischema_names["DateTime64"] = DATETIME
 
 register_custom_type(custom_types.common.Array, ArrayTypeClass)
 register_custom_type(custom_types.ip.IPv4, NumberTypeClass)