test returning buffer instead of bytes

cfpb · Jan 9, 2025 · 61aa497 · 61aa497
1 parent 9f843c5
commit 61aa497
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/regtech_data_validator/data_formatters.py b/src/regtech_data_validator/data_formatters.py
@@ -134,7 +134,8 @@ def df_to_download(
         )
         empty_df.write_csv(buffer, quote_style='non_numeric', include_header=True)
         buffer.seek(0)
-        return buffer.getvalue()
+        return buffer
+        # return buffer.getvalue()
 
     # get the check for the phase the results were in, so we can pull out static data from each
     # found check
@@ -196,7 +197,8 @@ def df_to_download(
 
     sorted_df.write_csv(buffer, quote_style='non_numeric', include_header=False)
     buffer.seek(0)
-    return buffer.getvalue()
+    return buffer
+    # return buffer.getvalue()
 
 
 def df_to_csv(df: pl.DataFrame) -> str: