zrb 0.24.0__py3-none-any.whl → 0.26.2__py3-none-any.whl

zrb/task/base_remote_cmd_task.py

@@ -1,19 +1,12 @@
 import os
 import pathlib
+from collections.abc import Callable, Iterable, Mapping
+from typing import Any, Optional, TypeVar, Union
 
 from zrb.helper.accessories.color import colored
 from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
-from zrb.helper.typing import (
-    Any,
-    Callable,
-    Iterable,
-    JinjaTemplate,
-    Mapping,
-    Optional,
-    TypeVar,
-    Union,
-)
+from zrb.helper.typing import JinjaTemplate
 from zrb.helper.util import to_snake_case
 from zrb.task.any_task import AnyTask
 from zrb.task.any_task_event_handler import (

zrb/task/base_task/base_task.py

@@ -2,6 +2,8 @@ import asyncio
 import copy
 import os
 import shutil
+from collections.abc import Callable, Iterable, Mapping
+from typing import Any, Optional, Union
 
 from zrb.advertisement import advertisements
 from zrb.config.config import SHOW_ADVERTISEMENT, TMP_DIR
@@ -12,7 +14,6 @@ from zrb.helper.map.conversion import to_str as map_to_str
 from zrb.helper.string.conversion import to_variable_name
 from zrb.helper.string.modification import double_quote
 from zrb.helper.typecheck import typechecked
-from zrb.helper.typing import Any, Callable, Iterable, List, Mapping, Optional, Union
 from zrb.task.any_task import AnyTask
 from zrb.task.any_task_event_handler import (
     OnFailed,
@@ -41,14 +42,14 @@ class BaseTask(FinishTracker, AttemptTracker, Renderer, BaseTaskModel, AnyTask):
     Every Task definition should be extended from this class.
     """
 
-    __running_tasks: List[AnyTask] = []
+    __running_tasks: list[AnyTask] = []
 
     def __init__(
         self,
         name: str,
         group: Optional[Group] = None,
         description: str = "",
-        inputs: List[AnyInput] = [],
+        inputs: list[AnyInput] = [],
         envs: Iterable[Env] = [],
         env_files: Iterable[EnvFile] = [],
         icon: Optional[str] = None,
@@ -119,7 +120,7 @@ class BaseTask(FinishTracker, AttemptTracker, Renderer, BaseTaskModel, AnyTask):
             operand.add_upstream(self)
             return operand
         if isinstance(operand, AnyParallel):
-            other_tasks: List[AnyTask] = operand.get_tasks()
+            other_tasks: list[AnyTask] = operand.get_tasks()
             for other_task in other_tasks:
                 other_task.add_upstream(self)
             return operand
@@ -286,6 +287,9 @@ class BaseTask(FinishTracker, AttemptTracker, Renderer, BaseTaskModel, AnyTask):
             result = results[-1]
             self._print_result(result)
             return result
+        except RuntimeError as e:
+            if raise_error and ("event loop is closed" not in str(e).lower()):
+                raise e
         except Exception as e:
             self.log_error(f"{e}")
             if raise_error:
@@ -415,7 +419,7 @@ class BaseTask(FinishTracker, AttemptTracker, Renderer, BaseTaskModel, AnyTask):
         # wait all upstream checkers to complete
         await asyncio.gather(*coroutines)
 
-    async def __trigger_failure(self, tasks: List[AnyTask]):
+    async def __trigger_failure(self, tasks: list[AnyTask]):
         coroutines = [
             task.on_failed(is_last_attempt=True, exception=Exception("canceled"))
             for task in tasks
@@ -423,7 +427,7 @@ class BaseTask(FinishTracker, AttemptTracker, Renderer, BaseTaskModel, AnyTask):
         await asyncio.gather(*coroutines)
 
     async def __trigger_fallbacks(
-        self, tasks: List[AnyTask], kwargs: Mapping[str, Any]
+        self, tasks: list[AnyTask], kwargs: Mapping[str, Any]
     ):
         coroutines: Iterable[asyncio.Task] = []
         for fallback in self.__get_all_fallbacks(tasks):
@@ -431,8 +435,8 @@ class BaseTask(FinishTracker, AttemptTracker, Renderer, BaseTaskModel, AnyTask):
             coroutines.append(asyncio.create_task(fallback._run_all(**kwargs)))
         await asyncio.gather(*coroutines)
 
-    def __get_all_fallbacks(self, tasks: List[AnyTask]) -> List[AnyTask]:
-        all_fallbacks: List[AnyTask] = []
+    def __get_all_fallbacks(self, tasks: list[AnyTask]) -> list[AnyTask]:
+        all_fallbacks: list[AnyTask] = []
         for task in tasks:
             task._lock_fallbacks()
             for fallback in task._get_fallbacks():

zrb/task/base_task/component/base_task_model.py

@@ -2,7 +2,9 @@ import datetime
 import logging
 import os
 import sys
+from collections.abc import Callable, Iterable, Mapping
 from functools import lru_cache
+from typing import Any, Optional, Union
 
 from zrb.config.config import ENV_PREFIX, LOGGING_LEVEL, SHOW_TIME
 from zrb.helper.accessories.color import colored
@@ -10,16 +12,7 @@ from zrb.helper.log import logger
 from zrb.helper.string.conversion import to_variable_name
 from zrb.helper.string.modification import double_quote
 from zrb.helper.typecheck import typechecked
-from zrb.helper.typing import (
-    Any,
-    Callable,
-    Iterable,
-    JinjaTemplate,
-    List,
-    Mapping,
-    Optional,
-    Union,
-)
+from zrb.helper.typing import JinjaTemplate
 from zrb.task.any_task import AnyTask
 from zrb.task.any_task_event_handler import (
     OnFailed,
@@ -48,7 +41,7 @@ class BaseTaskModel(CommonTaskModel, PidModel, TimeTracker):
         name: str,
         group: Optional[Group] = None,
         description: str = "",
-        inputs: List[AnyInput] = [],
+        inputs: list[AnyInput] = [],
         envs: Iterable[Env] = [],
         env_files: Iterable[EnvFile] = [],
         icon: Optional[str] = None,
@@ -102,7 +95,7 @@ class BaseTaskModel(CommonTaskModel, PidModel, TimeTracker):
         )
         PidModel.__init__(self)
         TimeTracker.__init__(self)
-        self.__args: List[Any] = []
+        self.__args: list[Any] = []
         self.__kwargs: Mapping[str, Any] = {}
 
     def _set_args(self, args: Iterable[Any]):
@@ -201,7 +194,7 @@ class BaseTaskModel(CommonTaskModel, PidModel, TimeTracker):
     def _show_run_command(self):
         if not self.__has_cli_interface:
             return
-        params: List[str] = [double_quote(arg) for arg in self.__args]
+        params: list[str] = [double_quote(arg) for arg in self.__args]
         for task_input in self._get_combined_inputs():
             if task_input.is_hidden():
                 continue

zrb/task/base_task/component/common_task_model.py

@@ -1,19 +1,12 @@
 import os
+from collections.abc import Callable, Iterable, Mapping
+from typing import Any, Optional, Union
 
 from zrb.helper.accessories.color import get_random_color
 from zrb.helper.accessories.icon import get_random_icon
 from zrb.helper.string.conversion import to_cli_name
 from zrb.helper.typecheck import typechecked
-from zrb.helper.typing import (
-    Any,
-    Callable,
-    Iterable,
-    JinjaTemplate,
-    List,
-    Mapping,
-    Optional,
-    Union,
-)
+from zrb.helper.typing import JinjaTemplate
 from zrb.helper.util import coalesce_str
 from zrb.task.any_task import AnyTask
 from zrb.task.any_task_event_handler import (
@@ -39,7 +32,7 @@ class CommonTaskModel:
         name: str,
         group: Optional[Group] = None,
         description: str = "",
-        inputs: List[AnyInput] = [],
+        inputs: list[AnyInput] = [],
         envs: Iterable[Env] = [],
         env_files: Iterable[EnvFile] = [],
         icon: Optional[str] = None,
@@ -65,7 +58,7 @@ class CommonTaskModel:
         self._group = group
         if group is not None:
             group._add_task(self)
-        checkers_cp: List[AnyTask] = [checker.copy() for checker in checkers]
+        checkers_cp: list[AnyTask] = [checker.copy() for checker in checkers]
         for checker in checkers_cp:
             checker.add_env(*envs)
             checker.add_env_file(*env_files)
@@ -104,7 +97,7 @@ class CommonTaskModel:
         self.__has_already_inject_inputs: bool = False
         self.__has_already_inject_upstreams: bool = False
         self.__has_already_inject_fallbacks: bool = False
-        self.__all_inputs: Optional[List[AnyInput]] = None
+        self.__all_inputs: Optional[list[AnyInput]] = None
 
     def _lock_checkers(self):
         self.__allow_add_checkers = False
@@ -185,7 +178,7 @@ class CommonTaskModel:
     def inject_inputs(self):
         pass
 
-    def _get_inputs(self) -> List[AnyInput]:
+    def _get_inputs(self) -> list[AnyInput]:
         if not self.__has_already_inject_inputs:
             self.inject_inputs()
             self.__has_already_inject_inputs = True
@@ -197,7 +190,7 @@ class CommonTaskModel:
         """
         if self.__all_inputs is not None:
             return self.__all_inputs
-        self.__all_inputs: List[AnyInput] = []
+        self.__all_inputs: list[AnyInput] = []
         existing_input_names: Mapping[str, bool] = {}
         # Add task inputs
         inputs = self._get_inputs()
@@ -244,7 +237,7 @@ class CommonTaskModel:
     def inject_envs(self):
         pass
 
-    def _get_envs(self) -> List[Env]:
+    def _get_envs(self) -> list[Env]:
         if not self.__has_already_inject_envs:
             self.inject_envs()
             self.__has_already_inject_envs = True
@@ -297,7 +290,7 @@ class CommonTaskModel:
     def inject_upstreams(self):
         pass
 
-    def _get_upstreams(self) -> List[AnyTask]:
+    def _get_upstreams(self) -> list[AnyTask]:
         if not self.__has_already_inject_upstreams:
             self.inject_upstreams()
             self.__has_already_inject_upstreams = True
@@ -316,7 +309,7 @@ class CommonTaskModel:
     def inject_fallbacks(self):
         pass
 
-    def _get_fallbacks(self) -> List[AnyTask]:
+    def _get_fallbacks(self) -> list[AnyTask]:
         if not self.__has_already_inject_fallbacks:
             self.inject_fallbacks()
             self.__has_already_inject_fallbacks = True
@@ -328,7 +321,7 @@ class CommonTaskModel:
     def get_color(self) -> str:
         return self._color
 
-    def _get_env_files(self) -> List[EnvFile]:
+    def _get_env_files(self) -> list[EnvFile]:
         if not self.__has_already_inject_env_files:
             self.inject_env_files()
             self.__has_already_inject_env_files = True
@@ -346,11 +339,11 @@ class CommonTaskModel:
         additional_checkers = self.__complete_new_checkers(checkers)
         self._checkers = self._checkers + additional_checkers
 
-    def __complete_new_checkers(self, new_checkers: List[AnyTask]) -> List[AnyTask]:
+    def __complete_new_checkers(self, new_checkers: Iterable[AnyTask]) -> list[AnyTask]:
         """
         For internal use: copy and completing new checkers
         """
-        checkers: List[AnyTask] = [checker.copy() for checker in new_checkers]
+        checkers: list[AnyTask] = [checker.copy() for checker in new_checkers]
         for checker in checkers:
             checker.add_input(*self._get_inputs())
             checker.add_env(*self._get_envs())
@@ -360,7 +353,7 @@ class CommonTaskModel:
     def inject_checkers(self):
         pass
 
-    def _get_checkers(self) -> List[AnyTask]:
+    def _get_checkers(self) -> list[AnyTask]:
         if not self.__allow_add_checkers:
             self.inject_checkers()
             self.__allow_add_checkers = True

zrb/task/base_task/component/renderer.py

@@ -1,4 +1,6 @@
 import os
+from collections.abc import Mapping
+from typing import Any, Optional, Union
 
 import jinja2
 
@@ -6,7 +8,7 @@ from zrb.helper.render_data import DEFAULT_RENDER_DATA
 from zrb.helper.string.conversion import to_boolean
 from zrb.helper.string.jinja import is_probably_jinja
 from zrb.helper.typecheck import typechecked
-from zrb.helper.typing import Any, JinjaTemplate, Mapping, Optional, Union
+from zrb.helper.typing import JinjaTemplate
 from zrb.task.any_task import AnyTask
 
 

zrb/task/base_task/component/trackers.py

@@ -1,8 +1,8 @@
 import asyncio
 import time
+from typing import Optional, Union
 
 from zrb.helper.typecheck import typechecked
-from zrb.helper.typing import Optional, Union
 
 LOG_NAME_LENGTH = 20
 

zrb/task/checker.py

@@ -1,9 +1,10 @@
 import asyncio
+from collections.abc import Callable, Iterable
+from typing import Any, Optional, Union
 
 from zrb.helper.accessories.color import colored
 from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
-from zrb.helper.typing import Any, Callable, Iterable, Optional, Union
 from zrb.task.any_task import AnyTask
 from zrb.task.any_task_event_handler import (
     OnFailed,

zrb/task/cmd_task.py

@@ -7,21 +7,15 @@ import signal
 import subprocess
 import sys
 import time
+from collections.abc import Callable, Iterable
+from typing import Any, Optional, TypeVar, Union
 
 from zrb.config.config import DEFAULT_SHELL, LOGGING_LEVEL
 from zrb.helper.accessories.color import colored
+from zrb.helper.asyncio_task import stop_asyncio_sync
 from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
-from zrb.helper.typing import (
-    Any,
-    Callable,
-    Iterable,
-    JinjaTemplate,
-    List,
-    Optional,
-    TypeVar,
-    Union,
-)
+from zrb.helper.typing import JinjaTemplate
 from zrb.task.any_task import AnyTask
 from zrb.task.any_task_event_handler import (
     OnFailed,
@@ -111,7 +105,7 @@ class CmdTask(BaseTask):
         >>> runner.register(hello)
     """
 
-    _pids: List[int] = []
+    _pids: list[int] = []
     _global_state = CmdGlobalState()
 
     def __init__(
@@ -296,13 +290,7 @@ class CmdTask(BaseTask):
         _print_out_dark(f"Getting signal {signum}")
         for pid in self._pids:
             self.__kill_by_pid(pid)
-        tasks = asyncio.all_tasks()
-        for task in tasks:
-            try:
-                task.cancel()
-            except Exception as e:
-                self.print_err(e)
-        time.sleep(0.3)
+        stop_asyncio_sync()
         _print_out_dark(f"Exiting with signal {signum}")
         sys.exit(signum)
 

zrb/task/decorator.py

@@ -1,8 +1,10 @@
+from collections.abc import Callable, Iterable
+from typing import Any, Optional, Union
+
 from zrb.action.runner import Runner
 from zrb.helper.accessories.color import colored
 from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
-from zrb.helper.typing import Any, Callable, Iterable, Optional, Union
 from zrb.task.any_task import AnyTask
 from zrb.task.any_task_event_handler import (
     OnFailed,

zrb/task/docker_compose_start_task.py

@@ -1,18 +1,12 @@
 import os
 import pathlib
+from collections.abc import Callable, Iterable, Mapping
+from typing import Any, Optional, Union
 
 from zrb.helper.accessories.color import colored
 from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
-from zrb.helper.typing import (
-    Any,
-    Callable,
-    Iterable,
-    JinjaTemplate,
-    Mapping,
-    Optional,
-    Union,
-)
+from zrb.helper.typing import JinjaTemplate
 from zrb.task.any_task import AnyTask
 from zrb.task.any_task_event_handler import (
     OnFailed,
@@ -123,29 +117,30 @@ class DockerComposeStartTask(DockerComposeTask):
         )
 
     def get_cmd_script(self, *args: Any, **kwargs: Any) -> str:
-        cmd_str = "\n".join(
-            [
-                # setup
-                self._create_cmd_script(
-                    self._setup_cmd_path, self._setup_cmd, *args, **kwargs
-                ),
-                # compose start
-                self._get_docker_compose_cmd_script(
-                    compose_cmd="up",
-                    compose_options=self._compose_options,
-                    compose_flags=list(self._compose_flags) + ["-d"],
-                    compose_args=self._compose_args,
-                    *args,
-                ),
-                # compose log
-                self._get_docker_compose_cmd_script(
-                    compose_cmd="logs",
-                    compose_options={},
-                    compose_flags=["-f"],
-                    compose_args=[],
-                    *args,
-                ),
-            ]
+        # setup
+        setup_cmd = self._create_cmd_script(
+            self._setup_cmd_path, self._setup_cmd, *args, **kwargs
         )
+        cmd_list = [setup_cmd] if setup_cmd.strip() != "" else []
+        # compose command
+        cmd_list = cmd_list + [
+            # compose start
+            self._get_docker_compose_cmd_script(
+                compose_cmd="up",
+                compose_options=self._compose_options,
+                compose_flags=list(self._compose_flags) + ["-d"],
+                compose_args=self._compose_args,
+                *args,
+            ),
+            # compose log
+            self._get_docker_compose_cmd_script(
+                compose_cmd="logs",
+                compose_options={},
+                compose_flags=["-f"],
+                compose_args=[],
+                *args,
+            ),
+        ]
+        cmd_str = "\n".join(cmd_list)
         self.log_info(f"Command: {cmd_str}")
         return cmd_str

zrb/task/docker_compose_task.py

@@ -1,5 +1,7 @@
 import os
 import pathlib
+from collections.abc import Callable, Iterable, Mapping
+from typing import Any, Optional, TypeVar, Union
 
 from zrb.config.config import CONTAINER_BACKEND
 from zrb.helper.accessories.color import colored
@@ -10,17 +12,7 @@ from zrb.helper.log import logger
 from zrb.helper.string.conversion import to_cli_name
 from zrb.helper.string.modification import double_quote
 from zrb.helper.typecheck import typechecked
-from zrb.helper.typing import (
-    Any,
-    Callable,
-    Iterable,
-    JinjaTemplate,
-    List,
-    Mapping,
-    Optional,
-    TypeVar,
-    Union,
-)
+from zrb.helper.typing import JinjaTemplate
 from zrb.helper.util import to_snake_case
 from zrb.task.any_task import AnyTask
 from zrb.task.any_task_event_handler import (
@@ -71,14 +63,14 @@ ensure_zrb_network_task = _get_ensure_zrb_network_task(CONTAINER_BACKEND)
 
 @typechecked
 class ServiceConfig:
-    def __init__(self, envs: List[Env] = [], env_files: List[EnvFile] = []):
+    def __init__(self, envs: list[Env] = [], env_files: list[EnvFile] = []):
         self._envs = envs
         self._env_files = env_files
 
-    def get_envs(self) -> List[Env]:
+    def get_envs(self) -> list[Env]:
         return self._envs
 
-    def get_env_files(self) -> List[EnvFile]:
+    def get_env_files(self) -> list[EnvFile]:
         return self._env_files
 
 
@@ -224,7 +216,7 @@ class DockerComposeTask(CmdTask):
     def __generate_compose_runtime_file(self):
         compose_data = read_compose_file(self._compose_template_file)
         for service, service_config in self._compose_service_configs.items():
-            envs: List[Env] = []
+            envs: list[Env] = []
             env_files = service_config.get_env_files()
             for env_file in env_files:
                 envs += env_file.get_envs()
@@ -233,7 +225,7 @@ class DockerComposeTask(CmdTask):
         write_compose_file(self._compose_runtime_file, compose_data)
 
     def __apply_service_env(
-        self, compose_data: Any, service_name: str, envs: List[Env]
+        self, compose_data: Any, service_name: str, envs: list[Env]
     ) -> Any:
         # service not found
         service_map = compose_data["services"]
@@ -269,7 +261,7 @@ class DockerComposeTask(CmdTask):
         return compose_data
 
     def __get_service_new_env_map(
-        self, service_name: str, service_env_map: Mapping[str, str], new_envs: List[Env]
+        self, service_name: str, service_env_map: Mapping[str, str], new_envs: list[Env]
     ) -> Mapping[str, str]:
         new_service_envs: Mapping[str, str] = {}
         for env in new_envs:
@@ -280,9 +272,9 @@ class DockerComposeTask(CmdTask):
         return new_service_envs
 
     def __get_service_new_env_list(
-        self, service_name: str, service_env_list: List[str], new_envs: List[Env]
-    ) -> List[str]:
-        new_service_envs: List[str] = []
+        self, service_name: str, service_env_list: list[str], new_envs: list[Env]
+    ) -> list[str]:
+        new_service_envs: list[str] = []
         for env in new_envs:
             should_be_added = 0 == len(
                 [
@@ -341,22 +333,22 @@ class DockerComposeTask(CmdTask):
         raise Exception(f"Invalid compose file: {compose_file}")
 
     def get_cmd_script(self, *args: Any, **kwargs: Any) -> str:
-        cmd_str = "\n".join(
-            [
-                # setup
-                self._create_cmd_script(
-                    self._setup_cmd_path, self._setup_cmd, *args, **kwargs
-                ),
-                # compose
-                self._get_docker_compose_cmd_script(
-                    compose_cmd=self._compose_cmd,
-                    compose_options=self._compose_options,
-                    compose_flags=self._compose_flags,
-                    compose_args=self._compose_args,
-                    *args,
-                ),
-            ]
+        # setup
+        setup_cmd = self._create_cmd_script(
+            self._setup_cmd_path, self._setup_cmd, *args, **kwargs
+        )
+        cmd_list = [setup_cmd] if setup_cmd.strip() != "" else []
+        # compose command
+        cmd_list.append(
+            self._get_docker_compose_cmd_script(
+                compose_cmd=self._compose_cmd,
+                compose_options=self._compose_options,
+                compose_flags=self._compose_flags,
+                compose_args=self._compose_args,
+                *args,
+            )
         )
+        cmd_str = "\n".join(cmd_list)
         self.log_info(f"Command: {cmd_str}")
         return cmd_str
 

zrb/task/flow_task.py

@@ -1,7 +1,9 @@
+from collections.abc import Callable, Iterable
+from typing import Optional, TypeVar, Union
+
 from zrb.helper.accessories.color import colored
 from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
-from zrb.helper.typing import Callable, Iterable, List, Optional, TypeVar, Union
 from zrb.task.any_task import AnyTask
 from zrb.task.any_task_event_handler import (
     OnFailed,
@@ -48,7 +50,7 @@ class FlowTask(BaseTask):
         checking_interval: Union[float, int] = 0.05,
         retry: int = 2,
         retry_interval: Union[float, int] = 1,
-        steps: List[Union[AnyTask, List[AnyTask]]] = [],
+        steps: list[Union[AnyTask, list[AnyTask]]] = [],
         should_execute: Union[bool, str, Callable[..., bool]] = True,
         return_upstream_result: bool = False,
     ):
@@ -83,7 +85,6 @@ class FlowTask(BaseTask):
             retry_interval=retry_interval,
             should_execute=should_execute,
             return_upstream_result=return_upstream_result,
-            run=lambda *args, **kwargs: kwargs.get("_task").print_out("🆗"),
         )
 
     def copy(self) -> TFlowTask:
@@ -91,12 +92,12 @@ class FlowTask(BaseTask):
 
     def _create_flow_upstreams(
         self,
-        steps: List[Union[AnyTask, List[AnyTask]]],
-        upstreams: List[AnyTask],
-        inputs: List[AnyInput],
-        envs: List[Env],
-        env_files: List[EnvFile],
-    ) -> List[AnyTask]:
+        steps: list[Union[AnyTask, list[AnyTask]]],
+        upstreams: list[AnyTask],
+        inputs: list[AnyInput],
+        envs: list[Env],
+        env_files: list[EnvFile],
+    ) -> list[AnyTask]:
         flow_upstreams = upstreams
         for step in steps:
             tasks = [task.copy() for task in self._step_to_tasks(step)]
@@ -110,20 +111,20 @@ class FlowTask(BaseTask):
             flow_upstreams = new_upstreams
         return flow_upstreams
 
-    def _step_to_tasks(self, step: Union[AnyTask, List[AnyTask]]) -> List[AnyTask]:
+    def _step_to_tasks(self, step: Union[AnyTask, list[AnyTask]]) -> list[AnyTask]:
         if isinstance(step, AnyTask):
             return [step]
         return step
 
     def _create_embeded_tasks(
         self,
-        tasks: List[AnyTask],
-        upstreams: List[AnyTask],
-        inputs: List[AnyInput],
-        envs: List[Env],
-        env_files: List[EnvFile],
-    ) -> List[AnyTask]:
-        embeded_tasks: List[AnyTask] = []
+        tasks: list[AnyTask],
+        upstreams: list[AnyTask],
+        inputs: list[AnyInput],
+        envs: list[Env],
+        env_files: list[EnvFile],
+    ) -> list[AnyTask]:
+        embeded_tasks: list[AnyTask] = []
         for embeded_task in tasks:
             embeded_task_upstreams = self._get_all_upstreams(tasks=[embeded_task])
             for embeded_task_upstream in embeded_task_upstreams:
@@ -134,7 +135,7 @@ class FlowTask(BaseTask):
             embeded_tasks.append(embeded_task)
         return embeded_tasks
 
-    def _get_all_upstreams(self, tasks: List[AnyTask]):
+    def _get_all_upstreams(self, tasks: list[AnyTask]):
         all_upstreams = []
         for task in tasks:
             upstreams = task._get_upstreams()