Skip to content

Commit b2d572e

Browse files
committed
added missing overloads
1 parent 46a34be commit b2d572e

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

_duckdb-stubs/__init__.pyi

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -286,6 +286,7 @@ class DuckDBPyConnection:
286286
union_by_name: bool = False,
287287
compression: str | None = None,
288288
) -> DuckDBPyRelation: ...
289+
@pytyping.overload
289290
def from_parquet(
290291
self,
291292
file_globs: collections.abc.Sequence[str],
@@ -397,6 +398,7 @@ class DuckDBPyConnection:
397398
union_by_name: bool = False,
398399
compression: str | None = None,
399400
) -> DuckDBPyRelation: ...
401+
@pytyping.overload
400402
def read_parquet(
401403
self,
402404
file_globs: collections.abc.Sequence[str],
@@ -812,6 +814,7 @@ class Expression:
812814
def __gt__(self, other: Expression) -> Expression: ...
813815
@pytyping.overload
814816
def __init__(self, arg0: str) -> None: ...
817+
@pytyping.overload
815818
def __init__(self, arg0: pytyping.Any) -> None: ...
816819
def __invert__(self) -> Expression: ...
817820
def __le__(self, other: Expression) -> Expression: ...
@@ -1011,6 +1014,7 @@ def LambdaExpression(lhs: Expression | str | tuple[str], rhs: Expression) -> Exp
10111014
def SQLExpression(expression: str) -> Expression: ...
10121015
@pytyping.overload
10131016
def StarExpression(*, exclude: Expression | str | tuple[str]) -> Expression: ...
1017+
@pytyping.overload
10141018
def StarExpression() -> Expression: ...
10151019
def aggregate(
10161020
df: pandas.DataFrame,
@@ -1032,6 +1036,7 @@ def array_type(
10321036
def arrow(
10331037
rows_per_batch: pytyping.SupportsInt = 1000000, *, connection: duckdb.DuckDBPyConnection | None = None
10341038
) -> pyarrow.lib.RecordBatchReader: ...
1039+
@pytyping.overload
10351040
def arrow(arrow_object: pytyping.Any, *, connection: duckdb.DuckDBPyConnection | None = None) -> DuckDBPyRelation: ...
10361041
def begin(*, connection: duckdb.DuckDBPyConnection | None = None) -> duckdb.DuckDBPyConnection: ...
10371042
def checkpoint(*, connection: duckdb.DuckDBPyConnection | None = None) -> duckdb.DuckDBPyConnection: ...
@@ -1062,6 +1067,7 @@ def default_connection() -> duckdb.DuckDBPyConnection: ...
10621067
def description(*, connection: duckdb.DuckDBPyConnection | None = None) -> list | None: ...
10631068
@pytyping.overload
10641069
def df(*, date_as_object: bool = False, connection: duckdb.DuckDBPyConnection | None = None) -> pandas.DataFrame: ...
1070+
@pytyping.overload
10651071
def df(df: pandas.DataFrame, *, connection: duckdb.DuckDBPyConnection | None = None) -> DuckDBPyRelation: ...
10661072
def distinct(df: pandas.DataFrame, *, connection: duckdb.DuckDBPyConnection | None = None) -> DuckDBPyRelation: ...
10671073
def dtype(type_str: str, *, connection: duckdb.DuckDBPyConnection | None = None) -> typing.DuckDBPyType: ...
@@ -1173,6 +1179,7 @@ def from_parquet(
11731179
compression: str | None = None,
11741180
connection: duckdb.DuckDBPyConnection | None = None,
11751181
) -> DuckDBPyRelation: ...
1182+
@pytyping.overload
11761183
def from_parquet(
11771184
file_globs: collections.abc.Sequence[str],
11781185
binary_as_string: bool = False,
@@ -1328,6 +1335,7 @@ def read_parquet(
13281335
compression: str | None = None,
13291336
connection: duckdb.DuckDBPyConnection | None = None,
13301337
) -> DuckDBPyRelation: ...
1338+
@pytyping.overload
13311339
def read_parquet(
13321340
file_globs: collections.abc.Sequence[str],
13331341
binary_as_string: bool = False,

0 commit comments

Comments
 (0)