diff --git a/src/common/csvreader.py b/src/common/csvreader.py index e700aa2..f702e84 100644 --- a/src/common/csvreader.py +++ b/src/common/csvreader.py @@ -36,6 +36,7 @@ def read_budgets(csv:str)-> list[Budget]: df = pd.read_csv(csv) + budgets = [] for _,row in df.iterrows(): budget = Budget( component_id=row['component_id'], @@ -47,6 +48,8 @@ def read_budgets(csv:str)-> list[Budget]: ) budgets.append(budget) + return budgets + def _get_csv_path(csv:str) -> str: if os.path.exists(csv): return csv