Update queryset_protocol.py

2020-11-01 18:02:25 +07:00
parent cce59acd99
commit 9bcf6a346a
1 changed files with 6 additions and 1 deletions
--- a/ormar/protocols/queryset_protocol.py
+++ b/ormar/protocols/queryset_protocol.py
@ -1,4 +1,9 @@
-from typing import Any, List, Optional, Protocol, Sequence, TYPE_CHECKING, Union
+from typing import Any, List, Optional, Sequence, TYPE_CHECKING, Union
+
+try:
+    from typing import Protocol
+except ImportError:
+    from typing_extentions import Protocol

 if TYPE_CHECKING:  # noqa: C901; #pragma nocover
    from ormar import QuerySet, Model