lightning-sdk 0.1.32__py3-none-any.whl → 0.1.34__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py

@@ -48,6 +48,7 @@ class V1Deployment(object):
         'id': 'str',
         'is_published': 'bool',
         'name': 'str',
+        'parameter_spec': 'V1ParameterizationSpec',
         'project_id': 'str',
         'release_id': 'str',
         'replicas': 'int',
@@ -66,6 +67,7 @@ class V1Deployment(object):
         'id': 'id',
         'is_published': 'isPublished',
         'name': 'name',
+        'parameter_spec': 'parameterSpec',
         'project_id': 'projectId',
         'release_id': 'releaseId',
         'replicas': 'replicas',
@@ -76,7 +78,7 @@ class V1Deployment(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, autoscaling: 'V1AutoscalingSpec' =None, created_at: 'datetime' =None, desired_state: 'V1DeploymentState' =None, endpoint: 'V1Endpoint' =None, id: 'str' =None, is_published: 'bool' =None, name: 'str' =None, project_id: 'str' =None, release_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, status: 'V1DeploymentStatus' =None, strategy: 'V1DeploymentStrategy' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, autoscaling: 'V1AutoscalingSpec' =None, created_at: 'datetime' =None, desired_state: 'V1DeploymentState' =None, endpoint: 'V1Endpoint' =None, id: 'str' =None, is_published: 'bool' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, project_id: 'str' =None, release_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, status: 'V1DeploymentStatus' =None, strategy: 'V1DeploymentStrategy' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """V1Deployment - a model defined in Swagger"""  # noqa: E501
         self._autoscaling = None
         self._created_at = None
@@ -85,6 +87,7 @@ class V1Deployment(object):
         self._id = None
         self._is_published = None
         self._name = None
+        self._parameter_spec = None
         self._project_id = None
         self._release_id = None
         self._replicas = None
@@ -108,6 +111,8 @@ class V1Deployment(object):
             self.is_published = is_published
         if name is not None:
             self.name = name
+        if parameter_spec is not None:
+            self.parameter_spec = parameter_spec
         if project_id is not None:
             self.project_id = project_id
         if release_id is not None:
@@ -272,6 +277,27 @@ class V1Deployment(object):
 
         self._name = name
 
+    @property
+    def parameter_spec(self) -> 'V1ParameterizationSpec':
+        """Gets the parameter_spec of this V1Deployment.  # noqa: E501
+
+
+        :return: The parameter_spec of this V1Deployment.  # noqa: E501
+        :rtype: V1ParameterizationSpec
+        """
+        return self._parameter_spec
+
+    @parameter_spec.setter
+    def parameter_spec(self, parameter_spec: 'V1ParameterizationSpec'):
+        """Sets the parameter_spec of this V1Deployment.
+
+
+        :param parameter_spec: The parameter_spec of this V1Deployment.  # noqa: E501
+        :type: V1ParameterizationSpec
+        """
+
+        self._parameter_spec = parameter_spec
+
     @property
     def project_id(self) -> 'str':
         """Gets the project_id of this V1Deployment.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_parameter.py

@@ -41,6 +41,7 @@ class V1DeploymentTemplateParameter(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'checkbox': 'V1Checkbox',
         'data_path': 'V1DataPath',
         'display_name': 'str',
         'input': 'V1Input',
@@ -55,6 +56,7 @@ class V1DeploymentTemplateParameter(object):
     }
 
     attribute_map = {
+        'checkbox': 'checkbox',
         'data_path': 'dataPath',
         'display_name': 'displayName',
         'input': 'input',
@@ -68,8 +70,9 @@ class V1DeploymentTemplateParameter(object):
         'type': 'type'
     }
 
-    def __init__(self, data_path: 'V1DataPath' =None, display_name: 'str' =None, input: 'V1Input' =None, long_description: 'str' =None, name: 'str' =None, placements: 'list[V1DeploymentTemplateParameterPlacement]' =None, readonly: 'bool' =None, required: 'bool' =None, select: 'V1Select' =None, short_description: 'str' =None, type: 'V1DeploymentTemplateParameterType' =None):  # noqa: E501
+    def __init__(self, checkbox: 'V1Checkbox' =None, data_path: 'V1DataPath' =None, display_name: 'str' =None, input: 'V1Input' =None, long_description: 'str' =None, name: 'str' =None, placements: 'list[V1DeploymentTemplateParameterPlacement]' =None, readonly: 'bool' =None, required: 'bool' =None, select: 'V1Select' =None, short_description: 'str' =None, type: 'V1DeploymentTemplateParameterType' =None):  # noqa: E501
         """V1DeploymentTemplateParameter - a model defined in Swagger"""  # noqa: E501
+        self._checkbox = None
         self._data_path = None
         self._display_name = None
         self._input = None
@@ -82,6 +85,8 @@ class V1DeploymentTemplateParameter(object):
         self._short_description = None
         self._type = None
         self.discriminator = None
+        if checkbox is not None:
+            self.checkbox = checkbox
         if data_path is not None:
             self.data_path = data_path
         if display_name is not None:
@@ -105,6 +110,27 @@ class V1DeploymentTemplateParameter(object):
         if type is not None:
             self.type = type
 
+    @property
+    def checkbox(self) -> 'V1Checkbox':
+        """Gets the checkbox of this V1DeploymentTemplateParameter.  # noqa: E501
+
+
+        :return: The checkbox of this V1DeploymentTemplateParameter.  # noqa: E501
+        :rtype: V1Checkbox
+        """
+        return self._checkbox
+
+    @checkbox.setter
+    def checkbox(self, checkbox: 'V1Checkbox'):
+        """Sets the checkbox of this V1DeploymentTemplateParameter.
+
+
+        :param checkbox: The checkbox of this V1DeploymentTemplateParameter.  # noqa: E501
+        :type: V1Checkbox
+        """
+
+        self._checkbox = checkbox
+
     @property
     def data_path(self) -> 'V1DataPath':
         """Gets the data_path of this V1DeploymentTemplateParameter.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_parameter_type.py

@@ -40,6 +40,7 @@ class V1DeploymentTemplateParameterType(object):
     SELECT = "DEPLOYMENT_TEMPLATE_SELECT"
     INPUT = "DEPLOYMENT_TEMPLATE_INPUT"
     DATA_PATH = "DEPLOYMENT_TEMPLATE_DATA_PATH"
+    CHECKBOX = "DEPLOYMENT_TEMPLATE_CHECKBOX"
     """
     Attributes:
       swagger_types (dict): The key is attribute name

lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py

@@ -44,7 +44,6 @@ class V1JobSpec(object):
         'cloudspace_id': 'str',
         'cluster_id': 'str',
         'command': 'str',
-        'created_symlinks': 'bool',
         'data_paths': 'list[V1DataPath]',
         'entrypoint': 'str',
         'env': 'list[V1EnvVar]',
@@ -69,7 +68,6 @@ class V1JobSpec(object):
         'cloudspace_id': 'cloudspaceId',
         'cluster_id': 'clusterId',
         'command': 'command',
-        'created_symlinks': 'createdSymlinks',
         'data_paths': 'dataPaths',
         'entrypoint': 'entrypoint',
         'env': 'env',
@@ -90,12 +88,11 @@ class V1JobSpec(object):
         'volumes': 'volumes'
     }
 
-    def __init__(self, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, created_symlinks: 'bool' =None, data_paths: 'list[V1DataPath]' =None, entrypoint: 'str' =None, env: 'list[V1EnvVar]' =None, image: 'str' =None, image_cluster_credentials: 'bool' =None, image_secret_ref: 'str' =None, include_credentials: 'bool' =None, instance_name: 'str' =None, instance_type: 'str' =None, modified_volume: 'bool' =None, readiness_probe: 'V1JobHealthCheckConfig' =None, resources: 'V1Resources' =None, restart_policy: 'str' =None, run_id: 'str' =None, skip_data_connections_setup: 'bool' =None, skip_filesystem_setup: 'bool' =None, spot: 'bool' =None, volumes: 'list[V1Volume]' =None):  # noqa: E501
+    def __init__(self, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, data_paths: 'list[V1DataPath]' =None, entrypoint: 'str' =None, env: 'list[V1EnvVar]' =None, image: 'str' =None, image_cluster_credentials: 'bool' =None, image_secret_ref: 'str' =None, include_credentials: 'bool' =None, instance_name: 'str' =None, instance_type: 'str' =None, modified_volume: 'bool' =None, readiness_probe: 'V1JobHealthCheckConfig' =None, resources: 'V1Resources' =None, restart_policy: 'str' =None, run_id: 'str' =None, skip_data_connections_setup: 'bool' =None, skip_filesystem_setup: 'bool' =None, spot: 'bool' =None, volumes: 'list[V1Volume]' =None):  # noqa: E501
         """V1JobSpec - a model defined in Swagger"""  # noqa: E501
         self._cloudspace_id = None
         self._cluster_id = None
         self._command = None
-        self._created_symlinks = None
         self._data_paths = None
         self._entrypoint = None
         self._env = None
@@ -121,8 +118,6 @@ class V1JobSpec(object):
             self.cluster_id = cluster_id
         if command is not None:
             self.command = command
-        if created_symlinks is not None:
-            self.created_symlinks = created_symlinks
         if data_paths is not None:
             self.data_paths = data_paths
         if entrypoint is not None:
@@ -223,27 +218,6 @@ class V1JobSpec(object):
 
         self._command = command
 
-    @property
-    def created_symlinks(self) -> 'bool':
-        """Gets the created_symlinks of this V1JobSpec.  # noqa: E501
-
-
-        :return: The created_symlinks of this V1JobSpec.  # noqa: E501
-        :rtype: bool
-        """
-        return self._created_symlinks
-
-    @created_symlinks.setter
-    def created_symlinks(self, created_symlinks: 'bool'):
-        """Sets the created_symlinks of this V1JobSpec.
-
-
-        :param created_symlinks: The created_symlinks of this V1JobSpec.  # noqa: E501
-        :type: bool
-        """
-
-        self._created_symlinks = created_symlinks
-
     @property
     def data_paths(self) -> 'list[V1DataPath]':
         """Gets the data_paths of this V1JobSpec.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_lightningwork_status.py

@@ -52,7 +52,6 @@ class V1LightningworkStatus(object):
         'server_instance_id': 'str',
         'start_timestamp': 'datetime',
         'stop_timestamp': 'datetime',
-        'symlinks_created': 'bool',
         'total_cost': 'float',
         'total_size_bytes': 'str'
     }
@@ -69,12 +68,11 @@ class V1LightningworkStatus(object):
         'server_instance_id': 'serverInstanceId',
         'start_timestamp': 'startTimestamp',
         'stop_timestamp': 'stopTimestamp',
-        'symlinks_created': 'symlinksCreated',
         'total_cost': 'totalCost',
         'total_size_bytes': 'totalSizeBytes'
     }
 
-    def __init__(self, ip_address: 'str' =None, last_state_status_transition_timestamp: 'datetime' =None, message: 'str' =None, modified_volume: 'bool' =None, number_of_files: 'str' =None, phase: 'V1LightningworkState' =None, reason: 'V1LightningworkStatusReason' =None, server_id: 'str' =None, server_instance_id: 'str' =None, start_timestamp: 'datetime' =None, stop_timestamp: 'datetime' =None, symlinks_created: 'bool' =None, total_cost: 'float' =None, total_size_bytes: 'str' =None):  # noqa: E501
+    def __init__(self, ip_address: 'str' =None, last_state_status_transition_timestamp: 'datetime' =None, message: 'str' =None, modified_volume: 'bool' =None, number_of_files: 'str' =None, phase: 'V1LightningworkState' =None, reason: 'V1LightningworkStatusReason' =None, server_id: 'str' =None, server_instance_id: 'str' =None, start_timestamp: 'datetime' =None, stop_timestamp: 'datetime' =None, total_cost: 'float' =None, total_size_bytes: 'str' =None):  # noqa: E501
         """V1LightningworkStatus - a model defined in Swagger"""  # noqa: E501
         self._ip_address = None
         self._last_state_status_transition_timestamp = None
@@ -87,7 +85,6 @@ class V1LightningworkStatus(object):
         self._server_instance_id = None
         self._start_timestamp = None
         self._stop_timestamp = None
-        self._symlinks_created = None
         self._total_cost = None
         self._total_size_bytes = None
         self.discriminator = None
@@ -113,8 +110,6 @@ class V1LightningworkStatus(object):
             self.start_timestamp = start_timestamp
         if stop_timestamp is not None:
             self.stop_timestamp = stop_timestamp
-        if symlinks_created is not None:
-            self.symlinks_created = symlinks_created
         if total_cost is not None:
             self.total_cost = total_cost
         if total_size_bytes is not None:
@@ -351,27 +346,6 @@ class V1LightningworkStatus(object):
 
         self._stop_timestamp = stop_timestamp
 
-    @property
-    def symlinks_created(self) -> 'bool':
-        """Gets the symlinks_created of this V1LightningworkStatus.  # noqa: E501
-
-
-        :return: The symlinks_created of this V1LightningworkStatus.  # noqa: E501
-        :rtype: bool
-        """
-        return self._symlinks_created
-
-    @symlinks_created.setter
-    def symlinks_created(self, symlinks_created: 'bool'):
-        """Sets the symlinks_created of this V1LightningworkStatus.
-
-
-        :param symlinks_created: The symlinks_created of this V1LightningworkStatus.  # noqa: E501
-        :type: bool
-        """
-
-        self._symlinks_created = symlinks_created
-
     @property
     def total_cost(self) -> 'float':
         """Gets the total_cost of this V1LightningworkStatus.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_metric_value.py

@@ -42,24 +42,29 @@ class V1MetricValue(object):
     """
     swagger_types = {
         'created_at': 'datetime',
+        'internal_step': 'str',
         'step': 'str',
         'value': 'float'
     }
 
     attribute_map = {
         'created_at': 'createdAt',
+        'internal_step': 'internalStep',
         'step': 'step',
         'value': 'value'
     }
 
-    def __init__(self, created_at: 'datetime' =None, step: 'str' =None, value: 'float' =None):  # noqa: E501
+    def __init__(self, created_at: 'datetime' =None, internal_step: 'str' =None, step: 'str' =None, value: 'float' =None):  # noqa: E501
         """V1MetricValue - a model defined in Swagger"""  # noqa: E501
         self._created_at = None
+        self._internal_step = None
         self._step = None
         self._value = None
         self.discriminator = None
         if created_at is not None:
             self.created_at = created_at
+        if internal_step is not None:
+            self.internal_step = internal_step
         if step is not None:
             self.step = step
         if value is not None:
@@ -86,6 +91,27 @@ class V1MetricValue(object):
 
         self._created_at = created_at
 
+    @property
+    def internal_step(self) -> 'str':
+        """Gets the internal_step of this V1MetricValue.  # noqa: E501
+
+
+        :return: The internal_step of this V1MetricValue.  # noqa: E501
+        :rtype: str
+        """
+        return self._internal_step
+
+    @internal_step.setter
+    def internal_step(self, internal_step: 'str'):
+        """Sets the internal_step of this V1MetricValue.
+
+
+        :param internal_step: The internal_step of this V1MetricValue.  # noqa: E501
+        :type: str
+        """
+
+        self._internal_step = internal_step
+
     @property
     def step(self) -> 'str':
         """Gets the step of this V1MetricValue.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_usage_details.py

@@ -49,11 +49,13 @@ class V1UsageDetails(object):
         'hourly_price': 'float',
         'instance_count': 'str',
         'instance_type': 'str',
+        'read_bytes': 'str',
         'resource_name': 'str',
         'saved_cost': 'float',
         'spot': 'bool',
         'started_at': 'datetime',
-        'user_id': 'str'
+        'user_id': 'str',
+        'written_bytes': 'str'
     }
 
     attribute_map = {
@@ -65,14 +67,16 @@ class V1UsageDetails(object):
         'hourly_price': 'hourlyPrice',
         'instance_count': 'instanceCount',
         'instance_type': 'instanceType',
+        'read_bytes': 'readBytes',
         'resource_name': 'resourceName',
         'saved_cost': 'savedCost',
         'spot': 'spot',
         'started_at': 'startedAt',
-        'user_id': 'userId'
+        'user_id': 'userId',
+        'written_bytes': 'writtenBytes'
     }
 
-    def __init__(self, billed_bytes: 'str' =None, billed_time_seconds: 'str' =None, cost: 'float' =None, ended_at: 'datetime' =None, free: 'bool' =None, hourly_price: 'float' =None, instance_count: 'str' =None, instance_type: 'str' =None, resource_name: 'str' =None, saved_cost: 'float' =None, spot: 'bool' =None, started_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, billed_bytes: 'str' =None, billed_time_seconds: 'str' =None, cost: 'float' =None, ended_at: 'datetime' =None, free: 'bool' =None, hourly_price: 'float' =None, instance_count: 'str' =None, instance_type: 'str' =None, read_bytes: 'str' =None, resource_name: 'str' =None, saved_cost: 'float' =None, spot: 'bool' =None, started_at: 'datetime' =None, user_id: 'str' =None, written_bytes: 'str' =None):  # noqa: E501
         """V1UsageDetails - a model defined in Swagger"""  # noqa: E501
         self._billed_bytes = None
         self._billed_time_seconds = None
@@ -82,11 +86,13 @@ class V1UsageDetails(object):
         self._hourly_price = None
         self._instance_count = None
         self._instance_type = None
+        self._read_bytes = None
         self._resource_name = None
         self._saved_cost = None
         self._spot = None
         self._started_at = None
         self._user_id = None
+        self._written_bytes = None
         self.discriminator = None
         if billed_bytes is not None:
             self.billed_bytes = billed_bytes
@@ -104,6 +110,8 @@ class V1UsageDetails(object):
             self.instance_count = instance_count
         if instance_type is not None:
             self.instance_type = instance_type
+        if read_bytes is not None:
+            self.read_bytes = read_bytes
         if resource_name is not None:
             self.resource_name = resource_name
         if saved_cost is not None:
@@ -114,6 +122,8 @@ class V1UsageDetails(object):
             self.started_at = started_at
         if user_id is not None:
             self.user_id = user_id
+        if written_bytes is not None:
+            self.written_bytes = written_bytes
 
     @property
     def billed_bytes(self) -> 'str':
@@ -283,6 +293,27 @@ class V1UsageDetails(object):
 
         self._instance_type = instance_type
 
+    @property
+    def read_bytes(self) -> 'str':
+        """Gets the read_bytes of this V1UsageDetails.  # noqa: E501
+
+
+        :return: The read_bytes of this V1UsageDetails.  # noqa: E501
+        :rtype: str
+        """
+        return self._read_bytes
+
+    @read_bytes.setter
+    def read_bytes(self, read_bytes: 'str'):
+        """Sets the read_bytes of this V1UsageDetails.
+
+
+        :param read_bytes: The read_bytes of this V1UsageDetails.  # noqa: E501
+        :type: str
+        """
+
+        self._read_bytes = read_bytes
+
     @property
     def resource_name(self) -> 'str':
         """Gets the resource_name of this V1UsageDetails.  # noqa: E501
@@ -388,6 +419,27 @@ class V1UsageDetails(object):
 
         self._user_id = user_id
 
+    @property
+    def written_bytes(self) -> 'str':
+        """Gets the written_bytes of this V1UsageDetails.  # noqa: E501
+
+
+        :return: The written_bytes of this V1UsageDetails.  # noqa: E501
+        :rtype: str
+        """
+        return self._written_bytes
+
+    @written_bytes.setter
+    def written_bytes(self, written_bytes: 'str'):
+        """Sets the written_bytes of this V1UsageDetails.
+
+
+        :param written_bytes: The written_bytes of this V1UsageDetails.  # noqa: E501
+        :type: str
+        """
+
+        self._written_bytes = written_bytes
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -60,6 +60,7 @@ class V1UserFeatures(object):
         'custom_app_domain': 'bool',
         'custom_instance_types': 'bool',
         'default_one_cluster': 'bool',
+        'deployment_customize_api': 'bool',
         'deployment_data_path': 'bool',
         'deployment_gallery': 'bool',
         'deployment_persistent_disk': 'bool',
@@ -81,6 +82,7 @@ class V1UserFeatures(object):
         'mmt_strategy_selector': 'bool',
         'model_store': 'bool',
         'multiple_deployment_versions': 'bool',
+        'multiple_studio_versions': 'bool',
         'org_level_member_permissions': 'bool',
         'plugin_biz_chat': 'bool',
         'plugin_distributed': 'bool',
@@ -136,6 +138,7 @@ class V1UserFeatures(object):
         'custom_app_domain': 'customAppDomain',
         'custom_instance_types': 'customInstanceTypes',
         'default_one_cluster': 'defaultOneCluster',
+        'deployment_customize_api': 'deploymentCustomizeApi',
         'deployment_data_path': 'deploymentDataPath',
         'deployment_gallery': 'deploymentGallery',
         'deployment_persistent_disk': 'deploymentPersistentDisk',
@@ -157,6 +160,7 @@ class V1UserFeatures(object):
         'mmt_strategy_selector': 'mmtStrategySelector',
         'model_store': 'modelStore',
         'multiple_deployment_versions': 'multipleDeploymentVersions',
+        'multiple_studio_versions': 'multipleStudioVersions',
         'org_level_member_permissions': 'orgLevelMemberPermissions',
         'plugin_biz_chat': 'pluginBizChat',
         'plugin_distributed': 'pluginDistributed',
@@ -192,7 +196,7 @@ class V1UserFeatures(object):
         'writable_data_connections': 'writableDataConnections'
     }
 
-    def __init__(self, advanced_deployment_autoscaling: 'bool' =None, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, aws_trainium: 'bool' =None, b2c_experience: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, cluster_proxy: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, cost_attribution_settings: 'bool' =None, crypto_monitoring: 'bool' =None, custom_app_domain: 'bool' =None, custom_instance_types: 'bool' =None, default_one_cluster: 'bool' =None, deployment_data_path: 'bool' =None, deployment_gallery: 'bool' =None, deployment_persistent_disk: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, enable_crypto_crackdown: 'bool' =None, enable_efs: 'bool' =None, enable_storage_limits: 'bool' =None, featured_studios_admin: 'bool' =None, filesystem_optimisation: 'bool' =None, gcp: 'bool' =None, inference_job_deployment_plugin: 'bool' =None, jobs_init: 'bool' =None, jobs_v2: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, lit_logger_storage_v2: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_store: 'bool' =None, multiple_deployment_versions: 'bool' =None, org_level_member_permissions: 'bool' =None, plugin_biz_chat: 'bool' =None, plugin_distributed: 'bool' =None, plugin_fiftyone: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_lightning_apps: 'bool' =None, plugin_lightning_apps_distributed: 'bool' =None, plugin_mage_ai: 'bool' =None, plugin_milvus: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_react: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, plugin_weviate: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, project_selector: 'bool' =None, restart_ide_on_hang: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, s3_folders: 'bool' =None, show_dev_admin: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, snowflake_connection: 'bool' =None, spot_v2: 'bool' =None, studio_config: 'bool' =None, studio_on_stop: 'bool' =None, teamspace_storage_tab: 'bool' =None, use_rclone_mounts_only: 'bool' =None, writable_data_connections: 'bool' =None):  # noqa: E501
+    def __init__(self, advanced_deployment_autoscaling: 'bool' =None, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, aws_trainium: 'bool' =None, b2c_experience: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, cluster_proxy: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, cost_attribution_settings: 'bool' =None, crypto_monitoring: 'bool' =None, custom_app_domain: 'bool' =None, custom_instance_types: 'bool' =None, default_one_cluster: 'bool' =None, deployment_customize_api: 'bool' =None, deployment_data_path: 'bool' =None, deployment_gallery: 'bool' =None, deployment_persistent_disk: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, enable_crypto_crackdown: 'bool' =None, enable_efs: 'bool' =None, enable_storage_limits: 'bool' =None, featured_studios_admin: 'bool' =None, filesystem_optimisation: 'bool' =None, gcp: 'bool' =None, inference_job_deployment_plugin: 'bool' =None, jobs_init: 'bool' =None, jobs_v2: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, lit_logger_storage_v2: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_store: 'bool' =None, multiple_deployment_versions: 'bool' =None, multiple_studio_versions: 'bool' =None, org_level_member_permissions: 'bool' =None, plugin_biz_chat: 'bool' =None, plugin_distributed: 'bool' =None, plugin_fiftyone: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_lightning_apps: 'bool' =None, plugin_lightning_apps_distributed: 'bool' =None, plugin_mage_ai: 'bool' =None, plugin_milvus: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_react: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, plugin_weviate: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, project_selector: 'bool' =None, restart_ide_on_hang: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, s3_folders: 'bool' =None, show_dev_admin: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, snowflake_connection: 'bool' =None, spot_v2: 'bool' =None, studio_config: 'bool' =None, studio_on_stop: 'bool' =None, teamspace_storage_tab: 'bool' =None, use_rclone_mounts_only: 'bool' =None, writable_data_connections: 'bool' =None):  # noqa: E501
         """V1UserFeatures - a model defined in Swagger"""  # noqa: E501
         self._advanced_deployment_autoscaling = None
         self._affiliate_links = None
@@ -213,6 +217,7 @@ class V1UserFeatures(object):
         self._custom_app_domain = None
         self._custom_instance_types = None
         self._default_one_cluster = None
+        self._deployment_customize_api = None
         self._deployment_data_path = None
         self._deployment_gallery = None
         self._deployment_persistent_disk = None
@@ -234,6 +239,7 @@ class V1UserFeatures(object):
         self._mmt_strategy_selector = None
         self._model_store = None
         self._multiple_deployment_versions = None
+        self._multiple_studio_versions = None
         self._org_level_member_permissions = None
         self._plugin_biz_chat = None
         self._plugin_distributed = None
@@ -306,6 +312,8 @@ class V1UserFeatures(object):
             self.custom_instance_types = custom_instance_types
         if default_one_cluster is not None:
             self.default_one_cluster = default_one_cluster
+        if deployment_customize_api is not None:
+            self.deployment_customize_api = deployment_customize_api
         if deployment_data_path is not None:
             self.deployment_data_path = deployment_data_path
         if deployment_gallery is not None:
@@ -348,6 +356,8 @@ class V1UserFeatures(object):
             self.model_store = model_store
         if multiple_deployment_versions is not None:
             self.multiple_deployment_versions = multiple_deployment_versions
+        if multiple_studio_versions is not None:
+            self.multiple_studio_versions = multiple_studio_versions
         if org_level_member_permissions is not None:
             self.org_level_member_permissions = org_level_member_permissions
         if plugin_biz_chat is not None:
@@ -814,6 +824,27 @@ class V1UserFeatures(object):
 
         self._default_one_cluster = default_one_cluster
 
+    @property
+    def deployment_customize_api(self) -> 'bool':
+        """Gets the deployment_customize_api of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The deployment_customize_api of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._deployment_customize_api
+
+    @deployment_customize_api.setter
+    def deployment_customize_api(self, deployment_customize_api: 'bool'):
+        """Sets the deployment_customize_api of this V1UserFeatures.
+
+
+        :param deployment_customize_api: The deployment_customize_api of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._deployment_customize_api = deployment_customize_api
+
     @property
     def deployment_data_path(self) -> 'bool':
         """Gets the deployment_data_path of this V1UserFeatures.  # noqa: E501
@@ -1255,6 +1286,27 @@ class V1UserFeatures(object):
 
         self._multiple_deployment_versions = multiple_deployment_versions
 
+    @property
+    def multiple_studio_versions(self) -> 'bool':
+        """Gets the multiple_studio_versions of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The multiple_studio_versions of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._multiple_studio_versions
+
+    @multiple_studio_versions.setter
+    def multiple_studio_versions(self, multiple_studio_versions: 'bool'):
+        """Sets the multiple_studio_versions of this V1UserFeatures.
+
+
+        :param multiple_studio_versions: The multiple_studio_versions of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._multiple_studio_versions = multiple_studio_versions
+
     @property
     def org_level_member_permissions(self) -> 'bool':
         """Gets the org_level_member_permissions of this V1UserFeatures.  # noqa: E501

lightning_sdk/studio.py

@@ -3,6 +3,7 @@ import warnings
 from typing import TYPE_CHECKING, Any, Mapping, Optional, Tuple, Union
 
 from lightning_sdk.api.studio_api import StudioApi
+from lightning_sdk.api.utils import _machine_to_compute_name
 from lightning_sdk.constants import _LIGHTNING_DEBUG
 from lightning_sdk.machine import Machine
 from lightning_sdk.organization import Organization
@@ -145,14 +146,14 @@ class Studio:
         """Returns the cluster the Studio is running on."""
         return self._studio.cluster_id
 
-    def start(self, machine: Machine = Machine.CPU, interruptible: bool = False) -> None:
+    def start(self, machine: Union[Machine, str] = Machine.CPU, interruptible: bool = False) -> None:
         """Starts a Studio on the specified machine type (default: CPU-4)."""
         status = self.status
         if status == Status.Running:
-            curr_machine = self.machine
-            if curr_machine != machine:
+            curr_machine = _machine_to_compute_name(self.machine) if self.machine is not None else None
+            if curr_machine != _machine_to_compute_name(machine):
                 raise RuntimeError(
-                    f"Requested to start studio on {machine}, but studio is already running on {curr_machine}."
+                    f"Requested to start studio on {machine}, but studio is already running on {self.machine}."
                     " Consider switching instead!"
                 )
             _logger.info(f"Studio {self.name} is already running")
@@ -180,7 +181,7 @@ class Studio:
         kwargs = self._studio_api.duplicate_studio(self._studio.id, self._teamspace.id, self._teamspace.id)
         return Studio(**kwargs)
 
-    def switch_machine(self, machine: Machine, interruptible: bool = False) -> None:
+    def switch_machine(self, machine: Union[Machine, str], interruptible: bool = False) -> None:
         """Switches machine to the provided machine type/.
 
         Args:

{lightning_sdk-0.1.32.dist-info → lightning_sdk-0.1.34.dist-info}/METADATA

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