fix bug on QueryContextFactory

apache · Feb 16, 2023 · 9b43492 · 9b43492
1 parent 4ddf67f
commit 9b43492
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/superset/common/query_context_factory.py b/superset/common/query_context_factory.py
@@ -121,7 +121,7 @@ def _apply_granularity(
         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)
         }