Ruff ANN204: return type annotations

Author: evertlammerts

duckdb/bytes_io_wrapper.py

@@ -1,5 +1,5 @@
 from io import StringIO, TextIOBase
-from typing import Union
+from typing import Any, Union
 
 """
 BSD 3-Clause License
@@ -48,7 +48,7 @@ def __init__(self, buffer: Union[StringIO, TextIOBase], encoding: str = "utf-8")
         # overflow to the front of the bytestring the next time reading is performed
         self.overflow = b""
 
-    def __getattr__(self, attr: str):
+    def __getattr__(self, attr: str) -> Any:
         return getattr(self.buffer, attr)
 
     def read(self, n: Union[int, None] = -1) -> bytes:

duckdb/experimental/spark/_globals.py

@@ -32,6 +32,8 @@ def foo(arg=pyducdkb.spark._NoValue):
 Note that this approach is taken after from NumPy.
 """
 
+from typing import Type
+
 __ALL__ = ["_NoValue"]
 
 
@@ -54,23 +56,23 @@ class _NoValueType:
 
     __instance = None
 
-    def __new__(cls):
+    def __new__(cls) -> '_NoValueType':
         # ensure that only one instance exists
         if not cls.__instance:
             cls.__instance = super(_NoValueType, cls).__new__(cls)
         return cls.__instance
 
     # Make the _NoValue instance falsey
-    def __nonzero__(self):
+    def __nonzero__(self) -> bool:
         return False
 
     __bool__ = __nonzero__
 
     # needed for python 2 to preserve identity through a pickle
-    def __reduce__(self):
+    def __reduce__(self) -> tuple[Type, tuple]:
         return (self.__class__, ())
 
-    def __repr__(self):
+    def __repr__(self) -> str:
         return "<no value>"
 
 

duckdb/experimental/spark/conf.py

@@ -3,7 +3,7 @@
 
 
 class SparkConf:
-    def __init__(self):
+    def __init__(self) -> None:
         raise NotImplementedError
 
     def contains(self, key: str) -> bool:

duckdb/experimental/spark/context.py

@@ -7,7 +7,7 @@
 
 
 class SparkContext:
-    def __init__(self, master: str):
+    def __init__(self, master: str) -> None:
         self._connection = duckdb.connect(':memory:')
         # This aligns the null ordering with Spark.
         self._connection.execute("set default_null_order='nulls_first_on_asc_last_on_desc'")

duckdb/experimental/spark/errors/exceptions/base.py

@@ -14,7 +14,7 @@ def __init__(
         error_class: Optional[str] = None,
         # The dictionary listing the arguments specified in the message (or the error_class)
         message_parameters: Optional[dict[str, str]] = None,
-    ):
+    ) -> None:
         # `message` vs `error_class` & `message_parameters` are mutually exclusive.
         assert (message is not None and (error_class is None and message_parameters is None)) or (
             message is None and (error_class is not None and message_parameters is not None)

duckdb/experimental/spark/exception.py

@@ -5,7 +5,7 @@ class ContributionsAcceptedError(NotImplementedError):
     feel free to open up a PR or a Discussion over on https://github.com/duckdb/duckdb
     """
 
-    def __init__(self, message=None):
+    def __init__(self, message=None) -> None:
         doc = self.__class__.__doc__
         if message:
             doc = message + '\n' + doc

duckdb/experimental/spark/sql/catalog.py

@@ -33,7 +33,7 @@ class Function(NamedTuple):
 
 
 class Catalog:
-    def __init__(self, session: SparkSession):
+    def __init__(self, session: SparkSession) -> None:
         self._session = session
 
     def listDatabases(self) -> list[Database]:

duckdb/experimental/spark/sql/column.py

@@ -95,11 +95,11 @@ class Column:
     .. versionadded:: 1.3.0
     """
 
-    def __init__(self, expr: Expression):
+    def __init__(self, expr: Expression) -> None:
         self.expr = expr
 
     # arithmetic operators
-    def __neg__(self):
+    def __neg__(self) -> 'Column':
         return Column(-self.expr)
 
     # `and`, `or`, `not` cannot be overloaded in Python,

duckdb/experimental/spark/sql/conf.py

@@ -4,7 +4,7 @@
 
 
 class RuntimeConfig:
-    def __init__(self, connection: DuckDBPyConnection):
+    def __init__(self, connection: DuckDBPyConnection) -> None:
         self._connection = connection
 
     def set(self, key: str, value: str) -> None:

duckdb/experimental/spark/sql/dataframe.py

@@ -37,7 +37,7 @@
 
 
 class DataFrame:
-    def __init__(self, relation: duckdb.DuckDBPyRelation, session: "SparkSession"):
+    def __init__(self, relation: duckdb.DuckDBPyRelation, session: "SparkSession") -> None:
         self.relation = relation
         self.session = session
         self._schema = None
@@ -870,7 +870,7 @@ def limit(self, num: int) -> "DataFrame":
         rel = self.relation.limit(num)
         return DataFrame(rel, self.session)
 
-    def __contains__(self, item: str):
+    def __contains__(self, item: str) -> bool:
         """
         Check if the :class:`DataFrame` contains a column by the name of `item`
         """