🐛 fix unavailable typing imports on python<3.11 (codemod) (#8)

Author: jorenham

src/numpy-stubs/__config__.pyi

@@ -1,6 +1,7 @@
 from enum import Enum
 from types import ModuleType
-from typing import Final, Literal as L, NotRequired, TypedDict, overload, type_check_only
+from typing import Final, Literal as L, TypedDict, overload, type_check_only
+from typing_extensions import NotRequired
 
 _CompilerConfigDictValue = TypedDict(
     "_CompilerConfigDictValue",

src/numpy-stubs/_array_api_info.pyi

@@ -1,5 +1,5 @@
-from typing import ClassVar, Literal, TypeAlias, TypeVar, TypedDict, final, overload, type_check_only
-from typing_extensions import Never
+from typing import ClassVar, Literal, TypeAlias, TypedDict, final, overload, type_check_only
+from typing_extensions import Never, TypeVar
 
 import numpy as np
 

src/numpy-stubs/_core/_asarray.pyi

@@ -1,5 +1,6 @@
 from collections.abc import Iterable
-from typing import Any, Literal, TypeAlias, TypeVar, overload
+from typing import Any, Literal, TypeAlias, overload
+from typing_extensions import TypeVar
 
 from numpy._typing import DTypeLike, NDArray, _SupportsArrayFunc
 

src/numpy-stubs/_core/defchararray.pyi

@@ -1,4 +1,5 @@
-from typing import Any, Literal as L, SupportsIndex, SupportsInt, TypeAlias, TypeVar, overload
+from typing import Any, Literal as L, SupportsIndex, SupportsInt, TypeAlias, overload
+from typing_extensions import TypeVar
 
 import numpy as np
 from numpy._core.multiarray import compare_chararrays

src/numpy-stubs/_core/einsumfunc.pyi

@@ -1,5 +1,6 @@
 from collections.abc import Sequence
-from typing import Any, Literal, TypeAlias, TypeVar, overload
+from typing import Any, Literal, TypeAlias, overload
+from typing_extensions import TypeVar
 
 import numpy as np
 from numpy._typing import (

src/numpy-stubs/_core/fromnumeric.pyi

@@ -1,6 +1,6 @@
 from collections.abc import Sequence
-from typing import Any, Literal, NoReturn, Protocol, SupportsIndex, TypeAlias, TypeVar, overload, type_check_only
-from typing_extensions import Never, deprecated
+from typing import Any, Literal, NoReturn, Protocol, SupportsIndex, TypeAlias, overload, type_check_only
+from typing_extensions import Never, TypeVar, deprecated
 
 import numpy as np
 from numpy._typing import (

src/numpy-stubs/_core/function_base.pyi

@@ -1,4 +1,5 @@
-from typing import Any, Literal as L, SupportsIndex, TypeVar, overload
+from typing import Any, Literal as L, SupportsIndex, overload
+from typing_extensions import TypeVar
 
 import numpy as np
 from numpy._typing import DTypeLike, NDArray, _ArrayLikeComplex_co, _ArrayLikeFloat_co, _DTypeLike

src/numpy-stubs/_core/multiarray.pyi

@@ -9,13 +9,12 @@ from typing import (
     Protocol,
     SupportsIndex,
     TypeAlias,
-    TypeVar,
     TypedDict,
     final,
     overload,
     type_check_only,
 )
-from typing_extensions import CapsuleType, Unpack
+from typing_extensions import CapsuleType, TypeVar, Unpack
 
 import numpy as np
 from numpy import (  # noqa: ICN003

src/numpy-stubs/_core/numeric.pyi

@@ -7,10 +7,9 @@ from typing import (
     SupportsAbs,
     SupportsIndex,
     TypeAlias,
-    TypeVar,
     overload,
 )
-from typing_extensions import TypeIs, Unpack
+from typing_extensions import TypeIs, TypeVar, Unpack
 
 import numpy as np
 from numpy import (  # noqa: ICN003

src/numpy-stubs/_core/records.pyi

@@ -1,7 +1,8 @@
 from _typeshed import Incomplete, StrOrBytesPath
 from collections.abc import Iterable, Sequence
 from types import EllipsisType
-from typing import Any, Literal, Protocol, SupportsIndex, TypeAlias, TypeVar, overload, type_check_only
+from typing import Any, Literal, Protocol, SupportsIndex, TypeAlias, overload, type_check_only
+from typing_extensions import TypeVar
 
 import numpy as np
 from numpy._typing import (

src/numpy-stubs/_core/shape_base.pyi

@@ -1,5 +1,6 @@
 from collections.abc import Sequence
-from typing import Any, SupportsIndex, TypeVar, overload
+from typing import Any, SupportsIndex, overload
+from typing_extensions import TypeVar
 
 import numpy as np
 from numpy._typing import ArrayLike, DTypeLike, NDArray, _ArrayLike, _DTypeLike

src/numpy-stubs/_typing/_array_like.pyi

@@ -1,6 +1,7 @@
 import sys
 from collections.abc import Callable, Collection, Sequence
-from typing import Any, Never as _UnknownType, Protocol, TypeAlias, TypeVar, runtime_checkable
+from typing import Any, Never as _UnknownType, Protocol, TypeAlias, runtime_checkable
+from typing_extensions import TypeVar
 
 import numpy as np
 from numpy.dtypes import StringDType

src/numpy-stubs/_typing/_callable.pyi

@@ -3,11 +3,11 @@ from typing import (
     NoReturn,
     Protocol,
     TypeAlias,
-    TypeVar,
     final,
     overload,
     type_check_only,
 )
+from typing_extensions import TypeVar
 
 import numpy as np
 

src/numpy-stubs/_typing/_dtype_like.pyi

@@ -1,5 +1,6 @@
 from collections.abc import Sequence
-from typing import Any, Protocol, TypeAlias, TypeVar, TypedDict, runtime_checkable
+from typing import Any, Protocol, TypeAlias, TypedDict, runtime_checkable
+from typing_extensions import TypeVar
 
 import numpy as np
 

src/numpy-stubs/_typing/_nested_sequence.pyi

@@ -1,5 +1,6 @@
 from collections.abc import Iterator
-from typing import Any, Protocol, TypeVar
+from typing import Any, Protocol
+from typing_extensions import TypeVar
 
 __all__ = ["_NestedSequence"]
 

src/numpy-stubs/_typing/_ufunc.pyi

@@ -2,17 +2,15 @@ from typing import (
     Any,
     Generic,
     Literal,
-    LiteralString,
     NoReturn,
     Protocol,
     SupportsIndex,
     TypeAlias,
-    TypeVar,
     TypedDict,
     overload,
     type_check_only,
 )
-from typing_extensions import Unpack
+from typing_extensions import LiteralString, TypeVar, Unpack
 
 import numpy as np
 import numpy.typing as npt

src/numpy-stubs/ctypeslib/_ctypeslib.pyi

@@ -1,7 +1,8 @@
 import ctypes as ct
 from _typeshed import StrOrBytesPath
 from collections.abc import Iterable, Sequence
-from typing import Any, ClassVar, Generic, Literal as L, TypeAlias, TypeVar, overload
+from typing import Any, ClassVar, Generic, Literal as L, TypeAlias, overload
+from typing_extensions import TypeVar
 
 import numpy as np
 from numpy._core._internal import _ctypes

src/numpy-stubs/dtypes.pyi

@@ -3,14 +3,12 @@ from typing import (
     Final,
     Generic,
     Literal as L,
-    LiteralString,
     NoReturn,
-    Self,
     TypeAlias,
     final,
     type_check_only,
 )
-from typing_extensions import TypeVar
+from typing_extensions import LiteralString, Self, TypeVar
 
 import numpy as np
 

src/numpy-stubs/fft/_helper.pyi

@@ -1,4 +1,5 @@
-from typing import Any, Literal as L, TypeVar, overload
+from typing import Any, Literal as L, overload
+from typing_extensions import TypeVar
 
 import numpy as np
 from numpy._typing import ArrayLike, NDArray, _ArrayLike, _ArrayLikeComplex_co, _ArrayLikeFloat_co, _ShapeLike

src/numpy-stubs/lib/_arraypad_impl.pyi

@@ -1,5 +1,6 @@
 from collections.abc import Mapping
-from typing import Any, Literal as L, Protocol, TypeAlias, TypeVar, overload, type_check_only
+from typing import Any, Literal as L, Protocol, TypeAlias, overload, type_check_only
+from typing_extensions import TypeVar
 
 import numpy as np
 from numpy._typing import ArrayLike, NDArray, _ArrayLike, _ArrayLikeInt

src/numpy-stubs/lib/_arraysetops_impl.pyi

@@ -1,5 +1,5 @@
-from typing import Any, Generic, Literal as L, NamedTuple, SupportsIndex, TypeVar, overload
-from typing_extensions import deprecated
+from typing import Any, Generic, Literal as L, NamedTuple, SupportsIndex, overload
+from typing_extensions import TypeVar, deprecated
 
 import numpy as np
 from numpy._typing import (

src/numpy-stubs/lib/_arrayterator_impl.pyi

@@ -1,6 +1,7 @@
 from collections.abc import Generator, Iterator
 from types import EllipsisType
-from typing import Any, TypeAlias, TypeVar
+from typing import Any, TypeAlias
+from typing_extensions import TypeVar
 
 import numpy as np
 

src/numpy-stubs/lib/_function_base_impl.pyi

@@ -3,16 +3,14 @@ from typing import (
     Any,
     Concatenate,
     Literal as L,
-    ParamSpec,
     Protocol,
     SupportsIndex,
     SupportsInt,
     TypeAlias,
-    TypeVar,
     overload,
     type_check_only,
 )
-from typing_extensions import TypeIs, deprecated
+from typing_extensions import ParamSpec, TypeIs, TypeVar, deprecated
 
 import numpy as np
 from numpy import vectorize  # noqa: ICN003

src/numpy-stubs/lib/_npyio_impl.pyi

@@ -3,8 +3,8 @@ import zipfile
 from _typeshed import StrOrBytesPath, StrPath, SupportsKeysAndGetItem, SupportsRead, SupportsWrite
 from collections.abc import Callable, Collection, Iterable, Iterator, Mapping, Sequence
 from re import Pattern
-from typing import IO, Any, Generic, Literal as L, Protocol, TypeVar, overload, type_check_only
-from typing_extensions import Self, deprecated
+from typing import IO, Any, Generic, Literal as L, Protocol, overload, type_check_only
+from typing_extensions import Self, TypeVar, deprecated
 
 import numpy as np
 from numpy._core.multiarray import packbits, unpackbits

src/numpy-stubs/lib/_polynomial_impl.pyi

@@ -1,4 +1,5 @@
-from typing import Any, Literal as L, NoReturn, SupportsIndex, SupportsInt, TypeAlias, TypeVar, overload
+from typing import Any, Literal as L, NoReturn, SupportsIndex, SupportsInt, TypeAlias, overload
+from typing_extensions import TypeVar
 
 import numpy as np
 from numpy import poly1d  # noqa: ICN003

src/numpy-stubs/lib/_shape_base_impl.pyi

@@ -1,5 +1,6 @@
 from collections.abc import Callable, Sequence
-from typing import Any, Concatenate, ParamSpec, Protocol, SupportsIndex, TypeVar, overload, type_check_only
+from typing import Any, Concatenate, Protocol, SupportsIndex, overload, type_check_only
+from typing_extensions import ParamSpec, TypeVar
 
 import numpy as np
 from numpy._core.shape_base import vstack as row_stack

src/numpy-stubs/lib/_stride_tricks_impl.pyi

@@ -1,5 +1,6 @@
 from collections.abc import Iterable
-from typing import Any, SupportsIndex, TypeVar, overload
+from typing import Any, SupportsIndex, overload
+from typing_extensions import TypeVar
 
 import numpy as np
 from numpy._typing import ArrayLike, NDArray, _ArrayLike, _Shape, _ShapeLike

src/numpy-stubs/lib/_twodim_base_impl.pyi

@@ -1,5 +1,6 @@
 from collections.abc import Callable, Sequence
-from typing import Any, Literal as L, TypeAlias, TypeVar, overload
+from typing import Any, Literal as L, TypeAlias, overload
+from typing_extensions import TypeVar
 
 import numpy as np
 from numpy._typing import (

src/numpy-stubs/lib/_type_check_impl.pyi

@@ -1,5 +1,6 @@
 from collections.abc import Container, Iterable
-from typing import Any, Literal as L, TypeVar, overload
+from typing import Any, Literal as L, overload
+from typing_extensions import TypeVar
 
 import numpy as np
 from numpy._typing import ArrayLike, NBitBase, NDArray, _64Bit, _ArrayLike, _ScalarLike_co, _SupportsDType

src/numpy-stubs/lib/_ufunclike_impl.pyi

@@ -1,4 +1,5 @@
-from typing import Any, TypeVar, overload
+from typing import Any, overload
+from typing_extensions import TypeVar
 
 import numpy as np
 from numpy._typing import NDArray, _ArrayLikeFloat_co, _ArrayLikeObject_co, _FloatLike_co

src/numpy-stubs/linalg/_linalg.pyi

@@ -6,9 +6,9 @@ from typing import (
     SupportsIndex,
     SupportsInt,
     TypeAlias,
-    TypeVar,
     overload,
 )
+from typing_extensions import TypeVar
 
 import numpy as np
 from numpy import vecdot  # noqa: ICN003

src/numpy-stubs/ma/core.pyi

@@ -1,7 +1,7 @@
 from _typeshed import Incomplete
 from collections.abc import Callable
-from typing import Any, TypeVar
-from typing_extensions import Self
+from typing import Any
+from typing_extensions import Self, TypeVar
 
 import numpy as np
 from numpy import amax, amin, angle, bool_, clip, expand_dims, indices, squeeze  # noqa: ICN003

src/numpy-stubs/ma/mrecords.pyi

@@ -1,5 +1,6 @@
 from _typeshed import Incomplete
-from typing import Any, TypeVar
+from typing import Any
+from typing_extensions import TypeVar
 
 import numpy as np
 

src/numpy-stubs/polynomial/_polybase.pyi

@@ -8,14 +8,12 @@ from typing import (
     Final,
     Generic,
     Literal,
-    LiteralString,
-    Self,
     SupportsIndex,
     TypeAlias,
     TypeGuard,
     overload,
 )
-from typing_extensions import TypeVar
+from typing_extensions import LiteralString, Self, TypeVar
 
 import numpy as np
 import numpy.typing as npt

src/numpy-stubs/polynomial/_polytypes.pyi

@@ -2,17 +2,15 @@ from collections.abc import Callable, Sequence
 from typing import (
     Any,
     Literal,
-    LiteralString,
     NoReturn,
     Protocol,
-    Self,
     SupportsIndex,
     SupportsInt,
     TypeAlias,
     overload,
     type_check_only,
 )
-from typing_extensions import TypeVar
+from typing_extensions import LiteralString, Self, TypeVar
 
 import numpy as np
 import numpy.typing as npt

src/numpy-stubs/polynomial/chebyshev.pyi

@@ -4,9 +4,9 @@ from typing import (
     Concatenate,
     Final,
     Literal as L,
-    TypeVar,
     overload,
 )
+from typing_extensions import TypeVar
 
 import numpy as np
 import numpy.typing as npt

src/numpy-stubs/polynomial/hermite.pyi

@@ -1,4 +1,5 @@
-from typing import Any, Final, Literal as L, TypeVar
+from typing import Any, Final, Literal as L
+from typing_extensions import TypeVar
 
 import numpy as np
 

src/numpy-stubs/polynomial/hermite_e.pyi

@@ -1,4 +1,5 @@
-from typing import Any, Final, Literal as L, TypeVar
+from typing import Any, Final, Literal as L
+from typing_extensions import TypeVar
 
 import numpy as np
 

src/numpy-stubs/polynomial/polyutils.pyi

@@ -5,9 +5,9 @@ from typing import (
     Literal,
     SupportsIndex,
     TypeAlias,
-    TypeVar,
     overload,
 )
+from typing_extensions import TypeVar
 
 import numpy as np
 import numpy.typing as npt