Ruff linting & format

Author: evertlammerts

duckdb/experimental/spark/sql/session.py

@@ -31,7 +31,7 @@
 
 # data is a List of rows
 # every value in each row needs to be turned into a Value
-def _combine_data_and_schema(data: Iterable[Any], schema: StructType) -> list['duckdb.Value']:
+def _combine_data_and_schema(data: Iterable[Any], schema: StructType) -> list["duckdb.Value"]:
     from duckdb import Value
 
     new_data = []

scripts/generate_connection_methods.py

@@ -5,7 +5,7 @@
 os.chdir(os.path.dirname(__file__))
 
 JSON_PATH = "connection_methods.json"
-PYCONNECTION_SOURCE = Path("..")  / "src" / "duckdb_py" / "pyconnection.cpp"
+PYCONNECTION_SOURCE = Path("..") / "src" / "duckdb_py" / "pyconnection.cpp"
 
 INITIALIZE_METHOD = (
     "static void InitializeConnectionMethods(py::class_<DuckDBPyConnection, shared_ptr<DuckDBPyConnection>> &m) {"

scripts/generate_connection_wrapper_methods.py

@@ -8,7 +8,7 @@
 os.chdir(os.path.dirname(__file__))
 
 JSON_PATH = "connection_methods.json"
-DUCKDB_PYTHON_SOURCE = Path("..")  / "src" / "duckdb_py" / "duckdb_python.cpp"
+DUCKDB_PYTHON_SOURCE = Path("..") / "src" / "duckdb_py" / "duckdb_python.cpp"
 
 START_MARKER = "\t// START_OF_CONNECTION_METHODS"
 END_MARKER = "\t// END_OF_CONNECTION_METHODS"

scripts/generate_import_cache_cpp.py

@@ -251,8 +251,9 @@ def get_module_file_path_includes(files: list[ModuleFile]):
 
 module_includes = get_module_file_path_includes(files)
 
-modules_header = Path(script_dir) / ".." / ("src/duckdb_py/include/duckdb_python/"
-                                            "import_cache/python_import_cache_modules.hpp")
+modules_header = (
+    Path(script_dir) / ".." / ("src/duckdb_py/include/duckdb_python/import_cache/python_import_cache_modules.hpp")
+)
 with open(modules_header, "w") as f:
     f.write(module_includes)
 

scripts/get_cpp_methods.py

@@ -52,8 +52,21 @@ def on_class_method(self, state, node):
 
 def get_methods(class_name: str) -> dict[str, ConnectionMethod]:
     CLASSES = {
-        "DuckDBPyConnection": Path(scripts_folder) / ".." / "src" / "duckdb_py" / "include" / "duckdb_python" / "pyconnection" / "pyconnection.hpp",  # noqa: E501
-        "DuckDBPyRelation": Path(scripts_folder) / ".." / "src" / "duckdb_py" / "include" / "duckdb_python" / "pyrelation.hpp",  # noqa: E501
+        "DuckDBPyConnection": Path(scripts_folder)
+        / ".."
+        / "src"
+        / "duckdb_py"
+        / "include"
+        / "duckdb_python"
+        / "pyconnection"
+        / "pyconnection.hpp",
+        "DuckDBPyRelation": Path(scripts_folder)
+        / ".."
+        / "src"
+        / "duckdb_py"
+        / "include"
+        / "duckdb_python"
+        / "pyrelation.hpp",
     }
 
     path = CLASSES[class_name]

tests/conftest.py

@@ -3,7 +3,7 @@
 import shutil
 import warnings
 from importlib import import_module
-from os.path import abspath, dirname, join, normpath
+from os.path import abspath
 from pathlib import Path
 from typing import Any