fixed naming bug

This commit is contained in:
Beale John Henry
2023-12-10 14:40:40 +01:00
parent 4f8adce28e
commit 8fe51dc111

View File

@@ -26,13 +26,13 @@ def h5_sample( lst, sample ):
# read h5.lst - note - removes // from image column
print( "reading SwissFEL ls file" )
# scrub file name
lst_name = os.path.basename( lst )
lst_name = os.path.splitext( os.path.basename( lst ) )[0]
cols = [ "h5", "image" ]
sample_df = pd.read_csv( lst, sep="\s//", engine="python", names=cols )
print( "done" )
print( "taking {0} image sample".format( sample ) )
print( "taking {0} image sample of {1} image lst".format( sample, len( sample_df ) ) )
# take defined sample
sample_df = sample_df.sample( sample )
@@ -71,3 +71,4 @@ if __name__ == "__main__":