Wraps run_dbt_ function in sync_compatible (#15740)

PrefectHQ · Oct 21, 2024 · 4583539 · 4583539
1 parent c78feea
commit 4583539
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/integrations/prefect-dbt/prefect_dbt/cli/commands.py b/src/integrations/prefect-dbt/prefect_dbt/cli/commands.py
@@ -397,6 +397,7 @@ def _compile_kwargs(self, **open_kwargs: Dict[str, Any]) -> Dict[str, Any]:
         return super(type(self), modified_self)._compile_kwargs(**open_kwargs)
 
 
+@sync_compatible
 @task
 async def run_dbt_build(
     profiles_dir: Optional[Union[Path, str]] = None,
@@ -470,6 +471,7 @@ def dbt_test_flow():
     return results
 
 
+@sync_compatible
 @task
 async def run_dbt_model(
     profiles_dir: Optional[Union[Path, str]] = None,
@@ -544,6 +546,7 @@ def dbt_test_flow():
     return results
 
 
+@sync_compatible
 @task
 async def run_dbt_test(
     profiles_dir: Optional[Union[Path, str]] = None,
@@ -618,6 +621,7 @@ def dbt_test_flow():
     return results
 
 
+@sync_compatible
 @task
 async def run_dbt_snapshot(
     profiles_dir: Optional[Union[Path, str]] = None,
@@ -692,6 +696,7 @@ def dbt_test_flow():
     return results
 
 
+@sync_compatible
 @task
 async def run_dbt_seed(
     profiles_dir: Optional[Union[Path, str]] = None,
@@ -766,6 +771,7 @@ def dbt_test_flow():
     return results
 
 
+@sync_compatible
 @task
 async def run_dbt_source_freshness(
     profiles_dir: Optional[Union[Path, str]] = None,