diff --git a/src/cal_tools/plotting.py b/src/cal_tools/plotting.py index 879dee04ff555b80002fe9b6cc0efe8d62216c81..838e1fc176fbbc8f5c564da1cbdf5e4092442f4a 100644 --- a/src/cal_tools/plotting.py +++ b/src/cal_tools/plotting.py @@ -392,9 +392,15 @@ def show_processed_modules_jungfrau( ax.set_axis_off() ax.get_legend().set_visible(False) + modules = [c for c in ax.collections if isinstance(c, PatchCollection)][0] + + # Remove non-tiles markings from figure + for coll in ax.collections: + if coll is not modules: + coll.remove() + # Set each module colour individually, # extra_geom provides a single color for all modules. - modules, = ax.collections = ax.collections[:1] facecolors = np.repeat(modules.get_facecolor(), len(expected_modules), axis=0) counter = 0 # Used as index within the `Noise` matrix