Skip to content

Commit c57c34c

Browse files
committed
Simplify rasterio rowcol operation typing
1 parent afdb012 commit c57c34c

1 file changed

Lines changed: 4 additions & 62 deletions

File tree

‎stubs/rasterio/rasterio/transform.pyi‎

Lines changed: 4 additions & 62 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ from rasterio.rpc import RPC
1313

1414
_Sextuple: TypeAlias = tuple[float, float, float, float, float, float]
1515
_OffsetOptions: TypeAlias = Literal["center", "ul", "ur", "ll", "lr"]
16-
_T = TypeVar("_T")
16+
_T = TypeVar("_T", default=int)
1717
_RoundOperation: TypeAlias = Callable[[float], _T]
1818

1919
IDENTITY: Final[Affine]
@@ -54,39 +54,15 @@ def xy(
5454
offset: _OffsetOptions = "center",
5555
**rpc_options: _GDALOption,
5656
) -> tuple[float, float] | tuple[list[float], list[float]]: ...
57-
58-
@overload
5957
def rowcol(
6058
transform: Affine | Sequence[GroundControlPoint] | RPC,
6159
xs: float | Sequence[float],
6260
ys: float | Sequence[float],
6361
zs: float | Sequence[float] | None = None,
64-
op: None = None,
65-
precision: int | None = None,
66-
**rpc_options: _GDALOption,
67-
) -> tuple[int, int] | tuple[list[int], list[int]]: ...
68-
@overload
69-
def rowcol(
70-
transform: Affine | Sequence[GroundControlPoint] | RPC,
71-
xs: float | Sequence[float],
72-
ys: float | Sequence[float],
73-
zs: float | Sequence[float] | None,
74-
op: _RoundOperation[_T],
62+
op: _RoundOperation[_T] | None = None,
7563
precision: int | None = None,
7664
**rpc_options: _GDALOption,
7765
) -> tuple[_T, _T] | tuple[list[_T], list[_T]]: ...
78-
@overload
79-
def rowcol(
80-
transform: Affine | Sequence[GroundControlPoint] | RPC,
81-
xs: float | Sequence[float],
82-
ys: float | Sequence[float],
83-
zs: float | Sequence[float] | None = None,
84-
*,
85-
op: _RoundOperation[_T],
86-
precision: int | None = None,
87-
**rpc_options: _GDALOption,
88-
) -> tuple[_T, _T] | tuple[list[_T], list[_T]]: ...
89-
9066
def get_transformer(
9167
transform: Affine | Sequence[GroundControlPoint] | RPC, **rpc_options: _GDALOption
9268
) -> type[TransformerBase]: ...
@@ -103,46 +79,13 @@ class TransformerBase:
10379
offset: _OffsetOptions = "center",
10480
) -> tuple[float, float] | tuple[list[float], list[float]]: ...
10581

106-
@overload
107-
def rowcol(
108-
self, xs: float | Sequence[float], ys: float | Sequence[float], zs: float | Sequence[float] | None = None, op: None = None
109-
) -> tuple[int, int] | tuple[list[int], list[int]]: ...
110-
@overload
111-
def rowcol(
112-
self,
113-
xs: float | Sequence[float],
114-
ys: float | Sequence[float],
115-
zs: float | Sequence[float] | None,
116-
op: _RoundOperation[_T],
117-
) -> tuple[_T, _T] | tuple[list[_T], list[_T]]: ...
11882
@overload
11983
def rowcol(
12084
self,
12185
xs: float | Sequence[float],
12286
ys: float | Sequence[float],
12387
zs: float | Sequence[float] | None = None,
124-
*,
125-
op: _RoundOperation[_T],
126-
) -> tuple[_T, _T] | tuple[list[_T], list[_T]]: ...
127-
@overload
128-
@deprecated("The `precision` parameter is unused since rasterio 1.3 and will be removed in 2.0.0.")
129-
def rowcol(
130-
self,
131-
xs: float | Sequence[float],
132-
ys: float | Sequence[float],
133-
zs: float | Sequence[float] | None = None,
134-
op: None = None,
135-
precision: int | None = None,
136-
) -> tuple[int, int] | tuple[list[int], list[int]]: ...
137-
@overload
138-
@deprecated("The `precision` parameter is unused since rasterio 1.3 and will be removed in 2.0.0.")
139-
def rowcol(
140-
self,
141-
xs: float | Sequence[float],
142-
ys: float | Sequence[float],
143-
zs: float | Sequence[float] | None,
144-
op: _RoundOperation[_T],
145-
precision: int | None = None,
88+
op: _RoundOperation[_T] | None = None,
14689
) -> tuple[_T, _T] | tuple[list[_T], list[_T]]: ...
14790
@overload
14891
@deprecated("The `precision` parameter is unused since rasterio 1.3 and will be removed in 2.0.0.")
@@ -151,8 +94,7 @@ class TransformerBase:
15194
xs: float | Sequence[float],
15295
ys: float | Sequence[float],
15396
zs: float | Sequence[float] | None = None,
154-
*,
155-
op: _RoundOperation[_T],
97+
op: _RoundOperation[_T] | None = None,
15698
precision: int | None = None,
15799
) -> tuple[_T, _T] | tuple[list[_T], list[_T]]: ...
158100

0 commit comments

Comments
 (0)