From bdd041ff3841153017f70688b45a01b8872e639e Mon Sep 17 00:00:00 2001 From: Zain Sohail Date: Wed, 4 Oct 2023 14:19:44 +0200 Subject: [PATCH] change back from augmented to converted (FlashLoader) (#156) --- sed/loader/flash/loader.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/sed/loader/flash/loader.py b/sed/loader/flash/loader.py index f6506280..f5f5c803 100644 --- a/sed/loader/flash/loader.py +++ b/sed/loader/flash/loader.py @@ -735,7 +735,7 @@ def parquet_handler( data_parquet_dir: Path, detector: str = "", parquet_path: Path = None, - augmented: bool = False, + converted: bool = False, load_parquet: bool = False, save_parquet: bool = False, ): @@ -747,7 +747,8 @@ def parquet_handler( detector (str, optional): Adds a identifier for parquets to distinguish multidetector systems. parquet_path (str, optional): Path to the combined parquet file. - augmented (bool, optional): True if data is augmented and saved into augmented folder. + converted (bool, optional): True if data is augmented by adding additional columns + externally and saved into converted folder. load_parquet (bool, optional): Loads the entire parquet into the dd dataframe. save_parquet (bool, optional): Saves the entire dataframe into a parquet. @@ -762,7 +763,7 @@ def parquet_handler( # Construct the parquet path if not provided if parquet_path is None: parquet_name = "_".join(str(run) for run in self.runs) - parquet_dir = data_parquet_dir.joinpath("augmented") if augmented else data_parquet_dir + parquet_dir = data_parquet_dir.joinpath("converted") if converted else data_parquet_dir parquet_path = parquet_dir.joinpath( "run_" + parquet_name + detector,