fix:import Self from typing vs. typing_extension

cunla · Apr 21, 2024 · f67cfa8 · f67cfa8
1 parent 05f9a41
commit f67cfa8
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/fakeredis/_server.py b/fakeredis/_server.py
@@ -1,6 +1,7 @@
 import inspect
 import logging
 import queue
+import sys
 import threading
 import time
 import uuid
@@ -9,10 +10,10 @@
 from collections import defaultdict
 from typing import Dict, Tuple, Any, List, Optional, Union, Set
 
-try:
-    from typing_extensions import Self
-except ImportError:
+if sys.version_info >= (3, 11):
     from typing import Self
+else:
+    from typing_extensions import Self
 
 import redis
 

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -45,6 +45,7 @@ include = [
 python = "^3.7"
 redis = ">=4"
 sortedcontainers = "^2"
+typing_extensions = { version = "^4.7", python = "<3.11" }
 lupa = { version = ">=1.14,<3.0", optional = true }
 jsonpath-ng = { version = "^1.6", optional = true }
 pyprobables = { version = "^0.6", optional = true }