manage setattr() for dictionaries

CEA-MetroCarac · Feb 12, 2024 · 18e7ae1 · 18e7ae1
1 parent 59948f2
commit 18e7ae1
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/fitspy/spectrum.py b/fitspy/spectrum.py
@@ -163,9 +163,14 @@ def set_attributes(self, model_dict):
         if 'xtol' in keys:
             self.fit_params['xtol'] = model_dict.pop('xtol')
 
-        for key in vars(self).keys():
-            if key in keys:
-                setattr(self, key, model_dict[key])
+        for key, val in vars(self).items():
+            if isinstance(val, dict):
+                for key2 in val.keys():
+                    if key2 in model_dict[key].keys():
+                        val[key2] = model_dict[key][key2]
+            else:
+                if key in keys:
+                    setattr(self, key, model_dict[key])
 
         if 'peak_models' in keys:
             self.peak_index = itertools.count(start=1)