From 7916857f331c40940549ea2916d20c21a2e6268d Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 30 Sep 2024 09:42:49 +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/operators/airflow_async.py | 11 +++-------- cosmos/operators/local.py | 1 - 2 files changed, 3 insertions(+), 9 deletions(-) diff --git a/cosmos/operators/airflow_async.py b/cosmos/operators/airflow_async.py index f278dcba8..9a6c3e862 100644 --- a/cosmos/operators/airflow_async.py +++ b/cosmos/operators/airflow_async.py @@ -6,25 +6,22 @@ from airflow.providers.google.cloud.operators.bigquery import BigQueryInsertJobOperator from airflow.utils.context import Context -from cosmos.operators.base import DbtCompileMixin from cosmos.operators.local import ( DbtBuildLocalOperator, + DbtCompileLocalOperator, DbtDepsLocalOperator, DbtDocsAzureStorageLocalOperator, DbtDocsCloudLocalOperator, DbtDocsGCSLocalOperator, DbtDocsLocalOperator, DbtDocsS3LocalOperator, - DbtLocalBaseOperator, DbtLSLocalOperator, + DbtRunOperationLocalOperator, DbtSeedLocalOperator, DbtSnapshotLocalOperator, DbtSourceLocalOperator, DbtTestLocalOperator, - DbtRunOperationLocalOperator, - DbtCompileLocalOperator, ) - from cosmos.settings import remote_target_path, remote_target_path_conn_id _SUPPORTED_DATABASES = ["bigquery"] @@ -95,7 +92,6 @@ def execute(self, context: Context) -> Any | None: super().execute(context) - class DbtTestAirflowAsyncOperator(DbtTestLocalOperator): pass @@ -127,7 +123,6 @@ class DbtDocsGCSAirflowAsyncOperator(DbtDocsGCSLocalOperator): class DbtCompileAirflowAsyncOperator(DbtCompileLocalOperator): pass + class DbtDepsAirflowAsyncOperator(DbtDepsLocalOperator): pass - - diff --git a/cosmos/operators/local.py b/cosmos/operators/local.py index ad320d5fc..ee0754f75 100644 --- a/cosmos/operators/local.py +++ b/cosmos/operators/local.py @@ -291,7 +291,6 @@ def _configure_remote_target_path() -> tuple[Path, str] | tuple[None, None]: return _configured_target_path, remote_conn_id - def upload_compiled_sql(self, tmp_project_dir: str, context: Context) -> None: """ Uploads the compiled SQL files from the dbt compile output to the remote store.