bug fix in save_fit

MarkusSifft · Sep 15, 2023 · dcba793 · dcba793
1 parent 5eda866
commit dcba793
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/signalsnap/spectrum_calculator.py b/src/signalsnap/spectrum_calculator.py
@@ -799,8 +799,11 @@ def save_spec(self, save_path):
         self.S_gpu = None
         self.S_err_gpu = None
         self.main_data = None
-        self.config.corr_data = None
-        self.config.data = None
+        # Only set to None if the attribute exists
+        if hasattr(self, 'config'):
+            self.config.corr_data = None
+            self.config.data = None
+
         self.S_errs = None
         self.S_stationarity_temp = None
         pickle_save(save_path, self)