Update canvas.py

Support for closing AI polygon when pressing enter
wkentaro · Jun 13, 2024 · ba5594a · ba5594a
1 parent 4329343
commit ba5594a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/labelme/widgets/canvas.py b/labelme/widgets/canvas.py
@@ -540,7 +540,7 @@ def setHiding(self, enable=True):
         self._hideBackround = self.hideBackround if enable else False
 
     def canCloseShape(self):
-        return self.drawing() and self.current and len(self.current) > 2
+        return self.drawing() and ((self.current and len(self.current) > 2) or self.createMode in ["ai_polygon", "ai_mask"])
 
     def mouseDoubleClickEvent(self, ev):
         if self.double_click != "close":