diff --git a/src/codeflare_sdk/ray/cluster/cluster.py b/src/codeflare_sdk/ray/cluster/cluster.py index fd059277..a3f34554 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)