Three state fitting

src/pychemelt/__init__.py

@@ -18,11 +18,13 @@
 from .utils.signals import (
     signal_two_state_tc_unfolding,
     signal_two_state_t_unfolding,
-    map_two_state_model_to_signal_fx
+    map_two_state_model_to_signal_fx,
+    map_three_state_model_to_signal_fx
 )
 
 from .utils.plotting import (
-    plot_unfolding
+    plot_unfolding,
+    plot_baselines,
 )
 
 

src/pychemelt/main.py

@@ -524,6 +524,9 @@ def estimate_baseline_parameters(
             'exponential': exponential_baseline
         }
 
+        self.window_range_native = window_range_native
+        self.window_range_unfolded = window_range_unfolded
+
         self.baseline_N_fx = baseline_fx_dic[native_baseline_type]
         self.baseline_U_fx = baseline_fx_dic[unfolded_baseline_type]