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

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -52,7 +52,6 @@ class V1UserFeatures(object):
         'cap_drop': 'list[str]',
         'capacity_reservation_byoc': 'bool',
         'capacity_reservation_dry_run': 'bool',
-        'cluster_proxy': 'bool',
         'code_tab': 'bool',
         'collab_screen_sharing': 'bool',
         'cost_attribution_settings': 'bool',
@@ -66,7 +65,6 @@ class V1UserFeatures(object):
         'deployment_version_visibility': 'bool',
         'docs_agent': 'bool',
         'drive_v2': 'bool',
-        'dynamic_workload_scheduling': 'bool',
         'enable_crypto_crackdown': 'bool',
         'enable_efs': 'bool',
         'enable_storage_limits': 'bool',
@@ -88,6 +86,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',
@@ -136,7 +135,6 @@ class V1UserFeatures(object):
         'cap_drop': 'capDrop',
         'capacity_reservation_byoc': 'capacityReservationByoc',
         'capacity_reservation_dry_run': 'capacityReservationDryRun',
-        'cluster_proxy': 'clusterProxy',
         'code_tab': 'codeTab',
         'collab_screen_sharing': 'collabScreenSharing',
         'cost_attribution_settings': 'costAttributionSettings',
@@ -150,7 +148,6 @@ class V1UserFeatures(object):
         'deployment_version_visibility': 'deploymentVersionVisibility',
         'docs_agent': 'docsAgent',
         'drive_v2': 'driveV2',
-        'dynamic_workload_scheduling': 'dynamicWorkloadScheduling',
         'enable_crypto_crackdown': 'enableCryptoCrackdown',
         'enable_efs': 'enableEfs',
         'enable_storage_limits': 'enableStorageLimits',
@@ -172,6 +169,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 +206,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, 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, 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
@@ -221,7 +219,6 @@ class V1UserFeatures(object):
         self._cap_drop = None
         self._capacity_reservation_byoc = None
         self._capacity_reservation_dry_run = None
-        self._cluster_proxy = None
         self._code_tab = None
         self._collab_screen_sharing = None
         self._cost_attribution_settings = None
@@ -235,7 +232,6 @@ class V1UserFeatures(object):
         self._deployment_version_visibility = None
         self._docs_agent = None
         self._drive_v2 = None
-        self._dynamic_workload_scheduling = None
         self._enable_crypto_crackdown = None
         self._enable_efs = None
         self._enable_storage_limits = None
@@ -257,6 +253,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
@@ -314,8 +311,6 @@ class V1UserFeatures(object):
             self.capacity_reservation_byoc = capacity_reservation_byoc
         if capacity_reservation_dry_run is not None:
             self.capacity_reservation_dry_run = capacity_reservation_dry_run
-        if cluster_proxy is not None:
-            self.cluster_proxy = cluster_proxy
         if code_tab is not None:
             self.code_tab = code_tab
         if collab_screen_sharing is not None:
@@ -342,8 +337,6 @@ class V1UserFeatures(object):
             self.docs_agent = docs_agent
         if drive_v2 is not None:
             self.drive_v2 = drive_v2
-        if dynamic_workload_scheduling is not None:
-            self.dynamic_workload_scheduling = dynamic_workload_scheduling
         if enable_crypto_crackdown is not None:
             self.enable_crypto_crackdown = enable_crypto_crackdown
         if enable_efs is not None:
@@ -386,6 +379,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:
@@ -686,27 +681,6 @@ class V1UserFeatures(object):
 
         self._capacity_reservation_dry_run = capacity_reservation_dry_run
 
-    @property
-    def cluster_proxy(self) -> 'bool':
-        """Gets the cluster_proxy of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The cluster_proxy of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._cluster_proxy
-
-    @cluster_proxy.setter
-    def cluster_proxy(self, cluster_proxy: 'bool'):
-        """Sets the cluster_proxy of this V1UserFeatures.
-
-
-        :param cluster_proxy: The cluster_proxy of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._cluster_proxy = cluster_proxy
-
     @property
     def code_tab(self) -> 'bool':
         """Gets the code_tab of this V1UserFeatures.  # noqa: E501
@@ -980,27 +954,6 @@ class V1UserFeatures(object):
 
         self._drive_v2 = drive_v2
 
-    @property
-    def dynamic_workload_scheduling(self) -> 'bool':
-        """Gets the dynamic_workload_scheduling of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The dynamic_workload_scheduling of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._dynamic_workload_scheduling
-
-    @dynamic_workload_scheduling.setter
-    def dynamic_workload_scheduling(self, dynamic_workload_scheduling: 'bool'):
-        """Sets the dynamic_workload_scheduling of this V1UserFeatures.
-
-
-        :param dynamic_workload_scheduling: The dynamic_workload_scheduling of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._dynamic_workload_scheduling = dynamic_workload_scheduling
-
     @property
     def enable_crypto_crackdown(self) -> 'bool':
         """Gets the enable_crypto_crackdown of this V1UserFeatures.  # noqa: E501
@@ -1442,6 +1395,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/openapi/models/v1_vultr_direct_v1.py

@@ -41,19 +41,29 @@ class V1VultrDirectV1(object):
                             and the value is json key in definition.
     """
     swagger_types = {
-        'credentials_secret_id': 'str'
+        'credentials_secret_id': 'str',
+        'primary_region': 'str',
+        'regions': 'list[str]'
     }
 
     attribute_map = {
-        'credentials_secret_id': 'credentialsSecretId'
+        'credentials_secret_id': 'credentialsSecretId',
+        'primary_region': 'primaryRegion',
+        'regions': 'regions'
     }
 
-    def __init__(self, credentials_secret_id: 'str' =None):  # noqa: E501
+    def __init__(self, credentials_secret_id: 'str' =None, primary_region: 'str' =None, regions: 'list[str]' =None):  # noqa: E501
         """V1VultrDirectV1 - a model defined in Swagger"""  # noqa: E501
         self._credentials_secret_id = None
+        self._primary_region = None
+        self._regions = None
         self.discriminator = None
         if credentials_secret_id is not None:
             self.credentials_secret_id = credentials_secret_id
+        if primary_region is not None:
+            self.primary_region = primary_region
+        if regions is not None:
+            self.regions = regions
 
     @property
     def credentials_secret_id(self) -> 'str':
@@ -78,6 +88,48 @@ class V1VultrDirectV1(object):
 
         self._credentials_secret_id = credentials_secret_id
 
+    @property
+    def primary_region(self) -> 'str':
+        """Gets the primary_region of this V1VultrDirectV1.  # noqa: E501
+
+
+        :return: The primary_region of this V1VultrDirectV1.  # noqa: E501
+        :rtype: str
+        """
+        return self._primary_region
+
+    @primary_region.setter
+    def primary_region(self, primary_region: 'str'):
+        """Sets the primary_region of this V1VultrDirectV1.
+
+
+        :param primary_region: The primary_region of this V1VultrDirectV1.  # noqa: E501
+        :type: str
+        """
+
+        self._primary_region = primary_region
+
+    @property
+    def regions(self) -> 'list[str]':
+        """Gets the regions of this V1VultrDirectV1.  # noqa: E501
+
+
+        :return: The regions of this V1VultrDirectV1.  # noqa: E501
+        :rtype: list[str]
+        """
+        return self._regions
+
+    @regions.setter
+    def regions(self, regions: 'list[str]'):
+        """Sets the regions of this V1VultrDirectV1.
+
+
+        :param regions: The regions of this V1VultrDirectV1.  # noqa: E501
+        :type: list[str]
+        """
+
+        self._regions = regions
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/version_default_body.py

@@ -0,0 +1,149 @@
+# 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 VersionDefaultBody(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 = {
+        'index': 'int',
+        'model_name': 'str'
+    }
+
+    attribute_map = {
+        'index': 'index',
+        'model_name': 'modelName'
+    }
+
+    def __init__(self, index: 'int' =None, model_name: 'str' =None):  # noqa: E501
+        """VersionDefaultBody - a model defined in Swagger"""  # noqa: E501
+        self._index = None
+        self._model_name = None
+        self.discriminator = None
+        if index is not None:
+            self.index = index
+        if model_name is not None:
+            self.model_name = model_name
+
+    @property
+    def index(self) -> 'int':
+        """Gets the index of this VersionDefaultBody.  # noqa: E501
+
+
+        :return: The index of this VersionDefaultBody.  # noqa: E501
+        :rtype: int
+        """
+        return self._index
+
+    @index.setter
+    def index(self, index: 'int'):
+        """Sets the index of this VersionDefaultBody.
+
+
+        :param index: The index of this VersionDefaultBody.  # noqa: E501
+        :type: int
+        """
+
+        self._index = index
+
+    @property
+    def model_name(self) -> 'str':
+        """Gets the model_name of this VersionDefaultBody.  # noqa: E501
+
+
+        :return: The model_name of this VersionDefaultBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._model_name
+
+    @model_name.setter
+    def model_name(self, model_name: 'str'):
+        """Sets the model_name of this VersionDefaultBody.
+
+
+        :param model_name: The model_name of this VersionDefaultBody.  # noqa: E501
+        :type: str
+        """
+
+        self._model_name = model_name
+
+    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(VersionDefaultBody, 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: 'VersionDefaultBody') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, VersionDefaultBody):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'VersionDefaultBody') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/versions_version_body.py

@@ -0,0 +1,123 @@
+# coding: utf-8
+
+"""
+    external/v1/auth_service.proto
+
+    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
+
+    OpenAPI spec version: version not set
+    
+    Generated by: https://github.com/swagger-api/swagger-codegen.git
+
+    NOTE
+    ----
+    standard swagger-codegen-cli for this python client has been modified
+    by custom templates. The purpose of these templates is to include
+    typing information in the API and Model code. Please refer to the
+    main grid repository for more info
+"""
+
+import pprint
+import re  # noqa: F401
+
+from typing import TYPE_CHECKING
+
+import six
+
+if TYPE_CHECKING:
+    from datetime import datetime
+    from lightning_sdk.lightning_cloud.openapi.models import *
+
+class VersionsVersionBody(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 = {
+        'name': 'str'
+    }
+
+    attribute_map = {
+        'name': 'name'
+    }
+
+    def __init__(self, name: 'str' =None):  # noqa: E501
+        """VersionsVersionBody - a model defined in Swagger"""  # noqa: E501
+        self._name = None
+        self.discriminator = None
+        if name is not None:
+            self.name = name
+
+    @property
+    def name(self) -> 'str':
+        """Gets the name of this VersionsVersionBody.  # noqa: E501
+
+
+        :return: The name of this VersionsVersionBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._name
+
+    @name.setter
+    def name(self, name: 'str'):
+        """Sets the name of this VersionsVersionBody.
+
+
+        :param name: The name of this VersionsVersionBody.  # noqa: E501
+        :type: str
+        """
+
+        self._name = name
+
+    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(VersionsVersionBody, 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: 'VersionsVersionBody') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, VersionsVersionBody):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'VersionsVersionBody') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/utils/data_connection.py

@@ -181,4 +181,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_container.py

@@ -0,0 +1,57 @@
+from typing import Dict, List, Optional
+
+from lightning_sdk.api.lit_container_api import LitContainerApi
+from lightning_sdk.utils.resolve import _resolve_teamspace
+
+
+class LitContainer:
+    def __init__(self) -> None:
+        self._api = LitContainerApi()
+
+    def list_containers(
+        self, teamspace: str, org: Optional[str] = None, user: Optional[str] = None
+    ) -> List[Dict[str, str]]:
+        """List available docker repositories.
+
+        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
+
+    def delete_container(
+        self, container: str, teamspace: str, org: Optional[str] = None, user: Optional[str] = None
+    ) -> None:
+        """Delete a docker container.
+
+        Args:
+            container: Name of the container to delete.
+            teamspace: The teamspace which contains the container.
+            org: The organization which contains the container.
+            user: The user which contains the container.
+        """
+        try:
+            teamspace = _resolve_teamspace(teamspace=teamspace, org=org, user=user)
+        except Exception as e:
+            raise ValueError("Could not resolve teamspace") from e
+        project_id = teamspace.id
+        return self._api.delete_container(project_id, container)

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)