style: import typing

Lee-W · Lee-W · commit 70e47dbf66be · 2025-05-22T15:39:43.000+08:00
diff --git a/airflow-core/src/airflow/models/expandinput.py b/airflow-core/src/airflow/models/expandinput.py
@@ -133,7 +133,7 @@ def get_total_map_length(self, run_id: str, *, session: Session) -> int:
         return length
 
 
-_EXPAND_INPUT_TYPES: dict[str, type] = {
+_EXPAND_INPUT_TYPES: dict[str, type[SchedulerExpandInput]] = {
     "dict-of-lists": SchedulerDictOfListsExpandInput,
     "list-of-dicts": SchedulerListOfDictsExpandInput,
 }

Original file line number	Diff line number	Diff line change
`@@ -133,7 +133,7 @@ def get_total_map_length(self, run_id: str, *, session: Session) -> int:`
`133`	`133`	`return length`
`134`	`134`
`135`	`135`
`136`		`-_EXPAND_INPUT_TYPES: dict[str, type] = {`
	`136`	`+_EXPAND_INPUT_TYPES: dict[str, type[SchedulerExpandInput]] = {`
`137`	`137`	`"dict-of-lists": SchedulerDictOfListsExpandInput,`
`138`	`138`	`"list-of-dicts": SchedulerListOfDictsExpandInput,`
`139`	`139`	`}`