Commit bfaf24cd authored by Baber's avatar Baber
Browse files

refactor: update type hints for 3.9

parent 45c11c31
...@@ -196,11 +196,11 @@ def _get_cached_config(yaml_path: Path, mode: str) -> dict: ...@@ -196,11 +196,11 @@ def _get_cached_config(yaml_path: Path, mode: str) -> dict:
def load_yaml_config( def load_yaml_config(
yaml_path: Union[Path, str, None] = None, yaml_path: Union[Path, str, None] = None,
yaml_config: dict | None = None, yaml_config: Optional[dict] = None,
yaml_dir: Path | None = None, yaml_dir: Optional[Path] = None,
mode: str = "full", mode: str = "full",
*, *,
_seen: set[tuple[Path, str]] | None = None, _seen: Optional[set[tuple[Path, str]]] = None,
resolve_includes: bool = True, resolve_includes: bool = True,
) -> dict: ) -> dict:
""" """
...@@ -719,7 +719,7 @@ class TaskManager: ...@@ -719,7 +719,7 @@ class TaskManager:
self, self,
cfg: dict, cfg: dict,
task_name: str, task_name: str,
yaml_path: str | None, yaml_path: Union[str, None],
) -> dict: ) -> dict:
""" """
Instantiate a single task (ConfigurableTask **or** python-task) from *cfg*. Instantiate a single task (ConfigurableTask **or** python-task) from *cfg*.
...@@ -766,7 +766,7 @@ class TaskManager: ...@@ -766,7 +766,7 @@ class TaskManager:
def _create_group_object( def _create_group_object(
self, self,
cfg: dict, cfg: dict,
parent_name: str | None = None, parent_name: Union[str, None] = None,
) -> tuple[GroupConfig, list[Union[str, dict]]]: ) -> tuple[GroupConfig, list[Union[str, dict]]]:
""" """
Build GroupConfig and return (group_obj, subtask_names). Build GroupConfig and return (group_obj, subtask_names).
...@@ -788,7 +788,7 @@ class TaskManager: ...@@ -788,7 +788,7 @@ class TaskManager:
self, self,
subtasks: list[Union[str, dict]], subtasks: list[Union[str, dict]],
parent_name: Union[str, GroupConfig, None], parent_name: Union[str, GroupConfig, None],
update_config: dict | None, update_config: Union[dict, None],
) -> Mapping: ) -> Mapping:
"""Return merged mapping of all subtasks, handling duplicates.""" """Return merged mapping of all subtasks, handling duplicates."""
fn = functools.partial( fn = functools.partial(
...@@ -800,10 +800,10 @@ class TaskManager: ...@@ -800,10 +800,10 @@ class TaskManager:
def _load_individual_task_or_group( def _load_individual_task_or_group(
self, self,
payload: str | dict, payload: Union[str, dict],
*, *,
parent_name: str | None = None, parent_name: Union[str, None] = None,
update_config: dict | None = None, update_config: Union[dict, None] = None,
) -> Mapping: ) -> Mapping:
""" """
Public helper that turns *payload* (str task/group/tag **or** dict config) Public helper that turns *payload* (str task/group/tag **or** dict config)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment