Skip to content

Commit

Permalink
Change print when loading parquet data
Browse files Browse the repository at this point in the history
  • Loading branch information
steinnymir committed Nov 30, 2020
1 parent db2ee4c commit 9dac64e
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion processor/DldProcessor.py
Original file line number Diff line number Diff line change
Expand Up @@ -301,6 +301,7 @@ def readDataframes(self, fileName=None, path=None, format='parquet'):
else:
fileName = 'run{}'.format(self.runNumber)
fullName = path + fileName # TODO: test if naming is correct
print(f'Loading {format} data from {fileName}')

if format == 'parquet':
self.dd = dask.dataframe.read_parquet(fullName + "_el")
Expand All @@ -316,7 +317,7 @@ def readDataframes(self, fileName=None, path=None, format='parquet'):
self.ddMicrobunches = dask.dataframe.read_hdf(
fullName, '/microbunches', mode='r', chunksize=self.CHUNK_SIZE)
self.printRunOverview()
print(f'Loaded data form {format} file')
print(f'Loading complete.')

def appendDataframeParquet(self, fileName, path=None):
""" Append data to an existing dask Parquet dataframe.
Expand Down

0 comments on commit 9dac64e

Please sign in to comment.