Fix multiple stub issues (#323)

Fix type stubs for 7 issues:

- #131 
- #194 
- #209 
- #210 
- #260 
- #279 
- #314

Author: evertlammerts

_duckdb-stubs/__init__.pyi

@@ -9,8 +9,6 @@ if pytyping.TYPE_CHECKING:
     import polars
     import pandas
     import pyarrow.lib
-    import torch as pytorch
-    import tensorflow
     from collections.abc import Callable, Sequence, Mapping
     from duckdb import sqltypes, func
 
@@ -247,7 +245,7 @@ class DuckDBPyConnection:
     def from_arrow(self, arrow_object: object) -> DuckDBPyRelation: ...
     def from_csv_auto(
         self,
-        path_or_buffer: str | bytes | os.PathLike[str] | os.PathLike[bytes],
+        path_or_buffer: str | bytes | os.PathLike[str] | os.PathLike[bytes] | pytyping.IO[bytes],
         header: bool | int | None = None,
         compression: str | None = None,
         sep: str | None = None,
@@ -350,7 +348,7 @@ class DuckDBPyConnection:
     def query_progress(self) -> float: ...
     def read_csv(
         self,
-        path_or_buffer: str | bytes | os.PathLike[str],
+        path_or_buffer: str | bytes | os.PathLike[str] | os.PathLike[bytes] | pytyping.IO[bytes],
         header: bool | int | None = None,
         compression: str | None = None,
         sep: str | None = None,
@@ -456,8 +454,8 @@ class DuckDBPyConnection:
     ) -> sqltypes.DuckDBPyType: ...
     def table(self, table_name: str) -> DuckDBPyRelation: ...
     def table_function(self, name: str, parameters: object = None) -> DuckDBPyRelation: ...
-    def tf(self) -> dict[str, tensorflow.Tensor]: ...
-    def torch(self) -> dict[str, pytorch.Tensor]: ...
+    def tf(self) -> dict[str, pytyping.Any]: ...
+    def torch(self) -> dict[str, pytyping.Any]: ...
     def type(self, type_str: str) -> sqltypes.DuckDBPyType: ...
     def union_type(
         self, members: list[sqltypes.DuckDBPyType] | dict[str, sqltypes.DuckDBPyType]
@@ -478,7 +476,7 @@ class DuckDBPyRelation:
     def __getitem__(self, name: str) -> DuckDBPyRelation: ...
     def __len__(self) -> int: ...
     def aggregate(
-        self, aggr_expr: Expression | str | list[Expression], group_expr: Expression | str = ""
+        self, aggr_expr: str | Expression | list[Expression | str], group_expr: Expression | str = ""
     ) -> DuckDBPyRelation: ...
     def any_value(
         self, expression: str, groups: str = "", window_spec: str = "", projected_columns: str = ""
@@ -710,7 +708,7 @@ class DuckDBPyRelation:
     def sum(
         self, expression: str, groups: str = "", window_spec: str = "", projected_columns: str = ""
     ) -> DuckDBPyRelation: ...
-    def tf(self) -> dict[str, tensorflow.Tensor]: ...
+    def tf(self) -> dict[str, pytyping.Any]: ...
     def to_csv(
         self,
         file_name: str,
@@ -751,7 +749,7 @@ class DuckDBPyRelation:
     ) -> None: ...
     def to_table(self, table_name: str) -> None: ...
     def to_view(self, view_name: str, replace: bool = True) -> DuckDBPyRelation: ...
-    def torch(self) -> dict[str, pytorch.Tensor]: ...
+    def torch(self) -> dict[str, pytyping.Any]: ...
     def union(self, union_rel: DuckDBPyRelation) -> DuckDBPyRelation: ...
     def unique(self, unique_aggr: str) -> DuckDBPyRelation: ...
     def update(self, set: Expression | str, *, condition: Expression | str | None = None) -> None: ...
@@ -771,6 +769,7 @@ class DuckDBPyRelation:
     def write_csv(
         self,
         file_name: str,
+        *,
         sep: str | None = None,
         na_rep: str | None = None,
         header: bool | None = None,
@@ -790,6 +789,7 @@ class DuckDBPyRelation:
     def write_parquet(
         self,
         file_name: str,
+        *,
         compression: str | None = None,
         field_ids: ParquetFieldIdsType | pytyping.Literal["auto"] | None = None,
         row_group_size_bytes: str | int | None = None,
@@ -810,7 +810,7 @@ class DuckDBPyRelation:
     @property
     def description(self) -> pytyping.List[tuple[str, sqltypes.DuckDBPyType, None, None, None, None, None]]: ...
     @property
-    def dtypes(self) -> pytyping.List[str]: ...
+    def dtypes(self) -> pytyping.List[sqltypes.DuckDBPyType]: ...
     @property
     def shape(self) -> tuple[int, int]: ...
     @property
@@ -1067,7 +1067,7 @@ def DefaultExpression() -> Expression: ...
 def FunctionExpression(function_name: str, *args: Expression) -> Expression: ...
 def LambdaExpression(lhs: pytyping.Any, rhs: Expression) -> Expression: ...
 def SQLExpression(expression: str) -> Expression: ...
-def StarExpression(*, exclude: pytyping.Any = None) -> Expression: ...
+def StarExpression(*, exclude: list[str | Expression] | None = None) -> Expression: ...
 def aggregate(
     df: pandas.DataFrame,
     aggr_expr: Expression | list[Expression] | str | list[str],
@@ -1194,7 +1194,7 @@ def from_arrow(
     connection: DuckDBPyConnection | None = None,
 ) -> DuckDBPyRelation: ...
 def from_csv_auto(
-    path_or_buffer: str | bytes | os.PathLike[str],
+    path_or_buffer: str | bytes | os.PathLike[str] | os.PathLike[bytes] | pytyping.IO[bytes],
     header: bool | int | None = None,
     compression: str | None = None,
     sep: str | None = None,
@@ -1346,7 +1346,7 @@ def query_df(
 ) -> DuckDBPyRelation: ...
 def query_progress(*, connection: DuckDBPyConnection | None = None) -> float: ...
 def read_csv(
-    path_or_buffer: str | bytes | os.PathLike[str],
+    path_or_buffer: str | bytes | os.PathLike[str] | os.PathLike[bytes] | pytyping.IO[bytes],
     header: bool | int | None = None,
     compression: str | None = None,
     sep: str | None = None,
@@ -1475,9 +1475,9 @@ def table_function(
     *,
     connection: DuckDBPyConnection | None = None,
 ) -> DuckDBPyRelation: ...
-def tf(*, connection: DuckDBPyConnection | None = None) -> dict[str, tensorflow.Tensor]: ...
+def tf(*, connection: DuckDBPyConnection | None = None) -> dict[str, pytyping.Any]: ...
 def tokenize(query: str) -> list[tuple[int, token_type]]: ...
-def torch(*, connection: DuckDBPyConnection | None = None) -> dict[str, pytorch.Tensor]: ...
+def torch(*, connection: DuckDBPyConnection | None = None) -> dict[str, pytyping.Any]: ...
 def type(type_str: str, *, connection: DuckDBPyConnection | None = None) -> sqltypes.DuckDBPyType: ...
 def union_type(
     members: dict[str, sqltypes.DuckDBPyType] | list[sqltypes.DuckDBPyType],

_duckdb-stubs/_sqltypes.pyi

@@ -43,7 +43,7 @@ class DuckDBPyType:
     @pytyping.overload
     def __init__(self, obj: object) -> None: ...
     @property
-    def children(self) -> list[tuple[str, object]]: ...
+    def children(self) -> list[tuple[str, DuckDBPyType | int | list[str]]]: ...
     @property
     def id(self) -> str: ...