chore: apply ruff format

src/dagcellent/__init__.py

@@ -1,4 +1,5 @@
 """Common Airflow dags and plugins for DFDS."""
+
 # SPDX-FileCopyrightText: 2024-present Timon Viola <viotimo@dfds.com>
 #
 # SPDX-License-Identifier: MIT

src/dagcellent/_connection.py

@@ -1,4 +1,5 @@
 """Airflow connection utilities."""
+
 from __future__ import annotations
 
 from enum import Enum

src/dagcellent/dag/_meta.py

@@ -1,4 +1,5 @@
 """DAG metadata related utilities."""
+
 from __future__ import annotations
 
 import os

src/dagcellent/data_utils/__init__.py

@@ -1,4 +1,5 @@
 """Utilities related to handling various data sources e.g.: databases, cloud blob storage."""
+
 from __future__ import annotations
 
 # pyright: reportUnknownVariableType=false

src/dagcellent/data_utils/sql_reflection.py

@@ -1,4 +1,5 @@
 """SqlAlchemy based utility functions to interact with databases."""
+
 from __future__ import annotations
 
 import logging

src/dagcellent/operators/__init__.py

@@ -1,4 +1,5 @@
 """Reusable operators to build Airflow DAGs."""
+
 from __future__ import annotations
 
 from dagcellent.operators.external_table_arrow import CreateExternalTableArrow

src/dagcellent/operators/mlflow/_utils.py

@@ -1,4 +1,5 @@
 """Mlflow helpers."""
+
 # ruff: noqa: G004
 # typing: ignore
 from __future__ import annotations
@@ -7,7 +8,6 @@
 from typing import TYPE_CHECKING, TypedDict
 
 if TYPE_CHECKING:
-
     import mlflow.entities.model_registry
 
 

src/dagcellent/operators/mlflow/operators.py

@@ -1,4 +1,3 @@
-
 from __future__ import annotations
 
 import logging
@@ -52,7 +51,7 @@ def __init__(
         self.version = version
         self.tag = tag
 
-        super().__init__(**kwargs) # type: ignore [reportUnknownMemberType]
+        super().__init__(**kwargs)  # type: ignore [reportUnknownMemberType]
 
     def execute(self, context: Any) -> None:
         """Operator execute method.
@@ -95,8 +94,7 @@ def __init__(
         self.stage = stage
         self.model_name = model_name
 
-        super().__init__(**kwargs) # type: ignore [reportUnknownMemberType]
-
+        super().__init__(**kwargs)  # type: ignore [reportUnknownMemberType]
 
     def execute(self, context: Any) -> list[SlimModelVersion]:
         """Connect to Mlflow hook and dump messages to tmp file."""
@@ -131,7 +129,7 @@ def __init__(
         self.tracking_uri = tracking_uri
         self.upstream_task_id = upstream_task_id
 
-        super().__init__(**kwargs)# type: ignore [reportUnknownMemberType]
+        super().__init__(**kwargs)  # type: ignore [reportUnknownMemberType]
 
     def execute(self, context: Any) -> Any:
         """Operator execute method.
@@ -147,13 +145,13 @@ def execute(self, context: Any) -> Any:
             key="return_value", task_ids=self.upstream_task_id
         )
         if isinstance(xcom_val, list):
-            _first_item = xcom_val.pop() # type: ignore [unknownTypeIssue]
-            run_id = _first_item["run_id"]# type: ignore [unknownTypeIssue]
+            _first_item = xcom_val.pop()  # type: ignore [unknownTypeIssue]
+            run_id = _first_item["run_id"]  # type: ignore [unknownTypeIssue]
             logging.warning(
                 "Multiple items returned from upstream task. Defaulting to first item"
             )
         elif isinstance(xcom_val, dict):
-            run_id = xcom_val["run_id"]# type: ignore [unknownTypeIssue]
+            run_id = xcom_val["run_id"]  # type: ignore [unknownTypeIssue]
         else:
             logging.error(
                 "Operator has no implementation for this type: %s", f"{type(xcom_val)}."
@@ -164,8 +162,7 @@ def execute(self, context: Any) -> Any:
             raise ValueError(f"{run_id} must be an instance of 'str'")
         data = client_hook.get_run(run_id)
         logging.info("%s", f"{data=}")
-        return data.data.metrics # type: ignore [unknownTypeIssue]
-
+        return data.data.metrics  # type: ignore [unknownTypeIssue]
 
 
 class GetLatestModelVersion(BaseOperator):
@@ -192,7 +189,7 @@ def __init__(
         self.tracking_uri = tracking_uri
         self.model_name = model_name
 
-        super().__init__(**kwargs)# type: ignore [reportUnknownMemberType]
+        super().__init__(**kwargs)  # type: ignore [reportUnknownMemberType]
 
     def execute(self, context: Any) -> SlimModelVersion:
         """Operator execute method.

tests/dags/test_msql_reflect_otherdb.py

@@ -16,7 +16,6 @@
     schedule="@once",
     catchup=False,
 ) as dag:
-
     reflect_table = SQLReflectOperator(
         task_id="reflect_database", conn_id=CONN_ID, table_name="kaka", database="model"
     )

tests/dags/test_mssql_reflect.py

@@ -1,6 +1,7 @@
 """Test DAG to show the usage of SQL reflection
 and executing the returned query.
 """
+
 from __future__ import annotations
 
 import datetime

tests/dags/test_psql_reflect.py

@@ -1,4 +1,5 @@
 """Test SQL reflection on PostrgeSQL."""
+
 from __future__ import annotations
 
 import datetime

tests/dags/test_psql_s3.py

@@ -1,4 +1,5 @@
 """Test SQL reflection on PostrgeSQL."""
+
 from __future__ import annotations
 
 import datetime

tests/test_mlflow.py

@@ -1,4 +1,5 @@
 """Unit tests for the operators.mlflow module."""
+
 from __future__ import annotations