fix: small fixes (#30)

Fixes: - upgrade `airflow-mcd` to 0.3.3 version to fix bug with DAG callbacks - fix correct usage of the default retry policy if none has been passed.
Toloka · Sep 24, 2024 · 7512685 · 7512685
1 parent 262e02a
commit 7512685
Show file tree

Hide file tree

Showing 3 changed files with 250 additions and 237 deletions.
diff --git a/dbt_af/operators/base.py b/dbt_af/operators/base.py
@@ -65,7 +65,9 @@ def __init__(
  af_pool = pool or f'dbt_{self.target_environment}' if dbt_af_config.use_dbt_target_specific_pools else None
 
  retry_policy = (
- retry_policy.as_dict() if retry_policy is not None else dbt_af_config.retries_config.default_retry_policy
+ retry_policy.as_dict()
+ if retry_policy is not None
+ else dbt_af_config.retries_config.default_retry_policy.as_dict()
  )
  super().__init__(
  max_active_tis_per_dag=max_active_tis_per_dag,