From 7eb2f54e0ed556e0c30a4e14ded75e32dcf3d531 Mon Sep 17 00:00:00 2001 From: wakonig_k Date: Wed, 4 Jun 2025 14:04:04 +0200 Subject: [PATCH] fix(image layer): add layer main if it does not exist --- bec_widgets/widgets/plots/image/image_base.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/bec_widgets/widgets/plots/image/image_base.py b/bec_widgets/widgets/plots/image/image_base.py index a27bc6e8..417347b9 100644 --- a/bec_widgets/widgets/plots/image/image_base.py +++ b/bec_widgets/widgets/plots/image/image_base.py @@ -223,6 +223,9 @@ class ImageLayerManager: """ if not isinstance(name, str): raise TypeError("name must be a string") + if name == "main" and name not in self.layers: + # If 'main' is requested, create a default layer if it doesn't exist + return self.add(name=name, z_position="top") return self.layers[name] def __len__(self) -> int: