diff --git a/src/scgenerator/spectra.py b/src/scgenerator/spectra.py index 82a26d6..7be6999 100644 --- a/src/scgenerator/spectra.py +++ b/src/scgenerator/spectra.py @@ -125,8 +125,7 @@ class Spectrum(np.ndarray): @property def center_of_gravity(self): - intensity = self.time_int - return np.sum(intensity * self.t) / np.sum(intensity) + return pulse.center_of_gravity(self.t, self.time_int) def mask_wl(self, pos: float, width: float) -> Spectrum: """