mypy passing

Author: evertlammerts

_duckdb-stubs/__init__.pyi

@@ -412,17 +412,21 @@ class DuckDBPyConnection:
     def register_filesystem(self, filesystem: fsspec.AbstractFileSystem) -> None: ...
     def remove_function(self, name: str) -> DuckDBPyConnection: ...
     def rollback(self) -> DuckDBPyConnection: ...
-    def row_type(self, fields: Expression | list[Expression] | str | list[str]) -> typing.DuckDBPyType: ...
+    def row_type(self, fields: dict[str, typing.DuckDBPyType] | list[typing.DuckDBPyType]) -> typing.DuckDBPyType: ...
     def sql(self, query: Statement | str, *, alias: str = "", params: object = None) -> DuckDBPyRelation: ...
     def sqltype(self, type_str: str) -> typing.DuckDBPyType: ...
     def string_type(self, collation: str = "") -> typing.DuckDBPyType: ...
-    def struct_type(self, fields: Expression | list[Expression] | str | list[str]) -> typing.DuckDBPyType: ...
+    def struct_type(
+        self, fields: dict[str, typing.DuckDBPyType] | list[typing.DuckDBPyType]
+    ) -> typing.DuckDBPyType: ...
     def table(self, table_name: str) -> DuckDBPyRelation: ...
     def table_function(self, name: str, parameters: object = None) -> DuckDBPyRelation: ...
     def tf(self) -> dict: ...
     def torch(self) -> dict: ...
     def type(self, type_str: str) -> typing.DuckDBPyType: ...
-    def union_type(self, members: typing.DuckDBPyType) -> typing.DuckDBPyType: ...
+    def union_type(
+        self, members: list[typing.DuckDBPyType] | dict[str, typing.DuckDBPyType]
+    ) -> typing.DuckDBPyType: ...
     def unregister(self, view_name: str) -> DuckDBPyConnection: ...
     def unregister_filesystem(self, name: str) -> None: ...
     def values(self, *args) -> DuckDBPyRelation: ...
@@ -1347,7 +1351,7 @@ def register_filesystem(
 def remove_function(name: str, *, connection: duckdb.DuckDBPyConnection | None = None) -> duckdb.DuckDBPyConnection: ...
 def rollback(*, connection: duckdb.DuckDBPyConnection | None = None) -> duckdb.DuckDBPyConnection: ...
 def row_type(
-    fields: Expression | list[Expression] | str | list[str],
+    fields: dict[str, typing.DuckDBPyType] | list[typing.DuckDBPyType],
     *,
     connection: duckdb.DuckDBPyConnection | None = None,
 ) -> typing.DuckDBPyType: ...
@@ -1363,7 +1367,7 @@ def sql(
 def sqltype(type_str: str, *, connection: duckdb.DuckDBPyConnection | None = None) -> typing.DuckDBPyType: ...
 def string_type(collation: str = "", *, connection: duckdb.DuckDBPyConnection | None = None) -> typing.DuckDBPyType: ...
 def struct_type(
-    fields: Expression | list[Expression] | str | list[str],
+    fields: dict[str, typing.DuckDBPyType] | list[typing.DuckDBPyType],
     *,
     connection: duckdb.DuckDBPyConnection | None = None,
 ) -> typing.DuckDBPyType: ...
@@ -1379,7 +1383,7 @@ def tokenize(query: str) -> list: ...
 def torch(*, connection: duckdb.DuckDBPyConnection | None = None) -> dict: ...
 def type(type_str: str, *, connection: duckdb.DuckDBPyConnection | None = None) -> typing.DuckDBPyType: ...
 def union_type(
-    members: typing.DuckDBPyType,
+    members: dict[str, typing.DuckDBPyType] | list[typing.DuckDBPyType],
     *,
     connection: duckdb.DuckDBPyConnection | None = None,
 ) -> typing.DuckDBPyType: ...

pyproject.toml

@@ -320,8 +320,8 @@ warn_unreachable = true
 pretty = true
 python_version = "3.9"
 exclude = [
-    "duckdb/experimental/",
-    "duckdb/query_graph/",
+    "duckdb/experimental/", # not checking the pyspark API
+    "duckdb/query_graph/",  # old and unmaintained (should probably remove)
 ]
 
 [[tool.mypy.overrides]]