diff --git a/src/scgenerator/spectra.py b/src/scgenerator/spectra.py index ef9888d..78581a6 100644 --- a/src/scgenerator/spectra.py +++ b/src/scgenerator/spectra.py @@ -162,7 +162,9 @@ class Spectrum(np.ndarray): return pulse.g12(self, axis)[..., self.wl_order] def spectrogram(self, delays: Iterable[float], gate_width: float = 2e-13) -> np.ndarray: - return pulse.spectrogram(self.t, delays, self.time_amp, gate_width) + return np.fft.fftshift( + pulse.spectrogram(self.t, delays, self.time_amp, gate_width), axes=-1 + ) freq_int = afreq_int freq_amp = afreq_amp