fix: dict bug on QueryContextFactory (#23093)

Co-authored-by: ok9897 <ok9897@krafton.com>
This commit is contained in:
Changhoon Oh
2023-02-16 23:04:39 +09:00
committed by GitHub
parent 4ddf67fc14
commit 21dd1601bf

View File

@@ -121,7 +121,7 @@ class QueryContextFactory: # pylint: disable=too-few-public-methods
datasource: BaseDatasource,
) -> None:
temporal_columns = {
column.column_name
column["column_name"] if isinstance(column, dict) else column.column_name
for column in datasource.columns
if (column["is_dttm"] if isinstance(column, dict) else column.is_dttm)
}