Fix import

astronomer · Jan 30, 2025 · 9fc5112 · 9fc5112
1 parent 93e7a8c
commit 9fc5112
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/operators/test_local.py b/tests/operators/test_local.py
@@ -34,7 +34,7 @@
     DbtDocsGCSLocalOperator,
     DbtDocsLocalOperator,
     DbtDocsS3LocalOperator,
-    DbtLocalBase,
+    DbtLocalBaseOperator,
     DbtLSLocalOperator,
     DbtRunLocalOperator,
     DbtRunOperationLocalOperator,
@@ -82,7 +82,7 @@ def failing_test_dbt_project(tmp_path):
     tmp_dir.cleanup()
 
 
-class ConcreteDbtLocalBaseOperator(DbtLocalBase):
+class ConcreteDbtLocalBaseOperator(DbtLocalBaseOperator):
     base_cmd = ["cmd"]
 
 
@@ -1293,7 +1293,7 @@ def test_configure_remote_target_path(mock_object_storage_path):
     mock_object_storage_path.return_value.mkdir.assert_called_with(parents=True, exist_ok=True)
 
 
-@patch.object(DbtLocalBase, "_configure_remote_target_path")
+@patch.object(DbtLocalBaseOperator, "_configure_remote_target_path")
 def test_no_compiled_sql_upload_for_other_operators(mock_configure_remote_target_path):
     operator = DbtSeedLocalOperator(
         task_id="fake-task",