diff --git a/pytd/query_engine.py b/pytd/query_engine.py index fdd471c..77ec5cc 100644 --- a/pytd/query_engine.py +++ b/pytd/query_engine.py @@ -95,7 +95,8 @@ def execute(self, query, **kwargs): cur = self.cursor(**kwargs) self.executed = cur.execute(query) rows = cur.fetchall() - columns = [desc[0] for desc in cur.description] + # cur.description is None for CREATE and DROP statements in recent version of Trino + columns = [desc[0] for desc in cur.description] if cur.description else None return {"data": rows, "columns": columns} def create_header(self, extra_lines=[]):