lightning-sdk 2025.8.8__py3-none-any.whl → 2025.8.14__py3-none-any.whl

lightning_sdk/__init__.py

@@ -32,6 +32,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "2025.08.08"
+__version__ = "2025.08.14"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/api/studio_api.py

@@ -253,8 +253,25 @@ class StudioApi:
         )
 
         # Wait until it's time to switch
+        requested_was_found = False
+        startup_status = None
         while True:
-            startup_status = self.get_studio_status(studio_id, teamspace_id).requested.startup_status
+            status = self.get_studio_status(studio_id, teamspace_id)
+            requested_machine = status.requested
+
+            if requested_machine is not None:
+                requested_was_found = True
+                startup_status = requested_machine.startup_status
+
+            # if the requested machine was found in the past, use the in_use status instead.
+            # it might be that it either was cancelled or it actually is ready.
+            # Either way, since we're actually blocking below for the in use startup status
+            # it's safe to switch at this point
+            elif requested_was_found:
+                in_use_machine = status.in_use
+                if in_use_machine is not None:
+                    startup_status = in_use_machine.startup_status
+
             if startup_status and startup_status.initial_restore_finished:
                 break
             time.sleep(1)
@@ -263,7 +280,10 @@ class StudioApi:
 
         # Wait until the new machine is ready to use
         while True:
-            startup_status = self.get_studio_status(studio_id, teamspace_id).in_use.startup_status
+            in_use = self.get_studio_status(studio_id, teamspace_id).in_use
+            if in_use is None:
+                continue
+            startup_status = in_use.startup_status
             if startup_status and startup_status.top_up_restore_finished:
                 break
             time.sleep(1)
@@ -441,7 +461,9 @@ class StudioApi:
             body=body,
         )
 
-    def duplicate_studio(self, studio_id: str, teamspace_id: str, target_teamspace_id: str) -> Dict[str, Any]:
+    def duplicate_studio(
+        self, studio_id: str, teamspace_id: str, target_teamspace_id: str, machine: Machine = Machine.CPU
+    ) -> Dict[str, Any]:
         """Duplicates the given Studio from a given Teamspace into a given target Teamspace."""
         target_teamspace = self._client.projects_service_get_project(target_teamspace_id)
         init_kwargs = {}
@@ -464,7 +486,7 @@ class StudioApi:
         init_kwargs["name"] = new_cloudspace.name
         init_kwargs["teamspace"] = target_teamspace.name
 
-        self.start_studio(new_cloudspace.id, target_teamspace_id, Machine.CPU, False, None)
+        self.start_studio(new_cloudspace.id, target_teamspace_id, machine, False, None)
         return init_kwargs
 
     def delete_studio(self, studio_id: str, teamspace_id: str) -> None:

lightning_sdk/job/base.py

@@ -4,7 +4,7 @@ from typing import TYPE_CHECKING, Any, Dict, Optional, TypedDict, Union
 
 from lightning_sdk.api.cloud_account_api import CloudAccountApi
 from lightning_sdk.api.utils import _get_cloud_url
-from lightning_sdk.utils.resolve import _resolve_deprecated_cluster, _resolve_teamspace, in_studio
+from lightning_sdk.utils.resolve import _resolve_deprecated_cluster, _resolve_teamspace, in_studio, skip_studio_setup
 
 if TYPE_CHECKING:
     from lightning_sdk.machine import CloudProvider, Machine
@@ -145,9 +145,15 @@ class _BaseJob(ABC):
 
         if image is None:
             if not isinstance(studio, Studio):
-                studio = Studio(
-                    name=studio, teamspace=teamspace, org=org, user=user, cloud_account=cloud_account, create_ok=False
-                )
+                with skip_studio_setup():
+                    studio = Studio(
+                        name=studio,
+                        teamspace=teamspace,
+                        org=org,
+                        user=user,
+                        cloud_account=cloud_account,
+                        create_ok=False,
+                    )
 
             # studio is a Studio instance at this point
             if teamspace is None:
@@ -192,7 +198,8 @@ class _BaseJob(ABC):
                 )
             if cloud_account is None and in_studio():
                 try:
-                    resolve_studio = Studio(teamspace=teamspace, user=user, org=org)
+                    with skip_studio_setup():
+                        resolve_studio = Studio(teamspace=teamspace, user=user, org=org)
                     cloud_account = resolve_studio.cloud_account
                 except (ValueError, ApiException):
                     warnings.warn("Could not infer cloud account from studio. Using teamspace default.")

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -148,6 +148,7 @@ from lightning_sdk.lightning_cloud.openapi.models.id_publications_body1 import I
 from lightning_sdk.lightning_cloud.openapi.models.id_release_body import IdReleaseBody
 from lightning_sdk.lightning_cloud.openapi.models.id_reportlogsactivity_body import IdReportlogsactivityBody
 from lightning_sdk.lightning_cloud.openapi.models.id_reportrestarttimings_body import IdReportrestarttimingsBody
+from lightning_sdk.lightning_cloud.openapi.models.id_sleepconfig_body import IdSleepconfigBody
 from lightning_sdk.lightning_cloud.openapi.models.id_start_body import IdStartBody
 from lightning_sdk.lightning_cloud.openapi.models.id_transfer_body import IdTransferBody
 from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body import IdUploadsBody
@@ -654,6 +655,9 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloudy_experts_respons
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_accelerators_response import V1ListClusterAcceleratorsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_availabilities_response import V1ListClusterAvailabilitiesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_capacity_reservations_response import V1ListClusterCapacityReservationsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_metrics_response import V1ListClusterMetricsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_namespace_metrics_response import V1ListClusterNamespaceMetricsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_namespace_user_metrics_response import V1ListClusterNamespaceUserMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_proxies_response import V1ListClusterProxiesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_usage_restrictions_response import V1ListClusterUsageRestrictionsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_clusters_response import V1ListClustersResponse
@@ -753,6 +757,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_login_response import V1Log
 from lightning_sdk.lightning_cloud.openapi.models.v1_logout_request import V1LogoutRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_logout_response import V1LogoutResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_logs_response import V1LogsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_lustre_data_connection import V1LustreDataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_machines_selector import V1MachinesSelector
 from lightning_sdk.lightning_cloud.openapi.models.v1_magic_link_login_request import V1MagicLinkLoginRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_magic_link_login_response import V1MagicLinkLoginResponse
@@ -783,6 +788,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_fault_tol
 from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_state import V1MultiMachineJobState
 from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_status import V1MultiMachineJobStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_named_get_logger_metrics import V1NamedGetLoggerMetrics
+from lightning_sdk.lightning_cloud.openapi.models.v1_namespace_metrics import V1NamespaceMetrics
+from lightning_sdk.lightning_cloud.openapi.models.v1_namespace_user_metrics import V1NamespaceUserMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_nebius_direct_v1 import V1NebiusDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_network_config import V1NetworkConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_new_feature import V1NewFeature

lightning_sdk/lightning_cloud/openapi/api/cloud_space_environment_template_service_api.py

@@ -248,6 +248,7 @@ class CloudSpaceEnvironmentTemplateServiceApi(object):
         :param async_req bool
         :param str id: (required)
         :param str org_id:
+        :param str project_id:
         :return: V1CloudSpaceEnvironmentTemplate
                  If the method is called asynchronously,
                  returns the request thread.
@@ -270,12 +271,13 @@ class CloudSpaceEnvironmentTemplateServiceApi(object):
         :param async_req bool
         :param str id: (required)
         :param str org_id:
+        :param str project_id:
         :return: V1CloudSpaceEnvironmentTemplate
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['id', 'org_id']  # noqa: E501
+        all_params = ['id', 'org_id', 'project_id']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -304,6 +306,8 @@ class CloudSpaceEnvironmentTemplateServiceApi(object):
         query_params = []
         if 'org_id' in params:
             query_params.append(('orgId', params['org_id']))  # noqa: E501
+        if 'project_id' in params:
+            query_params.append(('projectId', params['project_id']))  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py

@@ -9066,6 +9066,119 @@ class CloudSpaceServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def cloud_space_service_update_cloud_space_sleep_config(self, body: 'IdSleepconfigBody', project_id: 'str', id: 'str', **kwargs) -> 'V1CloudSpaceInstanceConfig':  # noqa: E501
+        """cloud_space_service_update_cloud_space_sleep_config  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.cloud_space_service_update_cloud_space_sleep_config(body, project_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param IdSleepconfigBody body: (required)
+        :param str project_id: (required)
+        :param str id: (required)
+        :return: V1CloudSpaceInstanceConfig
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.cloud_space_service_update_cloud_space_sleep_config_with_http_info(body, project_id, id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.cloud_space_service_update_cloud_space_sleep_config_with_http_info(body, project_id, id, **kwargs)  # noqa: E501
+            return data
+
+    def cloud_space_service_update_cloud_space_sleep_config_with_http_info(self, body: 'IdSleepconfigBody', project_id: 'str', id: 'str', **kwargs) -> 'V1CloudSpaceInstanceConfig':  # noqa: E501
+        """cloud_space_service_update_cloud_space_sleep_config  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.cloud_space_service_update_cloud_space_sleep_config_with_http_info(body, project_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param IdSleepconfigBody body: (required)
+        :param str project_id: (required)
+        :param str id: (required)
+        :return: V1CloudSpaceInstanceConfig
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', 'project_id', 'id']  # noqa: E501
+        all_params.append('async_req')
+        all_params.append('_return_http_data_only')
+        all_params.append('_preload_content')
+        all_params.append('_request_timeout')
+
+        params = locals()
+        for key, val in six.iteritems(params['kwargs']):
+            if key not in all_params:
+                raise TypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method cloud_space_service_update_cloud_space_sleep_config" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # verify the required parameter 'body' is set
+        if ('body' not in params or
+                params['body'] is None):
+            raise ValueError("Missing the required parameter `body` when calling `cloud_space_service_update_cloud_space_sleep_config`")  # noqa: E501
+        # verify the required parameter 'project_id' is set
+        if ('project_id' not in params or
+                params['project_id'] is None):
+            raise ValueError("Missing the required parameter `project_id` when calling `cloud_space_service_update_cloud_space_sleep_config`")  # noqa: E501
+        # verify the required parameter 'id' is set
+        if ('id' not in params or
+                params['id'] is None):
+            raise ValueError("Missing the required parameter `id` when calling `cloud_space_service_update_cloud_space_sleep_config`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'id' in params:
+            path_params['id'] = params['id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'body' in params:
+            body_params = params['body']
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # HTTP header `Content-Type`
+        header_params['Content-Type'] = self.api_client.select_header_content_type(  # noqa: E501
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/v1/projects/{projectId}/cloudspaces/{id}/sleepconfig', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1CloudSpaceInstanceConfig',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=params.get('async_req'),
+            _return_http_data_only=params.get('_return_http_data_only'),
+            _preload_content=params.get('_preload_content', True),
+            _request_timeout=params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def cloud_space_service_update_cloud_space_version(self, body: 'VersionsIdBody', project_id: 'str', cloud_space_id: 'str', id: 'str', **kwargs) -> 'V1CloudSpaceVersion':  # noqa: E501
         """cloud_space_service_update_cloud_space_version  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/api/cloudy_service_api.py

@@ -43,103 +43,6 @@ class CloudyServiceApi(object):
             api_client = ApiClient()
         self.api_client = api_client
 
-    def cloudy_service_create_user_cloudy_settings(self, body: 'V1CloudySettings', **kwargs) -> 'V1CloudySettings':  # noqa: E501
-        """cloudy_service_create_user_cloudy_settings  # noqa: E501
-
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.cloudy_service_create_user_cloudy_settings(body, async_req=True)
-        >>> result = thread.get()
-
-        :param async_req bool
-        :param V1CloudySettings body: (required)
-        :return: V1CloudySettings
-                 If the method is called asynchronously,
-                 returns the request thread.
-        """
-        kwargs['_return_http_data_only'] = True
-        if kwargs.get('async_req'):
-            return self.cloudy_service_create_user_cloudy_settings_with_http_info(body, **kwargs)  # noqa: E501
-        else:
-            (data) = self.cloudy_service_create_user_cloudy_settings_with_http_info(body, **kwargs)  # noqa: E501
-            return data
-
-    def cloudy_service_create_user_cloudy_settings_with_http_info(self, body: 'V1CloudySettings', **kwargs) -> 'V1CloudySettings':  # noqa: E501
-        """cloudy_service_create_user_cloudy_settings  # noqa: E501
-
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.cloudy_service_create_user_cloudy_settings_with_http_info(body, async_req=True)
-        >>> result = thread.get()
-
-        :param async_req bool
-        :param V1CloudySettings body: (required)
-        :return: V1CloudySettings
-                 If the method is called asynchronously,
-                 returns the request thread.
-        """
-
-        all_params = ['body']  # noqa: E501
-        all_params.append('async_req')
-        all_params.append('_return_http_data_only')
-        all_params.append('_preload_content')
-        all_params.append('_request_timeout')
-
-        params = locals()
-        for key, val in six.iteritems(params['kwargs']):
-            if key not in all_params:
-                raise TypeError(
-                    "Got an unexpected keyword argument '%s'"
-                    " to method cloudy_service_create_user_cloudy_settings" % key
-                )
-            params[key] = val
-        del params['kwargs']
-        # verify the required parameter 'body' is set
-        if ('body' not in params or
-                params['body'] is None):
-            raise ValueError("Missing the required parameter `body` when calling `cloudy_service_create_user_cloudy_settings`")  # noqa: E501
-
-        collection_formats = {}
-
-        path_params = {}
-
-        query_params = []
-
-        header_params = {}
-
-        form_params = []
-        local_var_files = {}
-
-        body_params = None
-        if 'body' in params:
-            body_params = params['body']
-        # HTTP header `Accept`
-        header_params['Accept'] = self.api_client.select_header_accept(
-            ['application/json'])  # noqa: E501
-
-        # HTTP header `Content-Type`
-        header_params['Content-Type'] = self.api_client.select_header_content_type(  # noqa: E501
-            ['application/json'])  # noqa: E501
-
-        # Authentication setting
-        auth_settings = []  # noqa: E501
-
-        return self.api_client.call_api(
-            '/v1/cloudy/settings/user', 'POST',
-            path_params,
-            query_params,
-            header_params,
-            body=body_params,
-            post_params=form_params,
-            files=local_var_files,
-            response_type='V1CloudySettings',  # noqa: E501
-            auth_settings=auth_settings,
-            async_req=params.get('async_req'),
-            _return_http_data_only=params.get('_return_http_data_only'),
-            _preload_content=params.get('_preload_content', True),
-            _request_timeout=params.get('_request_timeout'),
-            collection_formats=collection_formats)
-
     def cloudy_service_get_user_cloudy_settings(self, user_id: 'str', **kwargs) -> 'V1CloudySettings':  # noqa: E501
         """cloudy_service_get_user_cloudy_settings  # noqa: E501