Skip to content

Commit 7f284d3

Browse files
committed
fix: simplify import to match codebase pattern
- Import NOTSET and ArgNotSet directly from airflow.utils.types - Matches pattern used in other provider files
1 parent 2884e31 commit 7f284d3

1 file changed

Lines changed: 1 addition & 6 deletions

File tree

providers/standard/src/airflow/providers/standard/operators/trigger_dagrun.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -44,12 +44,7 @@
4444
from airflow.providers.standard.utils.openlineage import safe_inject_openlineage_properties_into_dagrun_conf
4545
from airflow.providers.standard.version_compat import AIRFLOW_V_3_0_PLUS, BaseOperator
4646
from airflow.utils.state import DagRunState
47-
from airflow.utils.types import DagRunType
48-
49-
try:
50-
from airflow.sdk.definitions._internal.types import NOTSET, ArgNotSet
51-
except ImportError:
52-
from airflow.utils.types import NOTSET, ArgNotSet # type: ignore[attr-defined,no-redef]
47+
from airflow.utils.types import ArgNotSet, DagRunType, NOTSET
5348

5449
XCOM_LOGICAL_DATE_ISO = "trigger_logical_date_iso"
5550
XCOM_RUN_ID = "trigger_run_id"

0 commit comments

Comments
 (0)