Add Union type annotation to goose.py

python_payload/st3m/goose.py

@@ -17,7 +17,7 @@ if TYPE_CHECKING:
     class ABCBase(metaclass=ABCMeta):
         pass
 
-    from typing import List, Optional, Tuple, Dict, Any, Callable, Iterator, Generator
+    from typing import List, Optional, Tuple, Dict, Any, Callable, Iterator, Generator, Union
     from enum import Enum
 else:
     # We're in CPython or Micropython.
@@ -40,6 +40,7 @@ else:
             Callable,
             Iterator,
             Generator,
+            Union,
         )
         from enum import Enum
     except ImportError:
@@ -52,6 +53,7 @@ else:
         Callable = None
         Iterator = None
         Generator = None
+        Union = None
 
         class Enum:
             pass
@@ -70,4 +72,5 @@ __all__ = [
     "Callable",
     "Iterator",
     "Generator",
+    "Union",
 ]