use list comprehension when returning output from read_local

1 files changed, 3 insertions, 9 deletions

diff --git a/obfuscator/csv_reader.py b/obfuscator/csv_reader.py
index 55fb892..3b4496b 100644
--- a/obfuscator/csv_reader.py
+++ b/obfuscator/csv_reader.py
@@ -5,25 +5,19 @@ from obfuscator.logger import get_logger
 
 logger = get_logger("CSVReader")
 
-
 class CSVReader:
     @staticmethod
     def read_local(path) -> List[Dict[str, str]]:
         logger.debug(f"Reading local CSV from: {path}")
-        data = []
 
         try:
-            with open(path, mode="r", encoding="utf-8") as file:
-                reader = csv.DictReader(file)
-                for row in reader:
-                    data.append(dict(row))
+            with open(path, mode="r", encoding="utf-8") as f:
+                reader = csv.DictReader(f)
+                return [dict(row) for row in reader]
         except FileNotFoundError:
             logger.error(f"File not found: {path}")
         except Exception as e:
             logger.error(f"Error reading file: {e}")
-
-        logger.debug(f"Total rows read: {len(data)}")
-        return data
     
     @staticmethod
     def read_s3(path) -> List[Dict[str, str]]: