update event reset data for consistency

Author: TomDonoghue

specparam/objs/algorithm.py

@@ -194,7 +194,7 @@ def _reset_internal_settings(self):
             self._gauss_std_limits = None
 
 
-    # Note: this currently overrides basefit - but once modes are used, this can be dropped (I think)
+    # ToCheck: this currently overrides basefit - but once modes are used, this can be dropped (I think)
     def _reset_results(self, clear_results=False):
         """Set, or reset, results attributes to empty.
 

specparam/objs/base.py

@@ -425,9 +425,8 @@ def load(self, file_name, file_path=None, peak_org=None):
             self.convert_results(peak_org)
 
 
-    # TO CHECK - DOES THIS GO HERE?
-    def _reset_data_results(self, clear_freqs=False, clear_spectrum=False,
-                            clear_results=False, clear_spectra=False):
+    def _reset_data_results(self, clear_freqs=False, clear_spectrum=False, clear_results=False,
+                            clear_spectra=False, clear_spectrograms=False):
         """Set, or reset, data & results attributes to empty.
 
         Parameters
@@ -440,6 +439,8 @@ def _reset_data_results(self, clear_freqs=False, clear_spectrum=False,
             Whether to clear model results attributes.
         clear_spectra : bool, optional, default: False
             Whether to clear power spectra attribute.
+        clear_spectrograms : bool, optional, default: False
+            Whether to clear spectrograms attribute.
         """
 
         self._reset_data(clear_freqs, clear_spectrum, clear_spectra)

specparam/objs/data.py

@@ -126,7 +126,7 @@ def set_check_modes(self, check_freqs=None, check_data=None):
 
 
     def _reset_data(self, clear_freqs=False, clear_spectrum=False):
-        """Set, or reset, data & results attributes to empty.
+        """Set, or reset, data attributes to empty.
 
         Parameters
         ----------
@@ -301,7 +301,7 @@ def plot(self, plt_log=False, **plt_kwargs):
 
 
     def _reset_data(self, clear_freqs=False, clear_spectrum=False, clear_spectra=False):
-        """Set, or reset, data & results attributes to empty.
+        """Set, or reset, data attributes to empty.
 
         Parameters
         ----------
@@ -452,3 +452,24 @@ def plot(self, event_ind):
         """Plot a selected spectrogram."""
 
         plot_spectrogram(self.freqs, self.spectrograms[event_ind, :, :], **plot_kwargs)
+
+
+    def _reset_data(self, clear_freqs=False, clear_spectrum=False,
+                    clear_spectra=False, clear_spectrograms=False):
+        """Set, or reset, data attributes to empty.
+
+        Parameters
+        ----------
+        clear_freqs : bool, optional, default: False
+            Whether to clear frequency attributes.
+        clear_spectrum : bool, optional, default: False
+            Whether to clear power spectrum attribute.
+        clear_spectra : bool, optional, default: False
+            Whether to clear power spectra attribute.
+        clear_spectrograms : bool, optional, default: False
+            Whether to clear spectrograms attribute.
+        """
+
+        super()._reset_data(clear_freqs, clear_spectrum, clear_spectra)
+        if clear_spectrograms:
+            self.spectrograms = None