backup commit - 21st @ 00:00

1 files changed, 5 insertions, 4 deletions

diff --git a/gdpr_obfuscator/__init__.py b/gdpr_obfuscator/__init__.py
index edfc665..69854cb 100644
--- a/gdpr_obfuscator/__init__.py
+++ b/gdpr_obfuscator/__init__.py
@@ -1,17 +1,18 @@
 from .read import DataReader
 from .obfuscate import obfuscate
-from .logger import get_logger
 from typing import List, Dict
+from .utils import Utilities
 
-class ImportData:
+
+class Obfuscator:
     def __init__(self, verbosity: bool = False):
         self.verbosity = verbosity
         self.log_level = "DEBUG" if verbosity else "INFO"
-        self.logger = get_logger("ImportData", self.log_level)
+        self.logger = Utilities.get_logger("ImportData", self.log_level)
         self.reader = DataReader()
 
     def import_s3(self, path: str, pii_fields: List[str]) -> bytes:
         return self.reader.read_s3(path)
 
     def import_local(self, path: str, pii_fields: List[str]) -> bytes:
-        return self.reader.read_local(path)
\ No newline at end of file
+        return self.reader.read_local(path)