Merge pull request #2 from jreanah/add_type_hints

Added Union for optional types

Author: jreanah

adafruit_display_notification/__init__.py

@@ -16,7 +16,7 @@
 
 # For older versions, typing may need to be imported
 try:
-    from typing import List, Tuple
+    from typing import List, Tuple, Optional, Union
 except ImportError:
     pass
 
@@ -31,7 +31,8 @@
 class NotificationFree(displayio.Group):
     """Widget to show when no notifications are active."""
 
-    def __init__(self, width: int, height: int, *, dark_mode=True):
+    def __init__(self, width: Union[int, float], height: Union[int, float], *, 
+                 dark_mode: Union[bool, int]=True):
         # pylint: disable=unused-argument
         super().__init__()
 
@@ -50,7 +51,8 @@ def __init__(self, width: int, height: int, *, dark_mode=True):
 class PlainNotification(displayio.Group):
     """Plain text widget with a title and message."""
 
-    def __init__(self, title: str, message: str, width: int, height: int, *, dark_mode: bool=True):
+    def __init__(self, title: str, message: str, width: Union[int, float], height: Union[int, float],
+                  *, dark_mode: Union[bool, int]=True):
         super().__init__()
 
         # Set text, font, and color

adafruit_display_notification/apple.py

@@ -13,12 +13,17 @@
 
 from . import PlainNotification
 
+try:
+    from typing import Union
+except ImportError:
+    pass
+
 __version__ = "0.0.0+auto.0"
 __repo__ = "https://github.com/adafruit/Adafruit_CircuitPython_Display_Notification.git"
 
 
 def create_notification_widget(
-    notification: object, max_width: int, max_height: int, 
+    notification: object, max_width: Union[int, float], max_height: Union[int, float], 
     *, color_count: int=2**16):
     """
         Creates a notification widget for the given Apple notification.