Add missing type annotations

Author: tcfranks

adafruit_trellism4.py

@@ -32,6 +32,7 @@
 
 try:
     from typing import List, Optional, Tuple, Union
+    from typing_extensions import Literal
     from microcontroller import Pin
 except ImportError:
     pass
@@ -43,7 +44,14 @@
 class _NeoPixelArray:
     """Creates a NeoPixel array for use in the ``TrellisM4Express`` class."""
 
-    def __init__(self, pin: Pin, *, width: int, height: int, rotation: int = 0) -> None:
+    def __init__(
+        self,
+        pin: Pin,
+        *,
+        width: int,
+        height: int,
+        rotation: Literal[0, 90, 180, 270] = 0,
+    ) -> None:
         self._neopixel = neopixel.NeoPixel(pin, width * height, auto_write=True)
         if rotation % 90 != 0:
             raise ValueError("Only 90 degree rotations supported")
@@ -234,7 +242,7 @@ class TrellisM4Express:
             current_press = pressed
     """
 
-    def __init__(self, rotation: int = 0) -> None:
+    def __init__(self, rotation: Literal[0, 90, 180, 270] = 0) -> None:
         self._rotation = rotation
 
         # Define NeoPixels

requirements.txt

@@ -5,3 +5,4 @@
 Adafruit-Blinka
 adafruit-circuitpython-matrixkeypad
 adafruit-circuitpython-neopixel
+typing-extensions~=4.0