From e2ff5c61e01560c6a9ea90c304504e66f9707747 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Mon, 24 Jul 2023 03:20:38 +0000
Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=A8=20[pre-commit.ci]=20Auto=20format?=
 =?UTF-8?q?=20from=20pre-commit.com=20hooks?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cosmos/profiles/__init__.py        | 1 -
 tests/dbt/test_graph.py            | 3 +--
 tests/operators/test_base.py       | 2 --
 tests/operators/test_kubernetes.py | 7 +------
 4 files changed, 2 insertions(+), 11 deletions(-)

diff --git a/cosmos/profiles/__init__.py b/cosmos/profiles/__init__.py
index 7af554308..784a2166d 100644
--- a/cosmos/profiles/__init__.py
+++ b/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
diff --git a/tests/dbt/test_graph.py b/tests/dbt/test_graph.py
index c51b7985a..66fc4a749 100644
--- a/tests/dbt/test_graph.py
+++ b/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()
 
 
diff --git a/tests/operators/test_base.py b/tests/operators/test_base.py
index a6c80eace..6dd580a1c 100644
--- a/tests/operators/test_base.py
+++ b/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
 
diff --git a/tests/operators/test_kubernetes.py b/tests/operators/test_kubernetes.py
index b63418d43..c336fbfe6 100644
--- a/tests/operators/test_kubernetes.py
+++ b/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,
 )