Use fallback import

maldoinc · Sep 13, 2023 · 74fe957 · 74fe957
1 parent 445923b
commit 74fe957
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/test/test_container.py b/test/test_container.py
@@ -1,6 +1,5 @@
 import unittest
 from dataclasses import dataclass
-from typing_extensions import Annotated
 from unittest.mock import Mock, patch
 
 from test import services
@@ -13,6 +12,11 @@
 from wireup.ioc.parameter import ParameterBag, TemplatedString
 from wireup.ioc.util import find_classes_in_module
 
+try:
+    from typing import Annotated
+except ImportError:
+    from typing_extensions import Annotated
+
 
 class TestContainer(unittest.IsolatedAsyncioTestCase):
     def setUp(self) -> None:

diff --git a/test/test_util.py b/test/test_util.py
@@ -1,6 +1,5 @@
 import inspect
 import unittest
-from typing_extensions import Annotated
 
 from test import services
 from test.services.baz_service import BazService
@@ -14,6 +13,11 @@
     AnnotatedParameter,
 )
 
+try:
+    from typing import Annotated
+except ImportError:
+    from typing_extensions import Annotated
+
 
 class TestUtilityFunctions(unittest.TestCase):
     def test_find_classes_in_package(self):