@@ -1841,13 +1841,15 @@ def compute_wrapper(
18411841 compressed_paths = []
18421842 mask_paths = []
18431843 masked_paths = []
1844+ waveplot_compressed_paths = []
1845+ waveplot_plots = []
18441846 if not fast_mode :
18451847 datas = [
18461848 (output_paths , 'jpg' , stft_db ),
18471849 ]
18481850 if plot_uncompressed_amplitude :
18491851 datas += [
1850- (output_paths , 'waveplot.jpg' , waveplot ),
1852+ (waveplot_plots , 'waveplot.jpg' , waveplot ),
18511853 ]
18521854 else :
18531855 datas = []
@@ -1857,7 +1859,7 @@ def compute_wrapper(
18571859 ]
18581860 if 'waveplot' in segments :
18591861 datas += [
1860- (compressed_paths , 'compressed.waveplot.jpg' , segments ['waveplot' ]),
1862+ (waveplot_compressed_paths , 'compressed.waveplot.jpg' , segments ['waveplot' ]),
18611863 ]
18621864
18631865 # Create masked image
@@ -1897,6 +1899,8 @@ def compute_wrapper(
18971899 'spectrogram' : {
18981900 'uncompressed.path' : output_paths ,
18991901 'compressed.path' : compressed_paths ,
1902+ 'waveplot.path' : waveplot_plots ,
1903+ 'waveplot.compressed.path' : waveplot_compressed_paths ,
19001904 'mask.path' : mask_paths ,
19011905 'masked.path' : masked_paths ,
19021906 },
0 commit comments