From 3a73ff2916518060194e296881797e946eaf7517 Mon Sep 17 00:00:00 2001 From: Salvatore Daniele Date: Wed, 3 Jul 2024 12:04:28 -0400 Subject: [PATCH] fix black errors Signed-off-by: Salvatore Daniele --- assistedInstallerService.py | 1 - clusterHost.py | 6 ++++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/assistedInstallerService.py b/assistedInstallerService.py index 752b5866..f8d5c5c7 100644 --- 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 index 38ebae09..d4e17bba 100644 --- 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: