lightning-sdk 2025.10.22__py3-none-any.whl → 2025.10.23__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_reset_api_key_response.py

@@ -0,0 +1,123 @@
+# coding: utf-8
+
+"""
+    external/v1/auth_service.proto
+
+    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
+
+    OpenAPI spec version: version not set
+    
+    Generated by: https://github.com/swagger-api/swagger-codegen.git
+
+    NOTE
+    ----
+    standard swagger-codegen-cli for this python client has been modified
+    by custom templates. The purpose of these templates is to include
+    typing information in the API and Model code. Please refer to the
+    main grid repository for more info
+"""
+
+import pprint
+import re  # noqa: F401
+
+from typing import TYPE_CHECKING
+
+import six
+
+if TYPE_CHECKING:
+    from datetime import datetime
+    from lightning_sdk.lightning_cloud.openapi.models import *
+
+class V1ResetAPIKeyResponse(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+    """
+    Attributes:
+      swagger_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    swagger_types = {
+        'api_key': 'str'
+    }
+
+    attribute_map = {
+        'api_key': 'apiKey'
+    }
+
+    def __init__(self, api_key: 'str' =None):  # noqa: E501
+        """V1ResetAPIKeyResponse - a model defined in Swagger"""  # noqa: E501
+        self._api_key = None
+        self.discriminator = None
+        if api_key is not None:
+            self.api_key = api_key
+
+    @property
+    def api_key(self) -> 'str':
+        """Gets the api_key of this V1ResetAPIKeyResponse.  # noqa: E501
+
+
+        :return: The api_key of this V1ResetAPIKeyResponse.  # noqa: E501
+        :rtype: str
+        """
+        return self._api_key
+
+    @api_key.setter
+    def api_key(self, api_key: 'str'):
+        """Sets the api_key of this V1ResetAPIKeyResponse.
+
+
+        :param api_key: The api_key of this V1ResetAPIKeyResponse.  # noqa: E501
+        :type: str
+        """
+
+        self._api_key = api_key
+
+    def to_dict(self) -> dict:
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.swagger_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+        if issubclass(V1ResetAPIKeyResponse, dict):
+            for key, value in self.items():
+                result[key] = value
+
+        return result
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self) -> str:
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other: 'V1ResetAPIKeyResponse') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1ResetAPIKeyResponse):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1ResetAPIKeyResponse') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -63,7 +63,6 @@ class V1UserFeatures(object):
         'deployment_persistent_disk': 'bool',
         'drive_v2': 'bool',
         'enterprise_compute_admin': 'bool',
-        'f227': 'bool',
         'f234': 'bool',
         'f236': 'bool',
         'f239': 'bool',
@@ -72,25 +71,25 @@ class V1UserFeatures(object):
         'f243': 'bool',
         'f245': 'bool',
         'f247': 'bool',
-        'f248': 'bool',
         'f250': 'bool',
         'f252': 'bool',
         'f253': 'bool',
         'f254': 'bool',
         'f255': 'bool',
-        'f257': 'bool',
         'f258': 'bool',
         'f259': 'bool',
         'f260': 'bool',
         'f261': 'bool',
         'f262': 'bool',
         'f263': 'bool',
-        'f264': 'bool',
         'f265': 'bool',
         'f266': 'bool',
+        'f267': 'bool',
+        'f268': 'bool',
+        'f269': 'bool',
+        'f270': 'bool',
         'fair_share': 'bool',
         'featured_studios_admin': 'bool',
-        'gcs_connections_optimized': 'bool',
         'job_artifacts_v2': 'bool',
         'kubernetes_cluster_ui': 'bool',
         'kubernetes_clusters': 'bool',
@@ -155,7 +154,6 @@ class V1UserFeatures(object):
         'deployment_persistent_disk': 'deploymentPersistentDisk',
         'drive_v2': 'driveV2',
         'enterprise_compute_admin': 'enterpriseComputeAdmin',
-        'f227': 'f227',
         'f234': 'f234',
         'f236': 'f236',
         'f239': 'f239',
@@ -164,25 +162,25 @@ class V1UserFeatures(object):
         'f243': 'f243',
         'f245': 'f245',
         'f247': 'f247',
-        'f248': 'f248',
         'f250': 'f250',
         'f252': 'f252',
         'f253': 'f253',
         'f254': 'f254',
         'f255': 'f255',
-        'f257': 'f257',
         'f258': 'f258',
         'f259': 'f259',
         'f260': 'f260',
         'f261': 'f261',
         'f262': 'f262',
         'f263': 'f263',
-        'f264': 'f264',
         'f265': 'f265',
         'f266': 'f266',
+        'f267': 'f267',
+        'f268': 'f268',
+        'f269': 'f269',
+        'f270': 'f270',
         'fair_share': 'fairShare',
         'featured_studios_admin': 'featuredStudiosAdmin',
-        'gcs_connections_optimized': 'gcsConnectionsOptimized',
         'job_artifacts_v2': 'jobArtifactsV2',
         'kubernetes_cluster_ui': 'kubernetesClusterUi',
         'kubernetes_clusters': 'kubernetesClusters',
@@ -224,7 +222,7 @@ class V1UserFeatures(object):
         'writable_s3_connections': 'writableS3Connections'
     }
 
-    def __init__(self, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, b2c_experience: 'bool' =None, byo_machine_type: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, chat_models: 'bool' =None, cloudspace_schedules: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, control_center_monitoring: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_persistent_disk: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, f227: 'bool' =None, f234: 'bool' =None, f236: 'bool' =None, f239: 'bool' =None, f240: 'bool' =None, f241: 'bool' =None, f243: 'bool' =None, f245: 'bool' =None, f247: 'bool' =None, f248: 'bool' =None, f250: 'bool' =None, f252: 'bool' =None, f253: 'bool' =None, f254: 'bool' =None, f255: 'bool' =None, f257: 'bool' =None, f258: 'bool' =None, f259: 'bool' =None, f260: 'bool' =None, f261: 'bool' =None, f262: 'bool' =None, f263: 'bool' =None, f264: 'bool' =None, f265: 'bool' =None, f266: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcs_connections_optimized: 'bool' =None, job_artifacts_v2: 'bool' =None, kubernetes_cluster_ui: 'bool' =None, kubernetes_clusters: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_api_dashboard: 'bool' =None, multiple_studio_versions: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, persistent_disk: 'bool' =None, plugin_distributed: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_sweeps: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, product_license: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, reserved_machines_tab: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, show_dev_admin: 'bool' =None, slurm: 'bool' =None, specialised_studios: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_sharing_v2: 'bool' =None, studio_version_visibility: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None, writable_s3_connections: 'bool' =None):  # noqa: E501
+    def __init__(self, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, b2c_experience: 'bool' =None, byo_machine_type: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, chat_models: 'bool' =None, cloudspace_schedules: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, control_center_monitoring: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_persistent_disk: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, f234: 'bool' =None, f236: 'bool' =None, f239: 'bool' =None, f240: 'bool' =None, f241: 'bool' =None, f243: 'bool' =None, f245: 'bool' =None, f247: 'bool' =None, f250: 'bool' =None, f252: 'bool' =None, f253: 'bool' =None, f254: 'bool' =None, f255: 'bool' =None, f258: 'bool' =None, f259: 'bool' =None, f260: 'bool' =None, f261: 'bool' =None, f262: 'bool' =None, f263: 'bool' =None, f265: 'bool' =None, f266: 'bool' =None, f267: 'bool' =None, f268: 'bool' =None, f269: 'bool' =None, f270: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, job_artifacts_v2: 'bool' =None, kubernetes_cluster_ui: 'bool' =None, kubernetes_clusters: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_api_dashboard: 'bool' =None, multiple_studio_versions: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, persistent_disk: 'bool' =None, plugin_distributed: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_sweeps: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, product_license: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, reserved_machines_tab: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, show_dev_admin: 'bool' =None, slurm: 'bool' =None, specialised_studios: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_sharing_v2: 'bool' =None, studio_version_visibility: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None, writable_s3_connections: 'bool' =None):  # noqa: E501
         """V1UserFeatures - a model defined in Swagger"""  # noqa: E501
         self._affiliate_links = None
         self._agents_v2 = None
@@ -248,7 +246,6 @@ class V1UserFeatures(object):
         self._deployment_persistent_disk = None
         self._drive_v2 = None
         self._enterprise_compute_admin = None
-        self._f227 = None
         self._f234 = None
         self._f236 = None
         self._f239 = None
@@ -257,25 +254,25 @@ class V1UserFeatures(object):
         self._f243 = None
         self._f245 = None
         self._f247 = None
-        self._f248 = None
         self._f250 = None
         self._f252 = None
         self._f253 = None
         self._f254 = None
         self._f255 = None
-        self._f257 = None
         self._f258 = None
         self._f259 = None
         self._f260 = None
         self._f261 = None
         self._f262 = None
         self._f263 = None
-        self._f264 = None
         self._f265 = None
         self._f266 = None
+        self._f267 = None
+        self._f268 = None
+        self._f269 = None
+        self._f270 = None
         self._fair_share = None
         self._featured_studios_admin = None
-        self._gcs_connections_optimized = None
         self._job_artifacts_v2 = None
         self._kubernetes_cluster_ui = None
         self._kubernetes_clusters = None
@@ -360,8 +357,6 @@ class V1UserFeatures(object):
             self.drive_v2 = drive_v2
         if enterprise_compute_admin is not None:
             self.enterprise_compute_admin = enterprise_compute_admin
-        if f227 is not None:
-            self.f227 = f227
         if f234 is not None:
             self.f234 = f234
         if f236 is not None:
@@ -378,8 +373,6 @@ class V1UserFeatures(object):
             self.f245 = f245
         if f247 is not None:
             self.f247 = f247
-        if f248 is not None:
-            self.f248 = f248
         if f250 is not None:
             self.f250 = f250
         if f252 is not None:
@@ -390,8 +383,6 @@ class V1UserFeatures(object):
             self.f254 = f254
         if f255 is not None:
             self.f255 = f255
-        if f257 is not None:
-            self.f257 = f257
         if f258 is not None:
             self.f258 = f258
         if f259 is not None:
@@ -404,18 +395,22 @@ class V1UserFeatures(object):
             self.f262 = f262
         if f263 is not None:
             self.f263 = f263
-        if f264 is not None:
-            self.f264 = f264
         if f265 is not None:
             self.f265 = f265
         if f266 is not None:
             self.f266 = f266
+        if f267 is not None:
+            self.f267 = f267
+        if f268 is not None:
+            self.f268 = f268
+        if f269 is not None:
+            self.f269 = f269
+        if f270 is not None:
+            self.f270 = f270
         if fair_share is not None:
             self.fair_share = fair_share
         if featured_studios_admin is not None:
             self.featured_studios_admin = featured_studios_admin
-        if gcs_connections_optimized is not None:
-            self.gcs_connections_optimized = gcs_connections_optimized
         if job_artifacts_v2 is not None:
             self.job_artifacts_v2 = job_artifacts_v2
         if kubernetes_cluster_ui is not None:
@@ -957,27 +952,6 @@ class V1UserFeatures(object):
 
         self._enterprise_compute_admin = enterprise_compute_admin
 
-    @property
-    def f227(self) -> 'bool':
-        """Gets the f227 of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The f227 of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._f227
-
-    @f227.setter
-    def f227(self, f227: 'bool'):
-        """Sets the f227 of this V1UserFeatures.
-
-
-        :param f227: The f227 of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._f227 = f227
-
     @property
     def f234(self) -> 'bool':
         """Gets the f234 of this V1UserFeatures.  # noqa: E501
@@ -1146,27 +1120,6 @@ class V1UserFeatures(object):
 
         self._f247 = f247
 
-    @property
-    def f248(self) -> 'bool':
-        """Gets the f248 of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The f248 of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._f248
-
-    @f248.setter
-    def f248(self, f248: 'bool'):
-        """Sets the f248 of this V1UserFeatures.
-
-
-        :param f248: The f248 of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._f248 = f248
-
     @property
     def f250(self) -> 'bool':
         """Gets the f250 of this V1UserFeatures.  # noqa: E501
@@ -1272,27 +1225,6 @@ class V1UserFeatures(object):
 
         self._f255 = f255
 
-    @property
-    def f257(self) -> 'bool':
-        """Gets the f257 of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The f257 of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._f257
-
-    @f257.setter
-    def f257(self, f257: 'bool'):
-        """Sets the f257 of this V1UserFeatures.
-
-
-        :param f257: The f257 of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._f257 = f257
-
     @property
     def f258(self) -> 'bool':
         """Gets the f258 of this V1UserFeatures.  # noqa: E501
@@ -1419,27 +1351,6 @@ class V1UserFeatures(object):
 
         self._f263 = f263
 
-    @property
-    def f264(self) -> 'bool':
-        """Gets the f264 of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The f264 of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._f264
-
-    @f264.setter
-    def f264(self, f264: 'bool'):
-        """Sets the f264 of this V1UserFeatures.
-
-
-        :param f264: The f264 of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._f264 = f264
-
     @property
     def f265(self) -> 'bool':
         """Gets the f265 of this V1UserFeatures.  # noqa: E501
@@ -1482,6 +1393,90 @@ class V1UserFeatures(object):
 
         self._f266 = f266
 
+    @property
+    def f267(self) -> 'bool':
+        """Gets the f267 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f267 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f267
+
+    @f267.setter
+    def f267(self, f267: 'bool'):
+        """Sets the f267 of this V1UserFeatures.
+
+
+        :param f267: The f267 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f267 = f267
+
+    @property
+    def f268(self) -> 'bool':
+        """Gets the f268 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f268 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f268
+
+    @f268.setter
+    def f268(self, f268: 'bool'):
+        """Sets the f268 of this V1UserFeatures.
+
+
+        :param f268: The f268 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f268 = f268
+
+    @property
+    def f269(self) -> 'bool':
+        """Gets the f269 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f269 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f269
+
+    @f269.setter
+    def f269(self, f269: 'bool'):
+        """Sets the f269 of this V1UserFeatures.
+
+
+        :param f269: The f269 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f269 = f269
+
+    @property
+    def f270(self) -> 'bool':
+        """Gets the f270 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f270 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f270
+
+    @f270.setter
+    def f270(self, f270: 'bool'):
+        """Sets the f270 of this V1UserFeatures.
+
+
+        :param f270: The f270 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f270 = f270
+
     @property
     def fair_share(self) -> 'bool':
         """Gets the fair_share of this V1UserFeatures.  # noqa: E501
@@ -1524,27 +1519,6 @@ class V1UserFeatures(object):
 
         self._featured_studios_admin = featured_studios_admin
 
-    @property
-    def gcs_connections_optimized(self) -> 'bool':
-        """Gets the gcs_connections_optimized of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The gcs_connections_optimized of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._gcs_connections_optimized
-
-    @gcs_connections_optimized.setter
-    def gcs_connections_optimized(self, gcs_connections_optimized: 'bool'):
-        """Sets the gcs_connections_optimized of this V1UserFeatures.
-
-
-        :param gcs_connections_optimized: The gcs_connections_optimized of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._gcs_connections_optimized = gcs_connections_optimized
-
     @property
     def job_artifacts_v2(self) -> 'bool':
         """Gets the job_artifacts_v2 of this V1UserFeatures.  # noqa: E501

lightning_sdk/machine.py

@@ -218,3 +218,6 @@ Machine.H200 = Machine(name="H200", slug="lit-h200x-1", family="H200", accelerat
 Machine.H200_X_8 = Machine(name="H200_X_8", slug="lit-h200x-8", family="H200", accelerator_count=8)
 # available B200 machines
 Machine.B200_X_8 = Machine(name="B200_X_8", slug="lit-b200x-8", family="B200", accelerator_count=8)
+
+
+DEFAULT_MACHINE = Machine.CPU.name

lightning_sdk/pipeline/steps.py

@@ -239,6 +239,7 @@ class JobStep:
             artifacts_remote=None,
             max_runtime=self.max_runtime,
             machine_image_version=machine_image_version,
+            reuse_snapshot=True,
         )
 
         return V1PipelineStep(

lightning_sdk/studio.py

@@ -11,7 +11,7 @@ from lightning_sdk.api.studio_api import StudioApi
 from lightning_sdk.base_studio import BaseStudio
 from lightning_sdk.constants import _LIGHTNING_DEBUG
 from lightning_sdk.lightning_cloud.openapi import V1ClusterType
-from lightning_sdk.machine import CloudProvider, Machine
+from lightning_sdk.machine import DEFAULT_MACHINE, CloudProvider, Machine
 from lightning_sdk.organization import Organization
 from lightning_sdk.owner import Owner
 from lightning_sdk.status import Status
@@ -103,7 +103,16 @@ class Studio:
         self._plugins = {}
         self._studio = None
 
-        cloud_account = _resolve_deprecated_cluster(cloud_account, cluster)
+        # Check to see if we're inside a studio
+        current_studio = None
+        studio_id = os.environ.get("LIGHTNING_CLOUD_SPACE_ID", None)
+        if studio_id is not None:
+            # We're inside a studio, get it by ID
+            current_studio = self._studio_api.get_studio_by_id(studio_id=studio_id, teamspace_id=self._teamspace.id)
+
+        cloud_account = _resolve_deprecated_cluster(
+            cloud_account, cluster, current_studio.cluster_id if current_studio else None
+        )
         cloud_provider = _resolve_deprecated_provider(cloud_provider, provider)
 
         cls_name = self._cls_name
@@ -135,14 +144,14 @@ class Studio:
                     f"Available studio types: "
                     f"{[bst.name.lower().replace(' ', '-') for bst in self._available_base_studios]}"
                 )
+        else:
+            if current_studio:
+                self._studio_type = current_studio.environment_template_id
 
         # Resolve studio name if not provided: explicit → env (LIGHTNING_CLOUD_SPACE_ID) → config defaults
         if name is None and not getattr(self._skip_init, "value", False):
-            studio_id = os.environ.get("LIGHTNING_CLOUD_SPACE_ID", None)
-            if studio_id is not None:
-                # We're inside a studio, get it by ID
-                self._studio = self._studio_api.get_studio_by_id(studio_id=studio_id, teamspace_id=self._teamspace.id)
-                name = self._studio.name
+            if current_studio:
+                name = current_studio.name
             else:
                 # Try config defaults
                 from lightning_sdk.utils.config import Config, DefaultConfigKeys
@@ -272,7 +281,7 @@ class Studio:
 
     def start(
         self,
-        machine: Union[Machine, str] = Machine.CPU,
+        machine: Optional[Union[Machine, str]] = None,
         interruptible: Optional[bool] = None,
         max_runtime: Optional[int] = None,
     ) -> None:
@@ -287,6 +296,22 @@ class Studio:
                 Defaults to 3h
 
         """
+        # Check to see if we're inside a studio and if its running
+        current_studio_machine = None
+        studio_id = os.environ.get("LIGHTNING_CLOUD_SPACE_ID", None)
+        if studio_id is not None:
+            # We're inside a studio, get the machine if it is running
+            current_studio = self._studio_api.get_studio_by_id(studio_id=studio_id, teamspace_id=self._teamspace.id)
+            current_status = self._studio_api._get_studio_instance_status_from_object(current_studio)
+
+            if current_status and _internal_status_to_external_status(current_status) == Status.Running:
+                current_studio_machine = self._studio_api.get_machine(
+                    current_studio.id,
+                    self._teamspace.id,
+                    current_studio.cluster_id,
+                    _get_org_id(self._teamspace),
+                )
+
         status = self.status
 
         if interruptible is None:
@@ -296,9 +321,14 @@ class Studio:
             else:
                 interruptible = self.teamspace.start_studios_on_interruptible
 
-        new_machine = machine
-        if not isinstance(machine, Machine):
-            new_machine = Machine.from_str(machine)
+        new_machine = DEFAULT_MACHINE
+        if machine is not None:
+            new_machine = machine
+        elif current_studio_machine is not None:
+            new_machine = current_studio_machine
+
+        if not isinstance(new_machine, Machine):
+            new_machine = Machine.from_str(new_machine)
 
         if status == Status.Running:
             if new_machine != self.machine: