Add typing to FeaturesScatterWidget

Author: dstansby

src/napari_matplotlib/base.py

@@ -1,5 +1,6 @@
 import os
 from pathlib import Path
+from typing import Tuple
 
 import matplotlib as mpl
 import napari
@@ -70,7 +71,7 @@ def __init__(self, napari_viewer: napari.viewer.Viewer):
     # Accept any number of input layers by default
     n_layers_input = Interval(None, None)
     # Accept any type of input layer by default
-    input_layer_types = (napari.layers.Layer,)
+    input_layer_types: Tuple[napari.layers.Layer, ...] = (napari.layers.Layer,)
 
     @property
     def n_selected_layers(self) -> int:

src/napari_matplotlib/scatter.py

@@ -116,7 +116,15 @@ def _get_data(self) -> Tuple[List[np.ndarray], str, str]:
 
 
 class FeaturesScatterWidget(ScatterBaseWidget):
-    n_layers_input = 1
+    n_layers_input = Interval(1, 1)
+    # All layers that have a .features attributes
+    input_layer_types = (
+        napari.layers.Labels,
+        napari.layers.Points,
+        napari.layers.Shapes,
+        napari.layers.Tracks,
+        napari.layers.Vectors,
+    )
 
     def __init__(
         self,
@@ -164,7 +172,8 @@ def _set_axis_keys(self, x_axis_key: str, y_axis_key: str):
         self._draw()
 
     def _get_valid_axis_keys(self, combo_widget=None) -> List[str]:
-        """Get the valid axis keys from the layer FeatureTable.
+        """
+        Get the valid axis keys from the layer FeatureTable.
 
         Returns
         -------