diff options
| -rw-r--r-- | obfuscator/csv_writer.py | 1 | ||||
| -rw-r--r-- | obfuscator/obfuscate.py | 4 |
2 files changed, 2 insertions, 3 deletions
diff --git a/obfuscator/csv_writer.py b/obfuscator/csv_writer.py index 2bff6e0..de7cd4b 100644 --- a/obfuscator/csv_writer.py +++ b/obfuscator/csv_writer.py @@ -9,7 +9,6 @@ logger = get_logger("CSVWRITER") def create_byte_stream(data: List[Dict[str, str]]) -> bytes: if not data: logger.error("Invalid or empty data was provided to write") - raise output = io.StringIO() diff --git a/obfuscator/obfuscate.py b/obfuscator/obfuscate.py index 73448ce..9cd3a03 100644 --- a/obfuscator/obfuscate.py +++ b/obfuscator/obfuscate.py @@ -13,10 +13,10 @@ def obfuscate( """ if not data: logger.error("Invalid or empty was provided to obfuscate") - raise + raise ValueError("Invalid data provided to obfuscate") if not pii_fields: logger.error("No PII fields provided to obfuscate") - raise + raise ValueError("No PII fields provided to obfuscate") return [ {k: ("***" if k in pii_fields else v) for k, v in record.items()} |