diff --git a/superset/semantic_layers/snowflake/semantic_view.py b/superset/semantic_layers/snowflake/semantic_view.py index 7bc609fe24f..8ce849af2f8 100644 --- a/superset/semantic_layers/snowflake/semantic_view.py +++ b/superset/semantic_layers/snowflake/semantic_view.py @@ -305,6 +305,7 @@ class SnowflakeSemanticView(SemanticViewImplementation): """ Execute a query and return the results as a Pandas DataFrame. """ + print("AM HERE") if not metrics and not dimensions: return DataFrame() @@ -321,6 +322,17 @@ class SnowflakeSemanticView(SemanticViewImplementation): with connect(**connection_parameters) as connection: df = connection.cursor().execute(query, parameters).fetch_pandas_all() + # map column names to dimension/metric names instead of IDs + mapping = { + **{dimension.id: dimension.name for dimension in dimensions}, + **{metric.id: metric.name for metric in metrics}, + } + print("BETO") + print(df.columns) + print(mapping) + df.rename(columns=mapping, inplace=True) + print(df.columns) + return SemanticResult( requests=[ SemanticRequest(