Skip to content

Commit 8085ae3

Browse files
code reformat
1 parent 08881d9 commit 8085ae3

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

keras_remote/backend/pathways_client.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@
2020
LWS_PLURAL = "leaderworkersets"
2121

2222

23-
2423
def _get_lws_version(group=LWS_GROUP):
2524
"""Get the preferred version for the LeaderWorkerSet API."""
2625
_load_kube_config()
@@ -159,7 +158,9 @@ def wait_for_job(
159158
# Pod might not be created yet
160159
pod = None
161160
else:
162-
raise RuntimeError(f"Failed to read leader pod status: {e.reason}") from e
161+
raise RuntimeError(
162+
f"Failed to read leader pod status: {e.reason}"
163+
) from e
163164

164165
if pod is not None and pod.status.container_statuses:
165166
container_status = pod.status.container_statuses[0]
@@ -179,7 +180,9 @@ def wait_for_job(
179180
# Check last state (in case it restarted)
180181
if container_status.last_state.terminated:
181182
if container_status.last_state.terminated.exit_code == 0:
182-
logger.info(f"[REMOTE] Job {job_name} completed successfully (restarted)")
183+
logger.info(
184+
f"[REMOTE] Job {job_name} completed successfully (restarted)"
185+
)
183186
return "success"
184187
else:
185188
_print_pod_logs(core_v1, job_name, namespace)

0 commit comments

Comments
 (0)