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

astronomer · Sep 26, 2023 · 8e157a5 · 8e157a5
1 parent 95a791c
commit 8e157a5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/cosmos/operators/local.py b/cosmos/operators/local.py
@@ -40,10 +40,7 @@
     FullOutputSubprocessHook,
     FullOutputSubprocessResult,
 )
-from cosmos.dbt.parser.output import (
-    extract_log_issues,
-    parse_output
-)
+from cosmos.dbt.parser.output import extract_log_issues
 
 logger = get_logger(__name__)
 

diff --git a/tests/operators/test_local.py b/tests/operators/test_local.py
@@ -303,6 +303,7 @@ def test_operator_execute_without_flags(mock_build_and_run_cmd, operator_class):
     task.execute(context={})
     mock_build_and_run_cmd.assert_called_once_with(context={})
 
+
 @pytest.mark.integration
 def test_dbt_test_local_operator_on_warning_callback():
     on_warning_callback_mock = MagicMock()
@@ -314,13 +315,14 @@ def test_dbt_test_local_operator_on_warning_callback():
             task_id="my-task",
             install_deps=True,
             dbt_cmd_flags=["--models", "stg_customers"],
-            on_warning_callback=on_warning_callback_mock
+            on_warning_callback=on_warning_callback_mock,
         )
         test_operator
 
     run_test_dag(dag)
     on_warning_callback_mock.assert_called_once()
 
+
 @pytest.mark.integration
 def test_on_warning_callback_not_triggered():
     on_warning_callback_mock = MagicMock()