lightning-sdk 0.1.46__py3-none-any.whl → 0.1.47__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -327,6 +327,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_capacity_res
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_encryption_key_response import V1DeleteClusterEncryptionKeyResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_proxy_response import V1DeleteClusterProxyResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_response import V1DeleteClusterResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_container_response import V1DeleteContainerResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_conversation_response import V1DeleteConversationResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_data_connection_response import V1DeleteDataConnectionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_dataset_response import V1DeleteDatasetResponse

lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py

@@ -148,6 +148,107 @@ class LitRegistryServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def lit_registry_service_delete_lit_container(self, project_id: 'str', lit_container_name: 'str', **kwargs) -> 'V1DeleteContainerResponse':  # noqa: E501
+        """lit_registry_service_delete_lit_container  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.lit_registry_service_delete_lit_container(project_id, lit_container_name, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str lit_container_name: (required)
+        :return: V1DeleteContainerResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.lit_registry_service_delete_lit_container_with_http_info(project_id, lit_container_name, **kwargs)  # noqa: E501
+        else:
+            (data) = self.lit_registry_service_delete_lit_container_with_http_info(project_id, lit_container_name, **kwargs)  # noqa: E501
+            return data
+
+    def lit_registry_service_delete_lit_container_with_http_info(self, project_id: 'str', lit_container_name: 'str', **kwargs) -> 'V1DeleteContainerResponse':  # noqa: E501
+        """lit_registry_service_delete_lit_container  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.lit_registry_service_delete_lit_container_with_http_info(project_id, lit_container_name, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str lit_container_name: (required)
+        :return: V1DeleteContainerResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_id', 'lit_container_name']  # 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 lit_registry_service_delete_lit_container" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # 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 `lit_registry_service_delete_lit_container`")  # noqa: E501
+        # verify the required parameter 'lit_container_name' is set
+        if ('lit_container_name' not in params or
+                params['lit_container_name'] is None):
+            raise ValueError("Missing the required parameter `lit_container_name` when calling `lit_registry_service_delete_lit_container`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'lit_container_name' in params:
+            path_params['litContainerName'] = params['lit_container_name']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/v1/projects/{projectId}/lit-registry/{litContainerName}', 'DELETE',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1DeleteContainerResponse',  # 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 lit_registry_service_get_lit_project_registry(self, project_id: 'str', **kwargs) -> 'V1LitRegistryProject':  # noqa: E501
         """lit_registry_service_get_lit_project_registry  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/models/__init__.py

@@ -290,6 +290,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_capacity_res
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_encryption_key_response import V1DeleteClusterEncryptionKeyResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_proxy_response import V1DeleteClusterProxyResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_response import V1DeleteClusterResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_container_response import V1DeleteContainerResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_conversation_response import V1DeleteConversationResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_data_connection_response import V1DeleteDataConnectionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_dataset_response import V1DeleteDatasetResponse

lightning_sdk/lightning_cloud/openapi/models/v1_delete_container_response.py

@@ -0,0 +1,97 @@
+# 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 V1DeleteContainerResponse(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 = {
+    }
+
+    attribute_map = {
+    }
+
+    def __init__(self):  # noqa: E501
+        """V1DeleteContainerResponse - 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(V1DeleteContainerResponse, 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: 'V1DeleteContainerResponse') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1DeleteContainerResponse):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1DeleteContainerResponse') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_deployment_api.py

@@ -44,6 +44,7 @@ class V1DeploymentAPI(object):
         'body': 'V1Body',
         'headers': 'list[V1Header]',
         'method': 'str',
+        'name': 'str',
         'path': 'str',
         '_query_params': 'list[V1QueryParam]'
     }
@@ -52,15 +53,17 @@ class V1DeploymentAPI(object):
         'body': 'body',
         'headers': 'headers',
         'method': 'method',
+        'name': 'name',
         'path': 'path',
         '_query_params': 'queryParams'
     }
 
-    def __init__(self, body: 'V1Body' =None, headers: 'list[V1Header]' =None, method: 'str' =None, path: 'str' =None, _query_params: 'list[V1QueryParam]' =None):  # noqa: E501
+    def __init__(self, body: 'V1Body' =None, headers: 'list[V1Header]' =None, method: 'str' =None, name: 'str' =None, path: 'str' =None, _query_params: 'list[V1QueryParam]' =None):  # noqa: E501
         """V1DeploymentAPI - a model defined in Swagger"""  # noqa: E501
         self._body = None
         self._headers = None
         self._method = None
+        self._name = None
         self._path = None
         self.__query_params = None
         self.discriminator = None
@@ -70,6 +73,8 @@ class V1DeploymentAPI(object):
             self.headers = headers
         if method is not None:
             self.method = method
+        if name is not None:
+            self.name = name
         if path is not None:
             self.path = path
         if _query_params is not None:
@@ -138,6 +143,27 @@ class V1DeploymentAPI(object):
 
         self._method = method
 
+    @property
+    def name(self) -> 'str':
+        """Gets the name of this V1DeploymentAPI.  # noqa: E501
+
+
+        :return: The name of this V1DeploymentAPI.  # noqa: E501
+        :rtype: str
+        """
+        return self._name
+
+    @name.setter
+    def name(self, name: 'str'):
+        """Sets the name of this V1DeploymentAPI.
+
+
+        :param name: The name of this V1DeploymentAPI.  # noqa: E501
+        :type: str
+        """
+
+        self._name = name
+
     @property
     def path(self) -> 'str':
         """Gets the path of this V1DeploymentAPI.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_deployment_state.py

@@ -41,6 +41,7 @@ class V1DeploymentState(object):
     RUNNING = "DEPLOYMENT_STATE_RUNNING"
     DELETED = "DEPLOYMENT_STATE_DELETED"
     FAILED = "DEPLOYMENT_STATE_FAILED"
+    SUSPENDED = "DEPLOYMENT_STATE_SUSPENDED"
     """
     Attributes:
       swagger_types (dict): The key is attribute name

lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py

@@ -42,7 +42,6 @@ class V1GoogleCloudDirectV1(object):
     """
     swagger_types = {
         'bucket_name': 'str',
-        'compute_list_bucket_project_role': 'str',
         'compute_project_role': 'str',
         'credentials_secret_id': 'str',
         'credentials_service_account_email': 'str',
@@ -55,7 +54,6 @@ class V1GoogleCloudDirectV1(object):
 
     attribute_map = {
         'bucket_name': 'bucketName',
-        'compute_list_bucket_project_role': 'computeListBucketProjectRole',
         'compute_project_role': 'computeProjectRole',
         'credentials_secret_id': 'credentialsSecretId',
         'credentials_service_account_email': 'credentialsServiceAccountEmail',
@@ -66,10 +64,9 @@ class V1GoogleCloudDirectV1(object):
         'source_cidr_ips': 'sourceCidrIps'
     }
 
-    def __init__(self, bucket_name: 'str' =None, compute_list_bucket_project_role: 'str' =None, compute_project_role: 'str' =None, credentials_secret_id: 'str' =None, credentials_service_account_email: 'str' =None, primary_region: 'str' =None, project_id: 'str' =None, regions: 'list[str]' =None, service_account_email: 'str' =None, source_cidr_ips: 'list[str]' =None):  # noqa: E501
+    def __init__(self, bucket_name: 'str' =None, compute_project_role: 'str' =None, credentials_secret_id: 'str' =None, credentials_service_account_email: 'str' =None, primary_region: 'str' =None, project_id: 'str' =None, regions: 'list[str]' =None, service_account_email: 'str' =None, source_cidr_ips: 'list[str]' =None):  # noqa: E501
         """V1GoogleCloudDirectV1 - a model defined in Swagger"""  # noqa: E501
         self._bucket_name = None
-        self._compute_list_bucket_project_role = None
         self._compute_project_role = None
         self._credentials_secret_id = None
         self._credentials_service_account_email = None
@@ -81,8 +78,6 @@ class V1GoogleCloudDirectV1(object):
         self.discriminator = None
         if bucket_name is not None:
             self.bucket_name = bucket_name
-        if compute_list_bucket_project_role is not None:
-            self.compute_list_bucket_project_role = compute_list_bucket_project_role
         if compute_project_role is not None:
             self.compute_project_role = compute_project_role
         if credentials_secret_id is not None:
@@ -121,27 +116,6 @@ class V1GoogleCloudDirectV1(object):
 
         self._bucket_name = bucket_name
 
-    @property
-    def compute_list_bucket_project_role(self) -> 'str':
-        """Gets the compute_list_bucket_project_role of this V1GoogleCloudDirectV1.  # noqa: E501
-
-
-        :return: The compute_list_bucket_project_role of this V1GoogleCloudDirectV1.  # noqa: E501
-        :rtype: str
-        """
-        return self._compute_list_bucket_project_role
-
-    @compute_list_bucket_project_role.setter
-    def compute_list_bucket_project_role(self, compute_list_bucket_project_role: 'str'):
-        """Sets the compute_list_bucket_project_role of this V1GoogleCloudDirectV1.
-
-
-        :param compute_list_bucket_project_role: The compute_list_bucket_project_role of this V1GoogleCloudDirectV1.  # noqa: E501
-        :type: str
-        """
-
-        self._compute_list_bucket_project_role = compute_list_bucket_project_role
-
     @property
     def compute_project_role(self) -> 'str':
         """Gets the compute_project_role of this V1GoogleCloudDirectV1.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_job.py

@@ -47,6 +47,7 @@ class V1Job(object):
         'id': 'str',
         'is_healthy': 'bool',
         'message': 'str',
+        'multi_machine_job_id': 'str',
         'name': 'str',
         'overprovisioned': 'bool',
         'project_id': 'str',
@@ -73,6 +74,7 @@ class V1Job(object):
         'id': 'id',
         'is_healthy': 'isHealthy',
         'message': 'message',
+        'multi_machine_job_id': 'multiMachineJobId',
         'name': 'name',
         'overprovisioned': 'overprovisioned',
         'project_id': 'projectId',
@@ -92,7 +94,7 @@ class V1Job(object):
         'user_logs_started_at': 'userLogsStartedAt'
     }
 
-    def __init__(self, created_at: 'datetime' =None, deployment_id: 'str' =None, endpoint: 'V1Endpoint' =None, id: 'str' =None, is_healthy: 'bool' =None, message: 'str' =None, name: 'str' =None, overprovisioned: 'bool' =None, project_id: 'str' =None, ready_at: 'datetime' =None, release_id: 'str' =None, server_error: 'str' =None, spec: 'V1JobSpec' =None, started_at: 'datetime' =None, state: 'str' =None, stopped_at: 'datetime' =None, timings: 'dict(str, V1JobTiming)' =None, total_cost: 'float' =None, total_size_bytes: 'str' =None, updated_at: 'datetime' =None, urls: 'list[str]' =None, user_id: 'str' =None, user_logs_started_at: 'datetime' =None):  # noqa: E501
+    def __init__(self, created_at: 'datetime' =None, deployment_id: 'str' =None, endpoint: 'V1Endpoint' =None, id: 'str' =None, is_healthy: 'bool' =None, message: 'str' =None, multi_machine_job_id: 'str' =None, name: 'str' =None, overprovisioned: 'bool' =None, project_id: 'str' =None, ready_at: 'datetime' =None, release_id: 'str' =None, server_error: 'str' =None, spec: 'V1JobSpec' =None, started_at: 'datetime' =None, state: 'str' =None, stopped_at: 'datetime' =None, timings: 'dict(str, V1JobTiming)' =None, total_cost: 'float' =None, total_size_bytes: 'str' =None, updated_at: 'datetime' =None, urls: 'list[str]' =None, user_id: 'str' =None, user_logs_started_at: 'datetime' =None):  # noqa: E501
         """V1Job - a model defined in Swagger"""  # noqa: E501
         self._created_at = None
         self._deployment_id = None
@@ -100,6 +102,7 @@ class V1Job(object):
         self._id = None
         self._is_healthy = None
         self._message = None
+        self._multi_machine_job_id = None
         self._name = None
         self._overprovisioned = None
         self._project_id = None
@@ -130,6 +133,8 @@ class V1Job(object):
             self.is_healthy = is_healthy
         if message is not None:
             self.message = message
+        if multi_machine_job_id is not None:
+            self.multi_machine_job_id = multi_machine_job_id
         if name is not None:
             self.name = name
         if overprovisioned is not None:
@@ -291,6 +296,27 @@ class V1Job(object):
 
         self._message = message
 
+    @property
+    def multi_machine_job_id(self) -> 'str':
+        """Gets the multi_machine_job_id of this V1Job.  # noqa: E501
+
+
+        :return: The multi_machine_job_id of this V1Job.  # noqa: E501
+        :rtype: str
+        """
+        return self._multi_machine_job_id
+
+    @multi_machine_job_id.setter
+    def multi_machine_job_id(self, multi_machine_job_id: 'str'):
+        """Sets the multi_machine_job_id of this V1Job.
+
+
+        :param multi_machine_job_id: The multi_machine_job_id of this V1Job.  # noqa: E501
+        :type: str
+        """
+
+        self._multi_machine_job_id = multi_machine_job_id
+
     @property
     def name(self) -> 'str':
         """Gets the name of this V1Job.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_resource_visibility.py

@@ -41,19 +41,24 @@ class V1ResourceVisibility(object):
                             and the value is json key in definition.
     """
     swagger_types = {
-        'all_users': 'bool'
+        'all_users': 'bool',
+        'user_ids': 'list[str]'
     }
 
     attribute_map = {
-        'all_users': 'allUsers'
+        'all_users': 'allUsers',
+        'user_ids': 'userIds'
     }
 
-    def __init__(self, all_users: 'bool' =None):  # noqa: E501
+    def __init__(self, all_users: 'bool' =None, user_ids: 'list[str]' =None):  # noqa: E501
         """V1ResourceVisibility - a model defined in Swagger"""  # noqa: E501
         self._all_users = None
+        self._user_ids = None
         self.discriminator = None
         if all_users is not None:
             self.all_users = all_users
+        if user_ids is not None:
+            self.user_ids = user_ids
 
     @property
     def all_users(self) -> 'bool':
@@ -76,6 +81,27 @@ class V1ResourceVisibility(object):
 
         self._all_users = all_users
 
+    @property
+    def user_ids(self) -> 'list[str]':
+        """Gets the user_ids of this V1ResourceVisibility.  # noqa: E501
+
+
+        :return: The user_ids of this V1ResourceVisibility.  # noqa: E501
+        :rtype: list[str]
+        """
+        return self._user_ids
+
+    @user_ids.setter
+    def user_ids(self, user_ids: 'list[str]'):
+        """Sets the user_ids of this V1ResourceVisibility.
+
+
+        :param user_ids: The user_ids of this V1ResourceVisibility.  # noqa: E501
+        :type: list[str]
+        """
+
+        self._user_ids = user_ids
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -88,6 +88,7 @@ class V1UserFeatures(object):
         'multiple_deployment_versions': 'bool',
         'multiple_studio_versions': 'bool',
         'org_level_member_permissions': 'bool',
+        'pipelines': 'bool',
         'plugin_biz_chat': 'bool',
         'plugin_distributed': 'bool',
         'plugin_fiftyone': 'bool',
@@ -172,6 +173,7 @@ class V1UserFeatures(object):
         'multiple_deployment_versions': 'multipleDeploymentVersions',
         'multiple_studio_versions': 'multipleStudioVersions',
         'org_level_member_permissions': 'orgLevelMemberPermissions',
+        'pipelines': 'pipelines',
         'plugin_biz_chat': 'pluginBizChat',
         'plugin_distributed': 'pluginDistributed',
         'plugin_fiftyone': 'pluginFiftyone',
@@ -208,7 +210,7 @@ class V1UserFeatures(object):
         'writable_data_connections': 'writableDataConnections'
     }
 
-    def __init__(self, advanced_deployment_autoscaling: 'bool' =None, 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, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: '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, 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, deployment_version_visibility: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, dynamic_workload_scheduling: '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, instant_capacity_reservation: 'bool' =None, jobs_init: 'bool' =None, jobs_v2: 'bool' =None, landing_studios: 'bool' =None, lightning_registry: 'bool' =None, lit_logger: 'bool' =None, lit_logger_storage_v2: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, mmt_v2: '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, show_dev_admin: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, snapshotter_service: 'bool' =None, snowflake_connection: 'bool' =None, spot_v2: 'bool' =None, studio_config: 'bool' =None, studio_on_stop: 'bool' =None, studio_version_visibility: 'bool' =None, teamspace_storage_tab: 'bool' =None, trainium2: '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, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, b2c_experience: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: '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, 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, deployment_version_visibility: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, dynamic_workload_scheduling: '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, instant_capacity_reservation: 'bool' =None, jobs_init: 'bool' =None, jobs_v2: 'bool' =None, landing_studios: 'bool' =None, lightning_registry: 'bool' =None, lit_logger: 'bool' =None, lit_logger_storage_v2: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, mmt_v2: 'bool' =None, model_store: 'bool' =None, multiple_deployment_versions: 'bool' =None, multiple_studio_versions: 'bool' =None, org_level_member_permissions: 'bool' =None, pipelines: '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, show_dev_admin: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, snapshotter_service: 'bool' =None, snowflake_connection: 'bool' =None, spot_v2: 'bool' =None, studio_config: 'bool' =None, studio_on_stop: 'bool' =None, studio_version_visibility: 'bool' =None, teamspace_storage_tab: 'bool' =None, trainium2: '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
@@ -257,6 +259,7 @@ class V1UserFeatures(object):
         self._multiple_deployment_versions = None
         self._multiple_studio_versions = None
         self._org_level_member_permissions = None
+        self._pipelines = None
         self._plugin_biz_chat = None
         self._plugin_distributed = None
         self._plugin_fiftyone = None
@@ -386,6 +389,8 @@ class V1UserFeatures(object):
             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 pipelines is not None:
+            self.pipelines = pipelines
         if plugin_biz_chat is not None:
             self.plugin_biz_chat = plugin_biz_chat
         if plugin_distributed is not None:
@@ -1442,6 +1447,27 @@ class V1UserFeatures(object):
 
         self._org_level_member_permissions = org_level_member_permissions
 
+    @property
+    def pipelines(self) -> 'bool':
+        """Gets the pipelines of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The pipelines of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._pipelines
+
+    @pipelines.setter
+    def pipelines(self, pipelines: 'bool'):
+        """Sets the pipelines of this V1UserFeatures.
+
+
+        :param pipelines: The pipelines of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._pipelines = pipelines
+
     @property
     def plugin_biz_chat(self) -> 'bool':
         """Gets the plugin_biz_chat of this V1UserFeatures.  # noqa: E501

lightning_sdk/lightning_cloud/utils/data_connection.py

@@ -133,7 +133,6 @@ def create_efs_folder(folder_name: str, region: str) -> None:
         cluster_id=cluster_id,
         access_cluster_ids=[cluster_id],
         force=True,
-        writable=True,
         efs=V1EfsConfig(region=region),
     )
     try:
@@ -181,4 +180,4 @@ def delete_data_connection(name: str):
         #  It may exhaust the attempts before the connection is actually unmounted from the studio.
         #  for now it's best to actually stop the studio and all other things where the connection
         #  is mounted before trying to delete it
-        raise e from None
+        raise e from None

lightning_sdk/lit_registry.py

@@ -0,0 +1,39 @@
+from typing import Dict, List, Optional
+
+from lightning_sdk.api.lit_registry_api import LitRegistryApi
+from lightning_sdk.utils.resolve import _resolve_teamspace
+
+
+class LitRegistry:
+    def __init__(self) -> None:
+        self._api = LitRegistryApi()
+
+    def list_containers(
+        self, teamspace: str, org: Optional[str] = None, user: Optional[str] = None
+    ) -> List[Dict[str, str]]:
+        """List available containers.
+
+        Args:
+            teamspace: The teamspace to list containers from.
+            org: The organization to list containers from.
+            user: The user to list the containers from.
+
+        Returns:
+            A list of dictionaries containing repository details.
+        """
+        try:
+            teamspace = _resolve_teamspace(teamspace=teamspace, org=org, user=user)
+        except Exception as e:
+            raise ValueError(f"Could not resolve teamspace: {e}") from e
+        project_id = teamspace.id
+        repositories = self._api.list_containers(project_id)
+        table = []
+        for repo in repositories:
+            table.append(
+                {
+                    "REPOSITORY": repo.name,
+                    "IMAGE ID": repo.id,
+                    "CREATED": repo.creation_time.strftime("%Y-%m-%d %H:%M:%S"),
+                }
+            )
+        return table

lightning_sdk/machine.py

@@ -26,3 +26,7 @@ class Machine(DeprecationEnum):
     A100_X_8 = "A100_X_8"
     H100_X_8 = "H100_X_8"
     H200_X_8 = "H200_X_8"
+
+    def __str__(self) -> str:
+        """String representation of the enum."""
+        return str(self.value)