Adding only valid DAG state for TriggerDagRunOperator (#1245)

Recently, we have noticed that our Master DAG and example_external_task DAG are throwing parsing errors due to a new change. We can only pass valid DAG states in TriggerDagRunOperator. This PR will address this issue
astronomer · Jul 13, 2023 · fedc81f · fedc81f
1 parent 3c032c9
commit fedc81f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.circleci/integration-tests/master_dag.py b/.circleci/integration-tests/master_dag.py
@@ -136,7 +136,7 @@ def prepare_dag_dependency(task_info, execution_time):
                 wait_for_completion=True,
                 reset_dag_run=True,
                 execution_date=execution_time,
-                allowed_states=["success", "failed", "skipped"],
+                allowed_states=["success", "failed"],
                 trigger_rule="all_done",
             )
         )

diff --git a/astronomer/providers/core/example_dags/example_external_task.py b/astronomer/providers/core/example_dags/example_external_task.py
@@ -71,7 +71,7 @@ def check_dag_status(**kwargs: Any) -> None:
         trigger_dag_id="example_external_task_async_waits_for_me",
         wait_for_completion=True,
         reset_dag_run=True,
-        allowed_states=["success", "failed", "skipped"],
+        allowed_states=["success", "failed"],
         execution_date="{{execution_date}}",
         poke_interval=1,
     )