From 04d67e47e047ac44920c04d6fb0e5ea6f5bbe149 Mon Sep 17 00:00:00 2001 From: Kalyan Chakravarthy Date: Mon, 13 May 2024 14:28:30 +0530 Subject: [PATCH] Refactor Summary class to update summary dataframe and handle file path --- langtest/utils/benchmark_utils.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/langtest/utils/benchmark_utils.py b/langtest/utils/benchmark_utils.py index 2bbab53de..4b07dae65 100644 --- a/langtest/utils/benchmark_utils.py +++ b/langtest/utils/benchmark_utils.py @@ -187,9 +187,7 @@ def __init__(self, path: str, *args, **kwargs) -> None: self.save_dir = path self.file_path = f"{path}summary.csv" - self.summary_df: pd.DataFrame = self.load_data_from_file( - self.file_path, *args, **kwargs - ) + self.summary_df: pd.DataFrame = self.__update_summary_df() def load_data_from_file(self, path: str, *args, **kwargs) -> pd.DataFrame: """ @@ -242,6 +240,8 @@ def add_report( """ Add a new report to the summary """ + # Load and Update the summary dataframe + self.summary_df = self.__update_summary_df() from datetime import datetime @@ -328,3 +328,12 @@ def __group_by_cols(self): @property def df(self) -> pd.DataFrame: return self.summary_df + + def __update_summary_df(self): + """ + Update the summary dataframe + """ + if self.file_path.startswith("~"): + self.file_path = os.path.expanduser(self.file_path) + self.summary_df = self.load_data_from_file(self.file_path) + return self.summary_df