fix black errors

Signed-off-by: Salvatore Daniele <[email protected]>
SalDaniele · Jul 3, 2024 · 3a73ff2 · 3a73ff2
1 parent 6f0fb56
commit 3a73ff2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/assistedInstallerService.py b/assistedInstallerService.py
@@ -54,7 +54,6 @@ class AssistedInstallerService:
     CONTROLLER_IMAGE = "registry.redhat.io/rhai-tech-preview/assisted-installer-reporter-rhel8:v1.0.0-418"
     AGENT_DOCKER_IMAGE = "registry.redhat.io/rhai-tech-preview/assisted-installer-agent-rhel8:v1.0.0-315"
 
-
     def __init__(self, version: str, ip: str, proxy: Optional[str] = None, noproxy: Optional[str] = None, branch: str = "master"):
         self._version = version
         self._ip = ip

diff --git a/clusterHost.py b/clusterHost.py
@@ -205,7 +205,7 @@ def _wait_for_boot(self, nodes: list[ClusterNode], desired_ip_range: tuple[str,
         if not nodes:
             return
 
-        def wait_state(state_name: str, get_states: Callable[[], Dict[str, bool]]):
+        def wait_state(state_name: str, get_states: Callable[[], Dict[str, bool]]) -> None:
             states = get_states()
             for try_count in itertools.count(0):
                 new_states = get_states()
@@ -220,10 +220,12 @@ def wait_state(state_name: str, get_states: Callable[[], Dict[str, bool]]):
 
         def boot_state() -> Dict[str, bool]:
             return {node.config.name: node.has_booted() for node in nodes}
+
         wait_state("boot", boot_state)
 
-        def post_boot_state():
+        def post_boot_state() -> Dict[str, bool]:
             return {node.config.name: node.post_boot(desired_ip_range) for node in nodes}
+
         wait_state("post_boot", post_boot_state)
 
         for node in nodes: