From d3c3fcb3ee90f0cfb82b04481c1cbc57b6cc3b49 Mon Sep 17 00:00:00 2001
From: Felix Metzner <felixmetzner@outlook.com>
Date: Thu, 18 Apr 2024 18:46:01 +0200
Subject: [PATCH] More bug fixes in axes handling.
---
.../fitting/dedicated_fit_approach/plotting_tools.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/rdstar/offline_analysis/fitting/dedicated_fit_approach/plotting_tools.py b/rdstar/offline_analysis/fitting/dedicated_fit_approach/plotting_tools.py
index b694b123a..716b8bd93 100644
--- a/rdstar/offline_analysis/fitting/dedicated_fit_approach/plotting_tools.py
+++ b/rdstar/offline_analysis/fitting/dedicated_fit_approach/plotting_tools.py
@@ -607,7 +607,11 @@ class SystematicsShapePlotter:
)
axis_index_pair: Tuple[int, int] = comp_to_axis_map[component.name]
- this_axis: AxesType = axes[axis_index_pair[0], axis_index_pair[1]]
+
+ if len(axes.shape) == 2:
+ this_axis: AxesType = axes[axis_index_pair[0], axis_index_pair[1]]
+ else:
+ this_axis = axes[axis_index_pair[1]]
self._plot_shape_overview(
ax=this_axis,
--
GitLab