lightning-sdk 2025.9.23__py3-none-any.whl → 2025.9.29__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_storage_transfer_status.py

@@ -0,0 +1,108 @@
+# 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 V1StorageTransferStatus(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+
+    """
+    allowed enum values
+    """
+    QUEUED = "STATUS_QUEUED"
+    IN_PROGRESS = "STATUS_IN_PROGRESS"
+    FAILED = "STATUS_FAILED"
+    ABORTED = "STATUS_ABORTED"
+    PAUSED = "STATUS_PAUSED"
+    COMPLETED = "STATUS_COMPLETED"
+    UNSPECIFIED = "STATUS_UNSPECIFIED"
+    """
+    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 = {
+    }
+
+    attribute_map = {
+    }
+
+    def __init__(self):  # noqa: E501
+        """V1StorageTransferStatus - a model defined in Swagger"""  # noqa: E501
+        self.discriminator = None
+
+    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(V1StorageTransferStatus, 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: 'V1StorageTransferStatus') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1StorageTransferStatus):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1StorageTransferStatus') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py

@@ -53,6 +53,7 @@ class V1UpdateUserRequest(object):
         'first_name': 'str',
         'general_audience_mode': 'bool',
         'last_name': 'str',
+        'linux_username': 'str',
         'non_developer_mode': 'bool',
         'opted_in_marketing_emails': 'bool',
         'organization': 'str',
@@ -82,6 +83,7 @@ class V1UpdateUserRequest(object):
         'first_name': 'firstName',
         'general_audience_mode': 'generalAudienceMode',
         'last_name': 'lastName',
+        'linux_username': 'linuxUsername',
         'non_developer_mode': 'nonDeveloperMode',
         'opted_in_marketing_emails': 'optedInMarketingEmails',
         'organization': 'organization',
@@ -98,7 +100,7 @@ class V1UpdateUserRequest(object):
         'website': 'website'
     }
 
-    def __init__(self, agree_to_terms_and_conditions: 'bool' =None, allow_credits_auto_replenish: 'bool' =None, auto_replenish_amount: 'float' =None, auto_replenish_threshold: 'float' =None, complete_sign_up: 'bool' =None, completed_project_onboarding: 'bool' =None, country: 'str' =None, email: 'str' =None, experimentation_id: 'str' =None, first_name: 'str' =None, general_audience_mode: 'bool' =None, last_name: 'str' =None, non_developer_mode: 'bool' =None, opted_in_marketing_emails: 'bool' =None, organization: 'str' =None, preferred_color_scheme: 'str' =None, preferred_ide: 'str' =None, preferred_shell: 'str' =None, preferred_vscode_marketplace: 'str' =None, role: 'str' =None, saw_create_first_project_dialog: 'bool' =None, saw_forums_login_merge_dialog: 'bool' =None, saw_free_credits_notification: 'bool' =None, user_metadata: 'str' =None, username: 'str' =None, website: 'str' =None):  # noqa: E501
+    def __init__(self, agree_to_terms_and_conditions: 'bool' =None, allow_credits_auto_replenish: 'bool' =None, auto_replenish_amount: 'float' =None, auto_replenish_threshold: 'float' =None, complete_sign_up: 'bool' =None, completed_project_onboarding: 'bool' =None, country: 'str' =None, email: 'str' =None, experimentation_id: 'str' =None, first_name: 'str' =None, general_audience_mode: 'bool' =None, last_name: 'str' =None, linux_username: 'str' =None, non_developer_mode: 'bool' =None, opted_in_marketing_emails: 'bool' =None, organization: 'str' =None, preferred_color_scheme: 'str' =None, preferred_ide: 'str' =None, preferred_shell: 'str' =None, preferred_vscode_marketplace: 'str' =None, role: 'str' =None, saw_create_first_project_dialog: 'bool' =None, saw_forums_login_merge_dialog: 'bool' =None, saw_free_credits_notification: 'bool' =None, user_metadata: 'str' =None, username: 'str' =None, website: 'str' =None):  # noqa: E501
         """V1UpdateUserRequest - a model defined in Swagger"""  # noqa: E501
         self._agree_to_terms_and_conditions = None
         self._allow_credits_auto_replenish = None
@@ -112,6 +114,7 @@ class V1UpdateUserRequest(object):
         self._first_name = None
         self._general_audience_mode = None
         self._last_name = None
+        self._linux_username = None
         self._non_developer_mode = None
         self._opted_in_marketing_emails = None
         self._organization = None
@@ -151,6 +154,8 @@ class V1UpdateUserRequest(object):
             self.general_audience_mode = general_audience_mode
         if last_name is not None:
             self.last_name = last_name
+        if linux_username is not None:
+            self.linux_username = linux_username
         if non_developer_mode is not None:
             self.non_developer_mode = non_developer_mode
         if opted_in_marketing_emails is not None:
@@ -432,6 +437,27 @@ class V1UpdateUserRequest(object):
 
         self._last_name = last_name
 
+    @property
+    def linux_username(self) -> 'str':
+        """Gets the linux_username of this V1UpdateUserRequest.  # noqa: E501
+
+
+        :return: The linux_username of this V1UpdateUserRequest.  # noqa: E501
+        :rtype: str
+        """
+        return self._linux_username
+
+    @linux_username.setter
+    def linux_username(self, linux_username: 'str'):
+        """Sets the linux_username of this V1UpdateUserRequest.
+
+
+        :param linux_username: The linux_username of this V1UpdateUserRequest.  # noqa: E501
+        :type: str
+        """
+
+        self._linux_username = linux_username
+
     @property
     def non_developer_mode(self) -> 'bool':
         """Gets the non_developer_mode of this V1UpdateUserRequest.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -67,7 +67,6 @@ class V1UserFeatures(object):
         'f227': 'bool',
         'f234': 'bool',
         'f236': 'bool',
-        'f237': 'bool',
         'f239': 'bool',
         'f240': 'bool',
         'f241': 'bool',
@@ -75,14 +74,16 @@ class V1UserFeatures(object):
         'f245': 'bool',
         'f247': 'bool',
         'f248': 'bool',
-        'f249': 'bool',
         'f250': 'bool',
         'f251': 'bool',
         'f252': 'bool',
         'f253': 'bool',
+        'f254': 'bool',
+        'f255': 'bool',
+        'f257': 'bool',
+        'f258': 'bool',
         'fair_share': 'bool',
         'featured_studios_admin': 'bool',
-        'gcp_overprovisioning': 'bool',
         'gcs_connections_optimized': 'bool',
         'instant_capacity_reservation': 'bool',
         'job_artifacts_v2': 'bool',
@@ -155,7 +156,6 @@ class V1UserFeatures(object):
         'f227': 'f227',
         'f234': 'f234',
         'f236': 'f236',
-        'f237': 'f237',
         'f239': 'f239',
         'f240': 'f240',
         'f241': 'f241',
@@ -163,14 +163,16 @@ class V1UserFeatures(object):
         'f245': 'f245',
         'f247': 'f247',
         'f248': 'f248',
-        'f249': 'f249',
         'f250': 'f250',
         'f251': 'f251',
         'f252': 'f252',
         'f253': 'f253',
+        'f254': 'f254',
+        'f255': 'f255',
+        'f257': 'f257',
+        'f258': 'f258',
         'fair_share': 'fairShare',
         'featured_studios_admin': 'featuredStudiosAdmin',
-        'gcp_overprovisioning': 'gcpOverprovisioning',
         'gcs_connections_optimized': 'gcsConnectionsOptimized',
         'instant_capacity_reservation': 'instantCapacityReservation',
         'job_artifacts_v2': 'jobArtifactsV2',
@@ -216,7 +218,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, auto_join_orgs: '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, f237: 'bool' =None, f239: 'bool' =None, f240: 'bool' =None, f241: 'bool' =None, f243: 'bool' =None, f245: 'bool' =None, f247: 'bool' =None, f248: 'bool' =None, f249: 'bool' =None, f250: 'bool' =None, f251: 'bool' =None, f252: 'bool' =None, f253: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcp_overprovisioning: 'bool' =None, gcs_connections_optimized: 'bool' =None, instant_capacity_reservation: '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, single_wallet: '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, trainium2: '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, auto_join_orgs: '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, f251: 'bool' =None, f252: 'bool' =None, f253: 'bool' =None, f254: 'bool' =None, f255: 'bool' =None, f257: 'bool' =None, f258: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcs_connections_optimized: 'bool' =None, instant_capacity_reservation: '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, single_wallet: '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, trainium2: '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
@@ -244,7 +246,6 @@ class V1UserFeatures(object):
         self._f227 = None
         self._f234 = None
         self._f236 = None
-        self._f237 = None
         self._f239 = None
         self._f240 = None
         self._f241 = None
@@ -252,14 +253,16 @@ class V1UserFeatures(object):
         self._f245 = None
         self._f247 = None
         self._f248 = None
-        self._f249 = None
         self._f250 = None
         self._f251 = None
         self._f252 = None
         self._f253 = None
+        self._f254 = None
+        self._f255 = None
+        self._f257 = None
+        self._f258 = None
         self._fair_share = None
         self._featured_studios_admin = None
-        self._gcp_overprovisioning = None
         self._gcs_connections_optimized = None
         self._instant_capacity_reservation = None
         self._job_artifacts_v2 = None
@@ -356,8 +359,6 @@ class V1UserFeatures(object):
             self.f234 = f234
         if f236 is not None:
             self.f236 = f236
-        if f237 is not None:
-            self.f237 = f237
         if f239 is not None:
             self.f239 = f239
         if f240 is not None:
@@ -372,8 +373,6 @@ class V1UserFeatures(object):
             self.f247 = f247
         if f248 is not None:
             self.f248 = f248
-        if f249 is not None:
-            self.f249 = f249
         if f250 is not None:
             self.f250 = f250
         if f251 is not None:
@@ -382,12 +381,18 @@ class V1UserFeatures(object):
             self.f252 = f252
         if f253 is not None:
             self.f253 = f253
+        if f254 is not None:
+            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 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 gcp_overprovisioning is not None:
-            self.gcp_overprovisioning = gcp_overprovisioning
         if gcs_connections_optimized is not None:
             self.gcs_connections_optimized = gcs_connections_optimized
         if instant_capacity_reservation is not None:
@@ -1021,27 +1026,6 @@ class V1UserFeatures(object):
 
         self._f236 = f236
 
-    @property
-    def f237(self) -> 'bool':
-        """Gets the f237 of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The f237 of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._f237
-
-    @f237.setter
-    def f237(self, f237: 'bool'):
-        """Sets the f237 of this V1UserFeatures.
-
-
-        :param f237: The f237 of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._f237 = f237
-
     @property
     def f239(self) -> 'bool':
         """Gets the f239 of this V1UserFeatures.  # noqa: E501
@@ -1189,27 +1173,6 @@ class V1UserFeatures(object):
 
         self._f248 = f248
 
-    @property
-    def f249(self) -> 'bool':
-        """Gets the f249 of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The f249 of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._f249
-
-    @f249.setter
-    def f249(self, f249: 'bool'):
-        """Sets the f249 of this V1UserFeatures.
-
-
-        :param f249: The f249 of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._f249 = f249
-
     @property
     def f250(self) -> 'bool':
         """Gets the f250 of this V1UserFeatures.  # noqa: E501
@@ -1294,6 +1257,90 @@ class V1UserFeatures(object):
 
         self._f253 = f253
 
+    @property
+    def f254(self) -> 'bool':
+        """Gets the f254 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f254 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f254
+
+    @f254.setter
+    def f254(self, f254: 'bool'):
+        """Sets the f254 of this V1UserFeatures.
+
+
+        :param f254: The f254 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f254 = f254
+
+    @property
+    def f255(self) -> 'bool':
+        """Gets the f255 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f255 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f255
+
+    @f255.setter
+    def f255(self, f255: 'bool'):
+        """Sets the f255 of this V1UserFeatures.
+
+
+        :param f255: The f255 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        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
+
+
+        :return: The f258 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f258
+
+    @f258.setter
+    def f258(self, f258: 'bool'):
+        """Sets the f258 of this V1UserFeatures.
+
+
+        :param f258: The f258 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f258 = f258
+
     @property
     def fair_share(self) -> 'bool':
         """Gets the fair_share of this V1UserFeatures.  # noqa: E501
@@ -1336,27 +1383,6 @@ class V1UserFeatures(object):
 
         self._featured_studios_admin = featured_studios_admin
 
-    @property
-    def gcp_overprovisioning(self) -> 'bool':
-        """Gets the gcp_overprovisioning of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The gcp_overprovisioning of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._gcp_overprovisioning
-
-    @gcp_overprovisioning.setter
-    def gcp_overprovisioning(self, gcp_overprovisioning: 'bool'):
-        """Sets the gcp_overprovisioning of this V1UserFeatures.
-
-
-        :param gcp_overprovisioning: The gcp_overprovisioning of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._gcp_overprovisioning = gcp_overprovisioning
-
     @property
     def gcs_connections_optimized(self) -> 'bool':
         """Gets the gcs_connections_optimized of this V1UserFeatures.  # noqa: E501

lightning_sdk/studio.py

@@ -102,6 +102,8 @@ class Studio:
         cloud_account = _resolve_deprecated_cluster(cloud_account, cluster)
         cloud_provider = _resolve_deprecated_provider(cloud_provider, provider)
 
+        cls_name = self._cls_name
+
         # if we're skipping init, we don't need to resolve the cloud account as then we're not creating a studio
         if self._teamspace is not None:
             _cloud_account = self._cloud_account_api.resolve_cloud_account(
@@ -126,7 +128,7 @@ class Studio:
                 name = config.get_value(DefaultConfigKeys.studio)
                 if name is None and not create_ok:
                     raise ValueError(
-                        "Cannot autodetect Studio. Either use the SDK from within a Studio or pass a name!"
+                        f"Cannot autodetect {cls_name}. Either use the SDK from within a {cls_name} or pass a name!"
                     )
 
         if self._studio is None and not getattr(self._skip_init, "value", False):
@@ -136,7 +138,7 @@ class Studio:
                     # if we don't have a name, raise an error to get
                     # to the exception path and optionally create a studio
                     raise ValueError(
-                        "Cannot autodetect Studio. Either use the SDK from within a Studio or pass a name!"
+                        f"Cannot autodetect {cls_name}. Either use the SDK from within a {cls_name} or pass a name!"
                     )
                 self._studio = self._studio_api.get_studio(name, self._teamspace.id)
             except ValueError as e:
@@ -227,7 +229,9 @@ class Studio:
     @property
     def cluster(self) -> str:
         """Returns the cluster the Studio is running on."""
-        warnings.warn("Studio.cluster is deprecated. Use Studio.cloud_account instead", DeprecationWarning)
+        warnings.warn(
+            f"{self._cls_name}.cluster is deprecated. Use {self._cls_name}.cloud_account instead", DeprecationWarning
+        )
         return self.cloud_account
 
     @property
@@ -266,14 +270,17 @@ class Studio:
                 new_machine = Machine.from_str(machine)
             if new_machine != self.machine:
                 raise RuntimeError(
-                    f"Requested to start studio on {new_machine}, but studio is already running on {self.machine}."
+                    f"Requested to start {self._cls_name} on {new_machine}, "
+                    "but {self._cls_name} is already running on {self.machine}."
                     " Consider switching instead!"
                 )
-            _logger.info(f"Studio {self.name} is already running")
+            _logger.info(f"{self._cls_name} {self.name} is already running")
             return
 
         if status != Status.Stopped:
-            raise RuntimeError(f"Cannot start a studio that is not stopped. Studio {self.name} is {status}.")
+            raise RuntimeError(
+                f"Cannot start a {self._cls_name} that is not stopped. {self._cls_name} {self.name} is {status}."
+            )
 
         # Show progress bar during startup
         if self.show_progress:
@@ -362,7 +369,8 @@ class Studio:
         status = self.status
         if status != Status.Running:
             raise RuntimeError(
-                f"Cannot switch machine on a studio that is not running. Studio {self.name} is {status}."
+                f"Cannot switch machine on a {self._cls_name} that is not running. "
+                "{self._cls_name} {self.name} is {status}."
             )
 
         current_cloud = self._cloud_account_api.get_cloud_account_non_org(
@@ -420,7 +428,10 @@ class Studio:
             print(f"Running {commands=}")
         status = self.status
         if status != Status.Running:
-            raise RuntimeError(f"Cannot run a command in a studio that is not running. Studio {self.name} is {status}.")
+            raise RuntimeError(
+                f"Cannot run a command in a {self._cls_name} that is not running. "
+                "{self._cls_name} {self.name} is {status}."
+            )
 
         iter_output = self._studio_api.run_studio_commands_and_yield(
             self._studio.id, self._teamspace.id, *commands, timeout=timeout, check_interval=check_interval
@@ -446,7 +457,10 @@ class Studio:
 
         status = self.status
         if status != Status.Running:
-            raise RuntimeError(f"Cannot run a command in a studio that is not running. Studio {self.name} is {status}.")
+            raise RuntimeError(
+                f"Cannot run a command in a {self._cls_name} that is not running. "
+                "{self._cls_name} {self.name} is {status}."
+            )
         output, exit_code = self._studio_api.run_studio_commands(self._studio.id, self._teamspace.id, *commands)
         output = output.strip()
 
@@ -616,7 +630,7 @@ class Studio:
     @auto_sleep.setter
     def auto_sleep(self, value: bool) -> None:
         if not value and self.machine == Machine.CPU:
-            warnings.warn("Disabling auto-sleep will convert the Studio from free to paid!")
+            warnings.warn(f"Disabling auto-sleep will convert the {self._cls_name} from free to paid!")
         self._studio_api.update_autoshutdown(self._studio.id, self._teamspace.id, enabled=value)
         self._update_studio_reference()
 
@@ -627,7 +641,7 @@ class Studio:
 
     @auto_sleep_time.setter
     def auto_sleep_time(self, value: int) -> None:
-        warnings.warn("Setting auto-sleep time will convert the Studio from free to paid!")
+        warnings.warn(f"Setting auto-sleep time will convert the {self._cls_name} from free to paid!")
         self._studio_api.update_autoshutdown(self._studio.id, self._teamspace.id, idle_shutdown_seconds=value)
         self._update_studio_reference()
 
@@ -758,6 +772,36 @@ class Studio:
     def _update_studio_reference(self) -> None:
         self._studio = self._studio_api.get_studio_by_id(studio_id=self._studio.id, teamspace_id=self._teamspace.id)
 
+    @property
+    def _cls_name(self) -> str:
+        return self.__class__.__qualname__
+
+
+class VM(Studio):
+    """A single Lightning AI VM.
+
+    Allows to fully control a vm, including retrieving the status, running commands
+    and switching machine types.
+
+    Args:
+        name: the name of the vm
+        teamspace: the name of the teamspace the vm is contained by
+        org: the name of the organization owning the :param`teamspace` in case it is owned by an org
+        user: the name of the user owning the :param`teamspace` in case it is owned directly by a user instead of an org
+        cloud_account: the name of the cloud account, the vm should be created on.
+            Doesn't matter when the vm already exists.
+        cloud_account_provider: The provider to select the cloud-account from.
+            If set, must be in agreement with the provider from the cloud_account (if specified).
+            If not specified, falls backto the teamspace default cloud account.
+        create_ok: whether the vm will be created if it does not yet exist. Defaults to True
+        provider: the provider of the machine, the vm should be created on.
+
+    Note:
+        Since a teamspace can either be owned by an org or by a user directly,
+        only one of the arguments can be provided.
+
+    """
+
 
 def _internal_status_to_external_status(internal_status: str) -> Status:
     """Converts internal status strings from HTTP requests to external enums."""

lightning_sdk/teamspace.py

@@ -28,7 +28,7 @@ from lightning_sdk.utils.resolve import (
 if TYPE_CHECKING:
     from lightning_sdk.job import Job
     from lightning_sdk.mmt import MMT
-    from lightning_sdk.studio import Studio
+    from lightning_sdk.studio import VM, Studio
 
 
 class FolderLocation(Enum):
@@ -144,13 +144,22 @@ class Teamspace:
         """All studios within that teamspace."""
         from lightning_sdk.studio import Studio
 
+        return self._get_studios(Studio)
+
+    @property
+    def vms(self) -> List["VM"]:
+        from lightning_sdk.studio import VM
+
+        return [x for x in self._get_studios(VM) if isinstance(x, VM)]
+
+    def _get_studios(self, target_cls: type) -> List[Union["Studio", "VM"]]:
         studios = []
         cloud_accounts = self._teamspace_api.list_cloud_accounts(teamspace_id=self.id)
         for cl in cloud_accounts:
             _studios = self._teamspace_api.list_studios(teamspace_id=self.id, cloud_account=cl.cluster_id)
             for s in _studios:
                 with skip_studio_init():
-                    studio = Studio(name=s.name, teamspace=self, cluster=cl.cluster_name, create_ok=False)
+                    studio = target_cls(name=s.name, teamspace=self, cluster=cl.cluster_name, create_ok=False)
                     studio._studio = s
                     studio._teamspace = self
                     studios.append(studio)

{lightning_sdk-2025.9.23.dist-info → lightning_sdk-2025.9.29.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning_sdk
-Version: 2025.9.23
+Version: 2025.9.29
 Summary: SDK to develop using Lightning AI Studios
 Author-email: Lightning-AI <justus@lightning.ai>
 License: MIT License