sunank200 commented on code in PR #55292:
URL: https://github.com/apache/airflow/pull/55292#discussion_r2450521201
##########
task-sdk/src/airflow/sdk/definitions/dag.py:
##########
@@ -1135,9 +1135,18 @@ def test(
from airflow.sdk import DagRunState, TaskInstanceState, timezone
from airflow.secrets.local_filesystem import LocalFilesystemBackend
from airflow.serialization.serialized_objects import SerializedDAG
- from airflow.utils.state import State
from airflow.utils.types import DagRunTriggeredByType, DagRunType
+ FINISHED_STATES = frozenset(
+ [
+ TaskInstanceState.SUCCESS,
+ TaskInstanceState.FAILED,
+ TaskInstanceState.SKIPPED,
+ TaskInstanceState.UPSTREAM_FAILED,
+ TaskInstanceState.REMOVED,
+ ]
+ )
Review Comment:
I can change this in follow up PR. But for now this place felt the right
place.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
