From 23f704780574d85ba47520fbf44b9c7ca8396ae4 Mon Sep 17 00:00:00 2001 From: jobdewitte Date: Thu, 23 Jan 2025 18:45:22 +0100 Subject: [PATCH] fixed import TypeAlias for python < 3.10 --- traceon/_typing/_array_like.py | 8 ++++++-- traceon/_typing/_backend.py | 10 ++++++++-- traceon/_typing/_geometric_types.py | 9 ++++++++- traceon/_typing/_mesh_types.py | 10 ++++++++-- 4 files changed, 30 insertions(+), 7 deletions(-) diff --git a/traceon/_typing/_array_like.py b/traceon/_typing/_array_like.py index c2dfdd5..19bad25 100644 --- a/traceon/_typing/_array_like.py +++ b/traceon/_typing/_array_like.py @@ -1,7 +1,11 @@ from __future__ import annotations - +import sys from collections.abc import Sequence -from typing import TypeAlias + +if sys.version_info >= (3, 10): + from typing import TypeAlias +else: + from typing_extensions import TypeAlias import numpy as np diff --git a/traceon/_typing/_backend.py b/traceon/_typing/_backend.py index bd5a03f..a72de54 100644 --- a/traceon/_typing/_backend.py +++ b/traceon/_typing/_backend.py @@ -1,7 +1,13 @@ from __future__ import annotations - -from typing import Literal, TypeAlias +import sys +from typing import Literal from collections.abc import Sequence + +if sys.version_info >= (3, 10): + from typing import TypeAlias +else: + from typing_extensions import TypeAlias + import numpy as np from ._array_like import _NumpyFloat diff --git a/traceon/_typing/_geometric_types.py b/traceon/_typing/_geometric_types.py index 776c222..b69dcb0 100644 --- a/traceon/_typing/_geometric_types.py +++ b/traceon/_typing/_geometric_types.py @@ -1,6 +1,13 @@ from __future__ import annotations -from typing import Literal, TypeAlias +import sys +from typing import Literal from collections.abc import Callable, Sequence + +if sys.version_info >= (3, 10): + from typing import TypeAlias +else: + from typing_extensions import TypeAlias + import numpy as np from ._array_like import _NumpyFloat diff --git a/traceon/_typing/_mesh_types.py b/traceon/_typing/_mesh_types.py index 34e75d9..a958f5e 100644 --- a/traceon/_typing/_mesh_types.py +++ b/traceon/_typing/_mesh_types.py @@ -1,7 +1,13 @@ from __future__ import annotations - -from typing import Literal, TypeAlias +import sys +from typing import Literal from collections.abc import Sequence, Mapping + +if sys.version_info >= (3, 10): + from typing import TypeAlias +else: + from typing_extensions import TypeAlias + import numpy as np from ._array_like import _NumpyFloat, _NumpyInt, ArrayInt1D