BUG: Fixed an issue treating an entire parameter as frozen could raise (

#219)
nlesc-nano · Feb 24, 2021 · fa5452b · fa5452b
1 parent 86c856f
commit fa5452b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/FOX/armc/sanitization.py b/FOX/armc/sanitization.py
@@ -338,7 +338,10 @@ def get_param(dct: ParamMapping_) -> Tuple[ParamMapping, dict, dict, ValidationD
     if _sub_prm_dict_frozen is not None:
         for *_key, value in _get_prm(_sub_prm_dict_frozen):
             key = tuple(_key)
-            unit = data.loc[key[:2], 'unit'].iat[0]
+            try:
+                unit = data.loc[key[:2], 'unit'].iat[0]
+            except KeyError:
+                unit = ''
             data.loc[key, :] = [value, value, True, False, -np.inf, np.inf, 0, unit]
     data.sort_index(inplace=True)