Also export Annotated

beartype · Jan 21, 2024 · 4b4e964 · 4b4e964
1 parent 727e968
commit 4b4e964
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 8 deletions.
diff --git a/plum/typing.py b/plum/typing.py
@@ -1,6 +1,6 @@
 import sys
 
-__all__ = ["get_type_hints", "get_origin", "get_args"]
+__all__ = ["get_type_hints", "get_origin", "get_args", "Annotated"]
 
 # Before Python 3.9, e.g. `get_type_hints` does not include the keyword argument
 # `include_extras`, so we take `get_type_hints` from `typing_extensions` instead,
@@ -12,10 +12,13 @@
     get_type_hints = typing_extensions.get_type_hints
     get_origin = typing_extensions.get_origin
     get_args = typing_extensions.get_args
+    Annotated = typing_extensions.Annotated
+
 
 else:  # pragma: specific no cover 3.8
     import typing
 
     get_type_hints = typing.get_type_hints
     get_origin = typing.get_origin
     get_args = typing.get_args
+    Annotated = typing.Annotated
diff --git a/tests/advanced/test_annotated.py b/tests/advanced/test_annotated.py
@@ -1,14 +1,8 @@
-import sys
-
-if sys.version_info < (3, 9):
-    from typing_extensions import Annotated
-else:
-    from typing import Annotated
-
 import pytest
 from beartype.vale import Is
 
 from plum import Dispatcher, NotFoundLookupError
+from plum.typing import Annotated
 
 
 def test_simple_annotated():