From 90542cfe838376988982fb5c9062fc8dee0b7c87 Mon Sep 17 00:00:00 2001
From: Alex Schofield <git@ajschof.me>
Date: Tue, 18 Feb 2025 23:34:26 +0000
Subject: update cli.py to use create_byte_stream

---
 cli.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/cli.py b/cli.py
index b1d4000..f2c8771 100644
--- a/cli.py
+++ b/cli.py
@@ -3,6 +3,7 @@ import json
 from obfuscator.csv_reader import CSVReader
 from obfuscator.obfuscate import obfuscate
 from obfuscator.logger import get_logger
+from obfuscator.csv_writer import create_byte_stream
 
 # Create the logger
 logger = get_logger("CLI")
@@ -44,6 +45,7 @@ def main():
     obfuscated_data = obfuscate(data, args.pii)
     # For debug purposes, log the obfuscated data as JSON for readability
     logger.debug("Obfuscated data (JSON): " + json.dumps(obfuscated_data, indent=4))
+    return create_byte_stream(obfuscated_data)
 
 
 # If the script is run directly (as it should be), call the main function
-- 
cgit v1.2.3