🎨 [pre-commit.ci] Auto format from pre-commit.com hooks

cosmos/profiles/__init__.py

@@ -2,7 +2,6 @@
 
 from __future__ import annotations
 
-from .base import BaseProfileMapping
 from .bigquery.service_account_file import GoogleCloudServiceAccountFileProfileMapping
 from .databricks.token import DatabricksTokenProfileMapping
 from .exasol.user_pass import ExasolUserPasswordProfileMapping

tests/dbt/test_graph.py

@@ -5,7 +5,6 @@
 
 from cosmos.constants import ExecutionMode, DbtResourceType
 from cosmos.dbt.graph import DbtGraph, LoadMode, CosmosLoadDbtException
-from cosmos.dbt.project import DbtProject
 from cosmos.config import CosmosConfig, ProjectConfig, ProfileConfig, RenderConfig, ExecutionConfig
 
 DBT_PROJECTS_ROOT_DIR = Path(__file__).parent.parent.parent / "dev/dags/dbt"
@@ -294,7 +293,7 @@ def test_load_via_dbt_ls_with_invalid_dbt_path():
             ),
         ),
     )
-    with pytest.raises(FileNotFoundError) as err_info:
+    with pytest.raises(FileNotFoundError):
         dbt_graph.load_via_dbt_ls()
 
 

tests/operators/test_base.py

@@ -1,8 +1,6 @@
 from unittest.mock import patch, MagicMock
 
 import pytest
-from airflow.exceptions import AirflowException, AirflowSkipException
-from airflow.hooks.subprocess import SubprocessResult
 from airflow.utils.context import Context
 from pendulum import datetime
 

tests/operators/test_kubernetes.py

@@ -1,16 +1,11 @@
-from pathlib import Path
-from unittest.mock import MagicMock, patch
+from unittest.mock import patch
 
 from airflow.utils.context import Context
 from pendulum import datetime
 
 from cosmos.config import CosmosConfig, ProjectConfig, ProfileConfig, ExecutionConfig
 from cosmos.operators.kubernetes import (
-    DbtKubernetesBaseOperator,
     DbtLSKubernetesOperator,
-    DbtRunKubernetesOperator,
-    DbtSeedKubernetesOperator,
-    DbtTestKubernetesOperator,
 )