The sql_data is a "SnowflakeQueryResult" type object and not a dataframe object which is why it is not subscriptable when you try to get the column_1 using data['COLUMN_1']