Fix dataframe columns

pennsignals · Feb 11, 2022 · e47b223 · e47b223
1 parent 0f04ed7
commit e47b223
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/dsdk/persistor.py b/src/dsdk/persistor.py
@@ -60,7 +60,7 @@ def df_from_query(
         if parameters is None:
             parameters = {}
         cur.execute(query, parameters)
-        columns = (each[0] for each in cur.description)
+        columns = tuple(each[0] for each in cur.description)
         rows = cur.fetchall()
         if rows:
             df = DataFrame(rows)
@@ -138,7 +138,7 @@ def df_from_query_by_keys(
         with NamedTemporaryFile("w", delete=False, suffix=".sql") as fout:
             fout.write(rendered)
         cur.execute(rendered)
-        columns = (each[0] for each in cur.description)
+        columns = tuple(each[0] for each in cur.description)
         rows = cur.fetchall()
         if rows:
             df = DataFrame(rows)