Ruff format fixes

Author: evertlammerts

adbc_driver_duckdb/dbapi.py

@@ -15,8 +15,7 @@
 # specific language governing permissions and limitations
 # under the License.
 
-"""DBAPI 2.0-compatible facade for the ADBC DuckDB driver.
-"""
+"""DBAPI 2.0-compatible facade for the ADBC DuckDB driver."""
 
 import typing
 

duckdb/experimental/spark/errors/__init__.py

@@ -15,8 +15,7 @@
 # limitations under the License.
 #
 
-"""PySpark exceptions.
-"""
+"""PySpark exceptions."""
 
 from .exceptions.base import (
     AnalysisException,

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

@@ -4,8 +4,7 @@
 
 
 class PySparkException(Exception):
-    """Base Exception for handling errors generated from PySpark.
-    """
+    """Base Exception for handling errors generated from PySpark."""
 
     def __init__(
         self,
@@ -78,115 +77,92 @@ def __str__(self) -> str:
 
 
 class AnalysisException(PySparkException):
-    """Failed to analyze a SQL query plan.
-    """
+    """Failed to analyze a SQL query plan."""
 
 
 class SessionNotSameException(PySparkException):
-    """Performed the same operation on different SparkSession.
-    """
+    """Performed the same operation on different SparkSession."""
 
 
 class TempTableAlreadyExistsException(AnalysisException):
-    """Failed to create temp view since it is already exists.
-    """
+    """Failed to create temp view since it is already exists."""
 
 
 class ParseException(AnalysisException):
-    """Failed to parse a SQL command.
-    """
+    """Failed to parse a SQL command."""
 
 
 class IllegalArgumentException(PySparkException):
-    """Passed an illegal or inappropriate argument.
-    """
+    """Passed an illegal or inappropriate argument."""
 
 
 class ArithmeticException(PySparkException):
-    """Arithmetic exception thrown from Spark with an error class.
-    """
+    """Arithmetic exception thrown from Spark with an error class."""
 
 
 class UnsupportedOperationException(PySparkException):
-    """Unsupported operation exception thrown from Spark with an error class.
-    """
+    """Unsupported operation exception thrown from Spark with an error class."""
 
 
 class ArrayIndexOutOfBoundsException(PySparkException):
-    """Array index out of bounds exception thrown from Spark with an error class.
-    """
+    """Array index out of bounds exception thrown from Spark with an error class."""
 
 
 class DateTimeException(PySparkException):
-    """Datetime exception thrown from Spark with an error class.
-    """
+    """Datetime exception thrown from Spark with an error class."""
 
 
 class NumberFormatException(IllegalArgumentException):
-    """Number format exception thrown from Spark with an error class.
-    """
+    """Number format exception thrown from Spark with an error class."""
 
 
 class StreamingQueryException(PySparkException):
-    """Exception that stopped a :class:`StreamingQuery`.
-    """
+    """Exception that stopped a :class:`StreamingQuery`."""
 
 
 class QueryExecutionException(PySparkException):
-    """Failed to execute a query.
-    """
+    """Failed to execute a query."""
 
 
 class PythonException(PySparkException):
-    """Exceptions thrown from Python workers.
-    """
+    """Exceptions thrown from Python workers."""
 
 
 class SparkRuntimeException(PySparkException):
-    """Runtime exception thrown from Spark with an error class.
-    """
+    """Runtime exception thrown from Spark with an error class."""
 
 
 class SparkUpgradeException(PySparkException):
-    """Exception thrown because of Spark upgrade.
-    """
+    """Exception thrown because of Spark upgrade."""
 
 
 class UnknownException(PySparkException):
-    """None of the above exceptions.
-    """
+    """None of the above exceptions."""
 
 
 class PySparkValueError(PySparkException, ValueError):
-    """Wrapper class for ValueError to support error classes.
-    """
+    """Wrapper class for ValueError to support error classes."""
 
 
 class PySparkIndexError(PySparkException, IndexError):
-    """Wrapper class for IndexError to support error classes.
-    """
+    """Wrapper class for IndexError to support error classes."""
 
 
 class PySparkTypeError(PySparkException, TypeError):
-    """Wrapper class for TypeError to support error classes.
-    """
+    """Wrapper class for TypeError to support error classes."""
 
 
 class PySparkAttributeError(PySparkException, AttributeError):
-    """Wrapper class for AttributeError to support error classes.
-    """
+    """Wrapper class for AttributeError to support error classes."""
 
 
 class PySparkRuntimeError(PySparkException, RuntimeError):
-    """Wrapper class for RuntimeError to support error classes.
-    """
+    """Wrapper class for RuntimeError to support error classes."""
 
 
 class PySparkAssertionError(PySparkException, AssertionError):
-    """Wrapper class for AssertionError to support error classes.
-    """
+    """Wrapper class for AssertionError to support error classes."""
 
 
 class PySparkNotImplementedError(PySparkException, NotImplementedError):
-    """Wrapper class for NotImplementedError to support error classes.
-    """
+    """Wrapper class for NotImplementedError to support error classes."""

duckdb/experimental/spark/errors/utils.py

@@ -21,15 +21,13 @@
 
 
 class ErrorClassesReader:
-    """A reader to load error information from error_classes.py.
-    """
+    """A reader to load error information from error_classes.py."""
 
     def __init__(self) -> None:
         self.error_info_map = ERROR_CLASSES_MAP
 
     def get_error_message(self, error_class: str, message_parameters: dict[str, str]) -> str:
-        """Returns the completed error message by applying message parameters to the message template.
-        """
+        """Returns the completed error message by applying message parameters to the message template."""
         message_template = self.get_message_template(error_class)
         # Verify message parameters.
         message_parameters_from_template = re.findall("<([a-zA-Z0-9_-]+)>", message_template)

duckdb/experimental/spark/sql/dataframe.py

@@ -843,8 +843,7 @@ def limit(self, num: int) -> "DataFrame":
         return DataFrame(rel, self.session)
 
     def __contains__(self, item: str) -> bool:
-        """Check if the :class:`DataFrame` contains a column by the name of `item`
-        """
+        """Check if the :class:`DataFrame` contains a column by the name of `item`"""
         return item in self.relation
 
     @property

duckdb/experimental/spark/sql/streaming.py

@@ -30,7 +30,6 @@ def load(
         schema: Union[StructType, str, None] = None,
         **options: OptionalPrimitiveType,
     ) -> "DataFrame":
-
         raise NotImplementedError
 
 

duckdb/experimental/spark/sql/types.py

@@ -102,13 +102,11 @@ def needConversion(self) -> bool:
         return False
 
     def toInternal(self, obj: Any) -> Any:
-        """Converts a Python object into an internal SQL object.
-        """
+        """Converts a Python object into an internal SQL object."""
         return obj
 
     def fromInternal(self, obj: Any) -> Any:
-        """Converts an internal SQL object into a native Python object.
-        """
+        """Converts an internal SQL object into a native Python object."""
         return obj
 
 
@@ -979,14 +977,12 @@ def typeName(cls) -> str:
 
     @classmethod
     def sqlType(cls) -> DataType:
-        """Underlying SQL storage type for this UDT.
-        """
+        """Underlying SQL storage type for this UDT."""
         raise NotImplementedError("UDT must implement sqlType().")
 
     @classmethod
     def module(cls) -> str:
-        """The Python module of the UDT.
-        """
+        """The Python module of the UDT."""
         raise NotImplementedError("UDT must implement module().")
 
     @classmethod
@@ -1001,8 +997,7 @@ def needConversion(self) -> bool:
 
     @classmethod
     def _cachedSqlType(cls) -> DataType:
-        """Cache the sqlType() into class, because it's heavily used in `toInternal`.
-        """
+        """Cache the sqlType() into class, because it's heavily used in `toInternal`."""
         if not hasattr(cls, "_cached_sql_type"):
             cls._cached_sql_type = cls.sqlType()  # type: ignore[attr-defined]
         return cls._cached_sql_type  # type: ignore[attr-defined]
@@ -1017,13 +1012,11 @@ def fromInternal(self, obj: Any) -> Any:
             return self.deserialize(v)
 
     def serialize(self, obj: Any) -> Any:
-        """Converts a user-type object into a SQL datum.
-        """
+        """Converts a user-type object into a SQL datum."""
         raise NotImplementedError("UDT must implement toInternal().")
 
     def deserialize(self, datum: Any) -> Any:
-        """Converts a SQL datum into a user-type object.
-        """
+        """Converts a SQL datum into a user-type object."""
         raise NotImplementedError("UDT must implement fromInternal().")
 
     def simpleString(self) -> str:

duckdb/polars_io.py

@@ -206,8 +206,7 @@ def _pl_tree_to_sql(tree: dict) -> str:
 
 
 def duckdb_source(relation: duckdb.DuckDBPyRelation, schema: pl.schema.Schema) -> pl.LazyFrame:
-    """A polars IO plugin for DuckDB.
-    """
+    """A polars IO plugin for DuckDB."""
 
     def source_generator(
         with_columns: Optional[list[str]],

duckdb/udf.py

@@ -1,6 +1,5 @@
 def vectorized(func):
-    """Decorate a function with annotated function parameters, so DuckDB can infer that the function should be provided with pyarrow arrays and should expect pyarrow array(s) as output
-    """
+    """Decorate a function with annotated function parameters, so DuckDB can infer that the function should be provided with pyarrow arrays and should expect pyarrow array(s) as output"""
     import types
     from inspect import signature
 

duckdb_packaging/pypi_cleanup.py

@@ -78,17 +78,14 @@ class PyPICleanupError(Exception):
     """Base exception for PyPI cleanup operations."""
 
 
-
 class AuthenticationError(PyPICleanupError):
     """Raised when authentication fails."""
 
 
-
 class ValidationError(PyPICleanupError):
     """Raised when input validation fails."""
 
 
-
 def setup_logging(verbose: bool = False) -> None:
     """Configure logging with appropriate level and format."""
     level = logging.DEBUG if verbose else logging.INFO