Import of Annotated and Literal from typing_extensions according to Python version

Author: Paul Duncan

predicthq/endpoints/v1/beam/schemas.py

@@ -2,9 +2,7 @@
 from datetime import datetime
 from predicthq.endpoints.schemas import ArgKwargResultSet
 from typing import Optional, List
-from typing import Annotated, Literal
 
-# Check if we're running on Python < 3.11
 # Python < 3.11 does not have StrEnum in the enum module
 import sys
 if sys.version_info < (3, 11):
@@ -15,6 +13,18 @@ class StrEnum(str, enum.Enum):
 else:
     from enum import StrEnum
 
+# Python < 3.9 does not have Annotated in the enum module
+if sys.version_info < (3, 9):
+    from typing_extensions import Annotated
+else:
+    from typing import Annotated, Literal
+
+# Python < 3.8 does not have Literal in the enum module
+if sys.version_info < (3, 8):
+    from typing_extensions import Literal
+else:
+    from typing import Literal
+
 
 class BeamPaginationResultSet(ArgKwargResultSet):
     def has_next(self):