Update _simpler_sf.py

Fixed a bug in _detect_fields()
benvigano · May 19, 2023 · 1e9d3f8 · 1e9d3f8
1 parent aa425cd
commit 1e9d3f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/simpler_sf/_simpler_sf.py b/simpler_sf/_simpler_sf.py
@@ -163,7 +163,7 @@ def _smart_query(
     output_df = pd.concat(dfs)
 
     # If the query didn't output any records, add the columns for output consistency
-    if len(dfs) == 0:
+    if output_df.shape[0] == 0:
         # Determine the columns from the query
         output_df = pd.DataFrame(columns=_determine_fields(query))
     else: