From 0fe8475b180ad052967ff44436aef2950643925a Mon Sep 17 00:00:00 2001 From: Steinn Ymir Agustsson Date: Fri, 6 Oct 2023 15:34:42 +0200 Subject: [PATCH] avoid progressbar when loading a single file --- sed/loader/flash/loader.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/sed/loader/flash/loader.py b/sed/loader/flash/loader.py index 8214a70c..7ab6c7ba 100644 --- a/sed/loader/flash/loader.py +++ b/sed/loader/flash/loader.py @@ -711,7 +711,7 @@ def fill_na( ) # Forward fill between consecutive dataframes - for i in tqdm(range(1, len(dataframes)), desc='Filling NaNs', leave=True, total=len(dataframes)-1): + for i in tqdm(range(1, len(dataframes)), desc='Filling NaNs', leave=True, total=len(dataframes)-1, disable=len(dataframes)==1): # Select pulse channels from current dataframe subset = dataframes[i][channels] # Find columns with NaN values in the first row @@ -788,6 +788,7 @@ def parquet_handler( data_parquet_dir, detector, ) + print(parquet_filenames) # Read all parquet files using dask and concatenate into one dataframe after filling dataframe = self.fill_na(