Fix annotation import in Python 3.8

machinable-org · May 24, 2024 · 6370d2b · 6370d2b
1 parent b387a94
commit 6370d2b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/machinable/config.py b/src/machinable/config.py
@@ -1,5 +1,5 @@
 __all__ = ["to_dict"]
-from typing import TYPE_CHECKING, Annotated, Any, Optional, Tuple, Union
+from typing import TYPE_CHECKING, Any, Optional, Tuple, Union
 
 import collections
 import re

diff --git a/tests/test_element.py b/tests/test_element.py
@@ -1,4 +1,11 @@
-from typing import Annotated, Optional
+from typing import Optional
+
+import sys
+
+if sys.version_info <= (3, 8):
+    from typing import Annotated
+else:
+    from typing_extensions import Annotated
 
 from uuid import uuid4