Skip to content

Commit

Permalink
make Self backward compatible
Browse files Browse the repository at this point in the history
  • Loading branch information
deanm0000 committed Oct 15, 2024
1 parent 73e59ff commit 66621b9
Show file tree
Hide file tree
Showing 15 changed files with 126 additions and 17 deletions.
12 changes: 10 additions & 2 deletions pyarrow-stubs/__lib_pxi/array.pyi
Original file line number Diff line number Diff line change
@@ -1,17 +1,25 @@
# mypy: disable-error-code="overload-overlap,misc,type-arg"

import datetime as dt
import sys

from collections.abc import Callable
from decimal import Decimal

if sys.version_info >= (3, 11):
from typing import Self
else:
from typing_extensions import Self
if sys.version_info >= (3, 10):
from typing import TypeAlias
else:
from typing_extensions import TypeAlias
from typing import (
Any,
Generic,
Iterable,
Iterator,
Literal,
Self,
TypeAlias,
TypeVar,
overload,
)
Expand Down
7 changes: 6 additions & 1 deletion pyarrow-stubs/__lib_pxi/error.pyi
Original file line number Diff line number Diff line change
@@ -1,4 +1,9 @@
from typing import Self
import sys

if sys.version_info >= (3, 11):
from typing import Self
else:
from typing_extensions import Self

class ArrowException(Exception): ...
class ArrowInvalid(ValueError, ArrowException): ...
Expand Down
13 changes: 12 additions & 1 deletion pyarrow-stubs/__lib_pxi/io.pyi
Original file line number Diff line number Diff line change
@@ -1,7 +1,18 @@
import sys

from collections.abc import Callable
from io import IOBase
from os import PathLike
from typing import Any, Literal, Self, SupportsIndex, TypeAlias, overload

if sys.version_info >= (3, 11):
from typing import Self
else:
from typing_extensions import Self
if sys.version_info >= (3, 10):
from typing import TypeAlias
else:
from typing_extensions import TypeAlias
from typing import Any, Literal, SupportsIndex, overload

from pyarrow._stubs_typing import Compression, SupportPyBuffer
from pyarrow.lib import MemoryPool, _Weakrefable
Expand Down
8 changes: 7 additions & 1 deletion pyarrow-stubs/__lib_pxi/ipc.pyi
Original file line number Diff line number Diff line change
@@ -1,7 +1,13 @@
import enum
import sys

from io import IOBase
from typing import Iterable, Iterator, Literal, Mapping, NamedTuple, Self

if sys.version_info >= (3, 11):
from typing import Self
else:
from typing_extensions import Self
from typing import Iterable, Iterator, Literal, Mapping, NamedTuple

import pandas as pd

Expand Down
12 changes: 11 additions & 1 deletion pyarrow-stubs/__lib_pxi/scalar.pyi
Original file line number Diff line number Diff line change
@@ -1,9 +1,19 @@
# mypy: disable-error-code="overload-overlap,misc,type-arg"
import collections.abc
import datetime as dt
import sys

from decimal import Decimal
from typing import Any, Generic, Iterator, Mapping, Self, TypeAlias, overload

if sys.version_info >= (3, 11):
from typing import Self
else:
from typing_extensions import Self
if sys.version_info >= (3, 10):
from typing import TypeAlias
else:
from typing_extensions import TypeAlias
from typing import Any, Generic, Iterator, Mapping, overload

import numpy as np

Expand Down
12 changes: 10 additions & 2 deletions pyarrow-stubs/__lib_pxi/table.pyi
Original file line number Diff line number Diff line change
@@ -1,8 +1,18 @@
# mypy: disable-error-code="overload-overlap,type-arg,misc"

import datetime as dt
import sys

from decimal import Decimal

if sys.version_info >= (3, 11):
from typing import Self
else:
from typing_extensions import Self
if sys.version_info >= (3, 10):
from typing import TypeAlias
else:
from typing_extensions import TypeAlias
from typing import (
Any,
Generator,
Expand All @@ -11,9 +21,7 @@ from typing import (
Iterator,
Literal,
Mapping,
Self,
Sequence,
TypeAlias,
TypeVar,
overload,
)
Expand Down
7 changes: 6 additions & 1 deletion pyarrow-stubs/__lib_pxi/tensor.pyi
Original file line number Diff line number Diff line change
@@ -1,6 +1,11 @@
# mypy: disable-error-code="import-untyped"

from typing import Self
import sys

if sys.version_info >= (3, 11):
from typing import Self
else:
from typing_extensions import Self

import numpy as np

Expand Down
8 changes: 7 additions & 1 deletion pyarrow-stubs/__lib_pxi/types.pyi
Original file line number Diff line number Diff line change
@@ -1,8 +1,14 @@
import datetime as dt
import sys

from collections.abc import Mapping
from decimal import Decimal
from typing import Any, Generic, Iterable, Iterator, Literal, Self, overload

if sys.version_info >= (3, 11):
from typing import Self
else:
from typing_extensions import Self
from typing import Any, Generic, Iterable, Iterator, Literal, overload

import numpy as np
import pandas as pd
Expand Down
8 changes: 7 additions & 1 deletion pyarrow-stubs/_dataset.pyi
Original file line number Diff line number Diff line change
@@ -1,4 +1,11 @@
import sys

from pathlib import Path

if sys.version_info >= (3, 11):
from typing import Self
else:
from typing_extensions import Self
from typing import (
IO,
Any,
Expand All @@ -7,7 +14,6 @@ from typing import (
Iterator,
Literal,
NamedTuple,
Self,
TypeVar,
overload,
)
Expand Down
7 changes: 6 additions & 1 deletion pyarrow-stubs/_flight.pyi
Original file line number Diff line number Diff line change
@@ -1,7 +1,12 @@
import asyncio
import enum
import sys

from typing import Generator, Generic, Iterable, Iterator, NamedTuple, Self, TypeVar
if sys.version_info >= (3, 11):
from typing import Self
else:
from typing_extensions import Self
from typing import Generator, Generic, Iterable, Iterator, NamedTuple, TypeVar

from typing_extensions import deprecated

Expand Down
12 changes: 11 additions & 1 deletion pyarrow-stubs/_fs.pyi
Original file line number Diff line number Diff line change
@@ -1,8 +1,18 @@
import datetime as dt
import enum
import sys

from abc import ABC, abstractmethod
from typing import Self, TypeAlias, Union, overload

if sys.version_info >= (3, 11):
from typing import Self
else:
from typing_extensions import Self
if sys.version_info >= (3, 10):
from typing import TypeAlias
else:
from typing_extensions import TypeAlias
from typing import Union, overload

from fsspec import AbstractFileSystem

Expand Down
12 changes: 11 additions & 1 deletion pyarrow-stubs/acero.pyi
Original file line number Diff line number Diff line change
@@ -1,4 +1,14 @@
from typing import Literal, Self, TypeAlias
import sys

if sys.version_info >= (3, 11):
from typing import Self
else:
from typing_extensions import Self
if sys.version_info >= (3, 10):
from typing import TypeAlias
else:
from typing_extensions import TypeAlias
from typing import Literal

from . import lib
from .compute import Expression, FunctionOptions
Expand Down
8 changes: 7 additions & 1 deletion pyarrow-stubs/interchange/dataframe.pyi
Original file line number Diff line number Diff line change
@@ -1,4 +1,10 @@
from typing import Any, Iterable, Self, Sequence
import sys

if sys.version_info >= (3, 11):
from typing import Self
else:
from typing_extensions import Self
from typing import Any, Iterable, Sequence

from pyarrow.interchange.column import _PyArrowColumn
from pyarrow.lib import RecordBatch, Table
Expand Down
8 changes: 7 additions & 1 deletion pyarrow-stubs/orc.pyi
Original file line number Diff line number Diff line change
@@ -1,4 +1,10 @@
from typing import IO, Literal, Self
import sys

if sys.version_info >= (3, 11):
from typing import Self
else:
from typing_extensions import Self
from typing import IO, Literal

from . import _orc
from ._fs import SupportedFileSystem
Expand Down
9 changes: 8 additions & 1 deletion pyarrow-stubs/parquet/core.pyi
Original file line number Diff line number Diff line change
@@ -1,5 +1,12 @@
import sys

from pathlib import Path
from typing import IO, Callable, Iterator, Literal, Self, Sequence

if sys.version_info >= (3, 11):
from typing import Self
else:
from typing_extensions import Self
from typing import IO, Callable, Iterator, Literal, Sequence

from pyarrow import _parquet
from pyarrow._compute import Expression
Expand Down

0 comments on commit 66621b9

Please sign in to comment.