fix: setPlate

pymmcore-plus · Aug 9, 2024 · 11dbc94 · 11dbc94
1 parent f3266a5
commit 11dbc94
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/pymmcore_widgets/hcs/_plate_navigator_widget.py b/src/pymmcore_widgets/hcs/_plate_navigator_widget.py
@@ -412,8 +412,13 @@ def __init__(
         self._plate_view.positionChanged.connect(self._on_position_changed)
         self._preset_movements.toggled.connect(self._on_preset_movements_toggled)
 
-    def setPlate(self, plan: useq.WellPlate | useq.WellPlatePlan) -> None:
+    def setPlate(self, plan: useq.WellPlate | useq.WellPlatePlan | None) -> None:
         """Set the plate to be displayed."""
+        if plan is None:
+            self.clear()
+            self._plate_view._plan = None
+            return
+
         self._plate_view.drawPlate(plan)
 
     def clear(self) -> None: