From d8e0b0604fcf562e34a422bfffc013fb95aa6af9 Mon Sep 17 00:00:00 2001 From: Abhi Suryawanshi Date: Fri, 7 Nov 2025 12:59:40 +0530 Subject: [PATCH] fix: corrected default path handling (closes #2) --- app/etl/extract.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/app/etl/extract.py b/app/etl/extract.py index f8ac96c..92c8f39 100644 --- a/app/etl/extract.py +++ b/app/etl/extract.py @@ -1,8 +1,11 @@ import pandas as pd import os -# TODO (Find & Fix) -def extract(path: str = "xyz.csv") -> pd.DataFrame : +# Get the base directory (app/) relative to this file (app/etl/extract.py) +BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) +DEFAULT_DATA_PATH = os.path.join(BASE_DIR, "data.csv") + +def extract(path: str = DEFAULT_DATA_PATH) -> pd.DataFrame : """ Extracts data from CSV, Excel, or JSON file.