From 399a2a103dfdb9ed8ccfc06ca23e1a2c8cd038e5 Mon Sep 17 00:00:00 2001 From: Ayush Shah Date: Thu, 18 Aug 2022 18:30:00 +0530 Subject: [PATCH] Fix profiler item_type sqlalchemy datatype issue (#6794) --- ingestion/src/metadata/orm_profiler/orm/converter.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ingestion/src/metadata/orm_profiler/orm/converter.py b/ingestion/src/metadata/orm_profiler/orm/converter.py index f59b5a1a19f..6a19a2e4fed 100644 --- a/ingestion/src/metadata/orm_profiler/orm/converter.py +++ b/ingestion/src/metadata/orm_profiler/orm/converter.py @@ -99,7 +99,9 @@ def build_orm_col(idx: int, col: Column, table_service_type) -> sqlalchemy.Colum """ return sqlalchemy.Column( name=str(col.name.__root__), - type_=_TYPE_MAP.get(col.dataType), + type_=_TYPE_MAP.get(col.dataType) + if not col.arrayDataType + else _TYPE_MAP.get(col.dataType)(item_type=col.arrayDataType), primary_key=not bool(idx), # The first col seen is used as PK quote=check_snowflake_case_sensitive(table_service_type, col.name.__root__), key=str(