MAINT: Add possibility to not add img (but background) and instructions.

pmgbergen · Nov 26, 2023 · 8221eff · 8221eff
1 parent 18135d3
commit 8221eff
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/darsia/assistants/point_selection_assistant.py b/src/darsia/assistants/point_selection_assistant.py
@@ -11,12 +11,16 @@
 
 class PointSelectionAssistant(darsia.BaseAssistant):
     def __init__(self, img: darsia.Image, **kwargs) -> None:
+        if img is None:
+            img = kwargs.get("background")
+            assert img is not None, "No image provided."
+
         # Only continue for 2d images
         assert img.space_dim == 2, "Only 2d images are supported."
 
         # Set name for titles in plots
-        self.name = "Point selection assistant"
-        """Name of assistant."""
+        self.name = kwargs.get("name", "Point selection assistant")
+        """Name of assistant / short version of instructions."""
 
         super().__init__(img, use_coordinates=False, **kwargs)