From 1e5f4d4a2dc4f42c6ee5913756eb843f6a19fe7a Mon Sep 17 00:00:00 2001 From: Bobbins228 Date: Wed, 20 Nov 2024 16:49:04 +0000 Subject: [PATCH] fix: get_cluster with job submission --- src/codeflare_sdk/ray/cluster/cluster.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/codeflare_sdk/ray/cluster/cluster.py b/src/codeflare_sdk/ray/cluster/cluster.py index fd0592771..a3f345547 100644 --- a/src/codeflare_sdk/ray/cluster/cluster.py +++ b/src/codeflare_sdk/ray/cluster/cluster.py @@ -72,6 +72,7 @@ def __init__(self, config: ClusterConfiguration): request. """ self.config = config + self._job_submission_client = None if self.config is None: warnings.warn( "Please provide a ClusterConfiguration to initialise the Cluster object" @@ -80,7 +81,6 @@ def __init__(self, config: ClusterConfiguration): else: self.resource_yaml = self.create_resource() - self._job_submission_client = None if is_notebook(): cluster_up_down_buttons(self)