Fix for empty results df_from_query_by_keys, ty Emeka

pennsignals · Feb 10, 2022 · e1ab7b2 · e1ab7b2
1 parent 22cda26
commit e1ab7b2
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/dsdk/persistor.py b/src/dsdk/persistor.py
@@ -139,9 +139,12 @@ def df_from_query_by_keys(
             fout.write(rendered)
         cur.execute(rendered)
         rows = cur.fetchall()
-        df = DataFrame(rows)
         columns = (each[0] for each in cur.description)
-        df.columns = columns
+        if rows:
+            df = DataFrame(rows)
+            df.columns = columns
+        else:
+            df = DataFrame(columns=columns)
         return df
 
     @classmethod