🔨 refactor: update import of override for compatibility with Python 3.11

adriamontoto · Dec 21, 2024 · bc66353 · bc66353
1 parent 1a0e582
commit bc66353
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/object_mother_pattern/mothers/primitives/bool_mother.py b/object_mother_pattern/mothers/primitives/bool_mother.py
@@ -2,7 +2,12 @@
 BoolMother module.
 """
 
-from typing import override
+from sys import version_info
+
+if version_info >= (3, 12):
+    from typing import override  # pragma: no cover
+else:
+    from typing_extensions import override  # pragma: no cover
 
 from ..base_mother import BaseMother
 

diff --git a/object_mother_pattern/mothers/primitives/integer_mother.py b/object_mother_pattern/mothers/primitives/integer_mother.py
@@ -2,7 +2,12 @@
 IntegerMother module.
 """
 
-from typing import override
+from sys import version_info
+
+if version_info >= (3, 12):
+    from typing import override  # pragma: no cover
+else:
+    from typing_extensions import override  # pragma: no cover
 
 from ..base_mother import BaseMother