|
@@ -164,7 +164,7 @@ from itertools import zip_longest
|
|
|
|
|
|
from numpy.typing import ArrayLike
|
|
|
|
|
|
-if sys.version_info >= (3, 8):
|
|
|
+if (sys.version_info.major == 3) & (sys.version_info.minor >= 8):
|
|
|
from typing import Callable, Dict, Iterable, Union
|
|
|
else:
|
|
|
from typing_extensions import Callable, Dict, Iterable, Union
|