feat: add skipped_states to DbtExternalSensor (#22)

add skipped_states to DbtExternalSensor
Toloka · Sep 9, 2024 · 375669c · 375669c
1 parent d6f294d
commit 375669c
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/dbt_af/__init__.py b/dbt_af/__init__.py
@@ -3,6 +3,6 @@
  'conf',
 ]
 
-__version__ = '0.7.1'
+__version__ = '0.7.2'
 
 from . import conf, dags # noqa
diff --git a/dbt_af/operators/sensors.py b/dbt_af/operators/sensors.py
@@ -367,7 +367,8 @@ def __init__(
  pool=DBT_SENSOR_POOL if dbt_af_config.use_dbt_target_specific_pools else None,
  mode='reschedule',
  retries=max(_RETRIES_COUNT, kwargs.get('retries', 0)),
- failed_states=[State.NONE, State.SKIPPED, State.FAILED, State.UPSTREAM_FAILED],
+ skipped_states=[State.NONE, State.SKIPPED],
+ failed_states=[State.FAILED, State.UPSTREAM_FAILED],
  timeout=6 * 60 * 60,
  poke_interval=_POKE_INTERVALS_SECONDS.get(dep_schedule.name, _DEFAULT_POKE_INTERVAL_SECONDS),
  retry_delay=datetime.timedelta(seconds=60 * 30),

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "dbt-af"
-version = "0.7.1"
+version = "0.7.2"
 description = "Distibuted dbt runs on Apache Airflow"
 authors = [
  "Nikita Yurasov <[email protected]>",