diff --git a/rdstar/offline_analysis/fitting/dedicated_fit_approach/dedicated_fit_routine.py b/rdstar/offline_analysis/fitting/dedicated_fit_approach/dedicated_fit_routine.py index 6b1b5e10a1a032779f1f7afb4e9db149fe3c9509..c4ba31311cdb81a2a944cb09de6588c2857dae2a 100644 --- a/rdstar/offline_analysis/fitting/dedicated_fit_approach/dedicated_fit_routine.py +++ b/rdstar/offline_analysis/fitting/dedicated_fit_approach/dedicated_fit_routine.py @@ -1120,6 +1120,7 @@ class RDStarFitEvaluator: def plot_sig_ff_sys_shape_effects( self, fit_status_container: RDStarFitStatusContainer, + add_statistical_uncertainties: bool, ) -> Generator[SpecificShapePlotInfoContainer, None, None]: sys_shape_plotter: SystematicsShapePlotter = SystematicsShapePlotter( base_output_dir_path=self.base_output_dir_path, @@ -1139,11 +1140,15 @@ class RDStarFitEvaluator: plot_subsets=False, ) - yield from sys_shape_plotter.create_systematics_shape_plots(sys_shape_info=example_sys_shape_info) + yield from sys_shape_plotter.create_systematics_shape_plots( + sys_shape_info=example_sys_shape_info, + add_statistical_uncertainties=add_statistical_uncertainties, + ) def plot_norm_ff_sys_shape_effects( self, fit_status_container: RDStarFitStatusContainer, + add_statistical_uncertainties: bool, ) -> Generator[SpecificShapePlotInfoContainer, None, None]: sys_shape_plotter: SystematicsShapePlotter = SystematicsShapePlotter( base_output_dir_path=self.base_output_dir_path, @@ -1163,11 +1168,15 @@ class RDStarFitEvaluator: plot_subsets=False, ) - yield from sys_shape_plotter.create_systematics_shape_plots(sys_shape_info=example_sys_shape_info) + yield from sys_shape_plotter.create_systematics_shape_plots( + sys_shape_info=example_sys_shape_info, + add_statistical_uncertainties=add_statistical_uncertainties, + ) def plot_l_id_sys_shape_effects( self, fit_status_container: RDStarFitStatusContainer, + add_statistical_uncertainties: bool, ) -> Generator[SpecificShapePlotInfoContainer, None, None]: sys_shape_plotter: SystematicsShapePlotter = SystematicsShapePlotter( base_output_dir_path=self.base_output_dir_path, @@ -1187,11 +1196,15 @@ class RDStarFitEvaluator: plot_subsets=False, ) - yield from sys_shape_plotter.create_systematics_shape_plots(sys_shape_info=example_sys_shape_info) + yield from sys_shape_plotter.create_systematics_shape_plots( + sys_shape_info=example_sys_shape_info, + add_statistical_uncertainties=add_statistical_uncertainties, + ) def plot_tracking_sys_shape_effects( self, fit_status_container: RDStarFitStatusContainer, + add_statistical_uncertainties: bool, ) -> Generator[SpecificShapePlotInfoContainer, None, None]: sys_shape_plotter: SystematicsShapePlotter = SystematicsShapePlotter( base_output_dir_path=self.base_output_dir_path, @@ -1211,11 +1224,15 @@ class RDStarFitEvaluator: plot_subsets=False, ) - yield from sys_shape_plotter.create_systematics_shape_plots(sys_shape_info=example_sys_shape_info) + yield from sys_shape_plotter.create_systematics_shape_plots( + sys_shape_info=example_sys_shape_info, + add_statistical_uncertainties=add_statistical_uncertainties, + ) def plot_k_short_sys_shape_effects( self, fit_status_container: RDStarFitStatusContainer, + add_statistical_uncertainties: bool, ) -> Generator[SpecificShapePlotInfoContainer, None, None]: sys_shape_plotter: SystematicsShapePlotter = SystematicsShapePlotter( base_output_dir_path=self.base_output_dir_path, @@ -1235,7 +1252,10 @@ class RDStarFitEvaluator: plot_subsets=False, ) - yield from sys_shape_plotter.create_systematics_shape_plots(sys_shape_info=example_sys_shape_info) + yield from sys_shape_plotter.create_systematics_shape_plots( + sys_shape_info=example_sys_shape_info, + add_statistical_uncertainties=add_statistical_uncertainties, + ) def dump_data_for_external_fitter( self,