diff --git a/pipe/pipe.py b/pipe/pipe.py index 751babd..c791901 100644 --- a/pipe/pipe.py +++ b/pipe/pipe.py @@ -28,12 +28,12 @@ def spectrogram_pipe(specpath: Path, namepattern: str, freq_samples: int) freq_samples (int): Number of frequency samples in each spectrogram. Returns: - spectrograms (np.ndarray): Stack of spectrograms (2D array). + spectrogram (np.ndarray): Stack of spectrograms (2D array). """ with h5py.File(specpath/namepattern.format(i), 'r') as f: - spectrograms = f['spectrogram'][:] + spectrogram = f['spectrogram'][:] - return spectrograms + return spectrogram # EOF