From 3a92f79a43660d2238d5c6b218e4b98712a79cbd Mon Sep 17 00:00:00 2001 From: Pim Nelissen Date: Fri, 20 Feb 2026 12:01:12 +0100 Subject: [PATCH] fix dataloader exception handling --- src/pg_rad/dataloader/dataloader.py | 15 ++++----------- 1 file changed, 4 insertions(+), 11 deletions(-) diff --git a/src/pg_rad/dataloader/dataloader.py b/src/pg_rad/dataloader/dataloader.py index dfe94a6..7bbd6c3 100644 --- a/src/pg_rad/dataloader/dataloader.py +++ b/src/pg_rad/dataloader/dataloader.py @@ -2,7 +2,6 @@ import logging import pandas as pd -from pg_rad.exceptions.exceptions import DataLoadError, InvalidCSVError logger = logging.getLogger(__name__) @@ -12,18 +11,12 @@ def load_data(filename: str) -> pd.DataFrame: try: df = pd.read_csv(filename, delimiter=',') - except FileNotFoundError as e: - logger.error(f"File not found: {filename}") - raise DataLoadError(f"File does not exist: {filename}") from e - + logger.critical(e) + raise except pd.errors.ParserError as e: - logger.error(f"Invalid CSV format: {filename}") - raise InvalidCSVError(f"Invalid CSV file: {filename}") from e - - except Exception as e: - logger.exception(f"Unexpected error while loading {filename}") - raise DataLoadError("Unexpected error while loading data") from e + logger.critical(e) + raise logger.debug(f"File loaded: {filename}") return df