Fix mypy issue

Author: jenshnielsen

plottr/node/autonode.py

@@ -44,22 +44,26 @@ def connectFloatSpinbox(
 
 
 class AutoNodeGui(AutoNodeGuiTemplate):
-
     widgetConnection = {
         int: connectIntegerSpinbox,
         float: connectFloatSpinbox,
     }
 
-    def __init__(self, parent: Optional[QtWidgets.QWidget] = None,
-                 node: Optional[Node] = None):
+    def __init__(
+        self, parent: Optional[QtWidgets.QWidget] = None, node: Optional[Node] = None
+    ):
         super().__init__(parent)
         layout = QtWidgets.QFormLayout()
         self.setLayout(layout)
 
     def addOption(self, name: str, specs: Dict[str, Any], confirm: bool) -> None:
-        optionType = specs.get('type', None)
+        optionType: type | None = specs.get("type", None)
         widget = None
-        func = self.widgetConnection.get(optionType, None)
+        func = (
+            self.widgetConnection[optionType]
+            if optionType in self.widgetConnection.keys()
+            else None
+        )
         if func is not None:
             widget = func(self, name, specs, confirm)
         layout = cast(QtWidgets.QFormLayout, self.layout())