Skip to content

Commit 364d81d

Browse files
address jeff's comment
1 parent b010eb6 commit 364d81d

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

examples/pathways_example.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
import keras_remote
1010

1111

12-
# A simple model that will be executed remotely
12+
# A simple model that will be executed remotely on pathways
1313
@keras_remote.run(accelerator="v5litepod-1", backend="pathways")
1414
def train_simple_model():
1515
print("Running Pathways job on JAX Backend!")

keras_remote/backend/pathways_client.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ def submit_pathways_job(
117117
raise RuntimeError(
118118
"LeaderWorkerSet CRD not found. Please ensure it is "
119119
"installed on the cluster. You can install it by running "
120-
"the `keras-remote infra up` command, or by following the "
120+
"the `keras-remote up` command, or by following the "
121121
"official LWS installation guide."
122122
) from e
123123
else:
@@ -218,7 +218,10 @@ def cleanup_job(job_name, namespace="default"):
218218
)
219219
logger.info(f"Deleted LeaderWorkerSet: {job_name}")
220220
except ApiException as e:
221-
if e.status != 404:
221+
if e.status == 404:
222+
# Job already deleted
223+
pass
224+
else:
222225
logger.warning(
223226
"Failed to delete LeaderWorkerSet %s: %s",
224227
job_name,

0 commit comments

Comments
 (0)