From c99f3fda48a518ace8c515930d129496356a01ed Mon Sep 17 00:00:00 2001 From: ChristianZaccaria Date: Wed, 10 Apr 2024 17:39:17 +0100 Subject: [PATCH] Bug-fix wait_ready functionality --- src/codeflare_sdk/cluster/cluster.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/codeflare_sdk/cluster/cluster.py b/src/codeflare_sdk/cluster/cluster.py index 81f7a7116..60f968d2a 100644 --- a/src/codeflare_sdk/cluster/cluster.py +++ b/src/codeflare_sdk/cluster/cluster.py @@ -848,7 +848,7 @@ def _get_app_wrappers( def _map_to_ray_cluster(rc) -> Optional[RayCluster]: - if "state" in rc["status"]: + if "status" in rc and "state" in rc["status"]: status = RayClusterStatus(rc["status"]["state"].lower()) else: status = RayClusterStatus.UNKNOWN