lightning-sdk 0.1.36__py3-none-any.whl → 0.1.38__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py

@@ -51,6 +51,7 @@ class V1ClusterSpec(object):
         'freeze_accelerators': 'bool',
         'google_cloud_v1': 'V1GoogleCloudDirectV1',
         'insurer_disabled': 'bool',
+        'lambda_labs_v1': 'V1LambdaLabsDirectV1',
         'overprovisioning': 'list[V1InstanceOverprovisioningSpec]',
         'pause_automation': 'bool',
         'security_options': 'V1ClusterSecurityOptions',
@@ -70,6 +71,7 @@ class V1ClusterSpec(object):
         'freeze_accelerators': 'freezeAccelerators',
         'google_cloud_v1': 'googleCloudV1',
         'insurer_disabled': 'insurerDisabled',
+        'lambda_labs_v1': 'lambdaLabsV1',
         'overprovisioning': 'overprovisioning',
         'pause_automation': 'pauseAutomation',
         'security_options': 'securityOptions',
@@ -78,7 +80,7 @@ class V1ClusterSpec(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, auth_token: 'str' =None, available_instance_types: 'list[str]' =None, aws_v1: 'V1AWSDirectV1' =None, cluster_type: 'V1ClusterType' =None, deletion_options: 'V1ClusterDeletionOptions' =None, desired_state: 'V1ClusterState' =None, domain: 'str' =None, freeze_accelerators: 'bool' =None, google_cloud_v1: 'V1GoogleCloudDirectV1' =None, insurer_disabled: 'bool' =None, overprovisioning: 'list[V1InstanceOverprovisioningSpec]' =None, pause_automation: 'bool' =None, security_options: 'V1ClusterSecurityOptions' =None, slurm_v1: 'V1SlurmV1' =None, tagging_options: 'V1ClusterTaggingOptions' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, auth_token: 'str' =None, available_instance_types: 'list[str]' =None, aws_v1: 'V1AWSDirectV1' =None, cluster_type: 'V1ClusterType' =None, deletion_options: 'V1ClusterDeletionOptions' =None, desired_state: 'V1ClusterState' =None, domain: 'str' =None, freeze_accelerators: 'bool' =None, google_cloud_v1: 'V1GoogleCloudDirectV1' =None, insurer_disabled: 'bool' =None, lambda_labs_v1: 'V1LambdaLabsDirectV1' =None, overprovisioning: 'list[V1InstanceOverprovisioningSpec]' =None, pause_automation: 'bool' =None, security_options: 'V1ClusterSecurityOptions' =None, slurm_v1: 'V1SlurmV1' =None, tagging_options: 'V1ClusterTaggingOptions' =None, user_id: 'str' =None):  # noqa: E501
         """V1ClusterSpec - a model defined in Swagger"""  # noqa: E501
         self._auth_token = None
         self._available_instance_types = None
@@ -90,6 +92,7 @@ class V1ClusterSpec(object):
         self._freeze_accelerators = None
         self._google_cloud_v1 = None
         self._insurer_disabled = None
+        self._lambda_labs_v1 = None
         self._overprovisioning = None
         self._pause_automation = None
         self._security_options = None
@@ -117,6 +120,8 @@ class V1ClusterSpec(object):
             self.google_cloud_v1 = google_cloud_v1
         if insurer_disabled is not None:
             self.insurer_disabled = insurer_disabled
+        if lambda_labs_v1 is not None:
+            self.lambda_labs_v1 = lambda_labs_v1
         if overprovisioning is not None:
             self.overprovisioning = overprovisioning
         if pause_automation is not None:
@@ -344,6 +349,27 @@ class V1ClusterSpec(object):
 
         self._insurer_disabled = insurer_disabled
 
+    @property
+    def lambda_labs_v1(self) -> 'V1LambdaLabsDirectV1':
+        """Gets the lambda_labs_v1 of this V1ClusterSpec.  # noqa: E501
+
+
+        :return: The lambda_labs_v1 of this V1ClusterSpec.  # noqa: E501
+        :rtype: V1LambdaLabsDirectV1
+        """
+        return self._lambda_labs_v1
+
+    @lambda_labs_v1.setter
+    def lambda_labs_v1(self, lambda_labs_v1: 'V1LambdaLabsDirectV1'):
+        """Sets the lambda_labs_v1 of this V1ClusterSpec.
+
+
+        :param lambda_labs_v1: The lambda_labs_v1 of this V1ClusterSpec.  # noqa: E501
+        :type: V1LambdaLabsDirectV1
+        """
+
+        self._lambda_labs_v1 = lambda_labs_v1
+
     @property
     def overprovisioning(self) -> 'list[V1InstanceOverprovisioningSpec]':
         """Gets the overprovisioning of this V1ClusterSpec.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_create_deployment_template_request.py

@@ -46,10 +46,11 @@ class V1CreateDeploymentTemplateRequest(object):
         'deployment_id': 'str',
         'description': 'str',
         'image_url': 'str',
-        'metrics': 'list[V1DeploymentMetrics]',
+        'metrics': 'V1DeploymentMetrics',
         'name': 'str',
         'org_id': 'str',
         'parameter_spec': 'V1ParameterizationSpec',
+        'pricing': 'V1ApiPricingSpec',
         'project_id': 'str',
         'spec': 'str',
         'tags': 'list[V1ResourceTag]',
@@ -69,6 +70,7 @@ class V1CreateDeploymentTemplateRequest(object):
         'name': 'name',
         'org_id': 'orgId',
         'parameter_spec': 'parameterSpec',
+        'pricing': 'pricing',
         'project_id': 'projectId',
         'spec': 'spec',
         'tags': 'tags',
@@ -78,7 +80,7 @@ class V1CreateDeploymentTemplateRequest(object):
         'visibility': 'visibility'
     }
 
-    def __init__(self, about_page_content: 'str' =None, categories: 'list[str]' =None, deployment_id: 'str' =None, description: 'str' =None, image_url: 'str' =None, metrics: 'list[V1DeploymentMetrics]' =None, name: 'str' =None, org_id: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, project_id: 'str' =None, spec: 'str' =None, tags: 'list[V1ResourceTag]' =None, thumbnail: 'str' =None, thumbnail_file_type: 'str' =None, version: 'str' =None, visibility: 'V1DeploymentTemplateType' =None):  # noqa: E501
+    def __init__(self, about_page_content: 'str' =None, categories: 'list[str]' =None, deployment_id: 'str' =None, description: 'str' =None, image_url: 'str' =None, metrics: 'V1DeploymentMetrics' =None, name: 'str' =None, org_id: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, pricing: 'V1ApiPricingSpec' =None, project_id: 'str' =None, spec: 'str' =None, tags: 'list[V1ResourceTag]' =None, thumbnail: 'str' =None, thumbnail_file_type: 'str' =None, version: 'str' =None, visibility: 'V1DeploymentTemplateType' =None):  # noqa: E501
         """V1CreateDeploymentTemplateRequest - a model defined in Swagger"""  # noqa: E501
         self._about_page_content = None
         self._categories = None
@@ -89,6 +91,7 @@ class V1CreateDeploymentTemplateRequest(object):
         self._name = None
         self._org_id = None
         self._parameter_spec = None
+        self._pricing = None
         self._project_id = None
         self._spec = None
         self._tags = None
@@ -115,6 +118,8 @@ class V1CreateDeploymentTemplateRequest(object):
             self.org_id = org_id
         if parameter_spec is not None:
             self.parameter_spec = parameter_spec
+        if pricing is not None:
+            self.pricing = pricing
         if project_id is not None:
             self.project_id = project_id
         if spec is not None:
@@ -236,22 +241,22 @@ class V1CreateDeploymentTemplateRequest(object):
         self._image_url = image_url
 
     @property
-    def metrics(self) -> 'list[V1DeploymentMetrics]':
+    def metrics(self) -> 'V1DeploymentMetrics':
         """Gets the metrics of this V1CreateDeploymentTemplateRequest.  # noqa: E501
 
 
         :return: The metrics of this V1CreateDeploymentTemplateRequest.  # noqa: E501
-        :rtype: list[V1DeploymentMetrics]
+        :rtype: V1DeploymentMetrics
         """
         return self._metrics
 
     @metrics.setter
-    def metrics(self, metrics: 'list[V1DeploymentMetrics]'):
+    def metrics(self, metrics: 'V1DeploymentMetrics'):
         """Sets the metrics of this V1CreateDeploymentTemplateRequest.
 
 
         :param metrics: The metrics of this V1CreateDeploymentTemplateRequest.  # noqa: E501
-        :type: list[V1DeploymentMetrics]
+        :type: V1DeploymentMetrics
         """
 
         self._metrics = metrics
@@ -319,6 +324,27 @@ class V1CreateDeploymentTemplateRequest(object):
 
         self._parameter_spec = parameter_spec
 
+    @property
+    def pricing(self) -> 'V1ApiPricingSpec':
+        """Gets the pricing of this V1CreateDeploymentTemplateRequest.  # noqa: E501
+
+
+        :return: The pricing of this V1CreateDeploymentTemplateRequest.  # noqa: E501
+        :rtype: V1ApiPricingSpec
+        """
+        return self._pricing
+
+    @pricing.setter
+    def pricing(self, pricing: 'V1ApiPricingSpec'):
+        """Sets the pricing of this V1CreateDeploymentTemplateRequest.
+
+
+        :param pricing: The pricing of this V1CreateDeploymentTemplateRequest.  # noqa: E501
+        :type: V1ApiPricingSpec
+        """
+
+        self._pricing = pricing
+
     @property
     def project_id(self) -> 'str':
         """Gets the project_id of this V1CreateDeploymentTemplateRequest.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py

@@ -46,8 +46,7 @@ class V1DataConnection(object):
         'aws': 'V1AwsDataConnection',
         'cluster_id': 'str',
         'created_at': 'datetime',
-        'efs': 'V1EfsDataConnection',
-        'efs_folder': 'V1EFSFolderDataConnection',
+        'efs': 'V1EfsConfig',
         'gcp': 'V1GcpDataConnection',
         'gcs_folder': 'V1GCSFolderDataConnection',
         'id': 'str',
@@ -74,7 +73,6 @@ class V1DataConnection(object):
         'cluster_id': 'clusterId',
         'created_at': 'createdAt',
         'efs': 'efs',
-        'efs_folder': 'efsFolder',
         'gcp': 'gcp',
         'gcs_folder': 'gcsFolder',
         'id': 'id',
@@ -94,7 +92,7 @@ class V1DataConnection(object):
         'writable': 'writable'
     }
 
-    def __init__(self, access_cluster_ids: 'list[str]' =None, accessible: 'bool' =None, aws: 'V1AwsDataConnection' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, efs: 'V1EfsDataConnection' =None, efs_folder: 'V1EFSFolderDataConnection' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, id: 'str' =None, index: 'V1Index' =None, is_managed: 'bool' =None, lock_out: 'bool' =None, name: 'str' =None, number_of_files: 'str' =None, project_id: 'str' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, state: 'V1DataConnectionState' =None, total_size_bytes: 'str' =None, type: 'str' =None, updated_at: 'datetime' =None, writable: 'bool' =None):  # noqa: E501
+    def __init__(self, access_cluster_ids: 'list[str]' =None, accessible: 'bool' =None, aws: 'V1AwsDataConnection' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, efs: 'V1EfsConfig' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, id: 'str' =None, index: 'V1Index' =None, is_managed: 'bool' =None, lock_out: 'bool' =None, name: 'str' =None, number_of_files: 'str' =None, project_id: 'str' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, state: 'V1DataConnectionState' =None, total_size_bytes: 'str' =None, type: 'str' =None, updated_at: 'datetime' =None, writable: 'bool' =None):  # noqa: E501
         """V1DataConnection - a model defined in Swagger"""  # noqa: E501
         self._access_cluster_ids = None
         self._accessible = None
@@ -102,7 +100,6 @@ class V1DataConnection(object):
         self._cluster_id = None
         self._created_at = None
         self._efs = None
-        self._efs_folder = None
         self._gcp = None
         self._gcs_folder = None
         self._id = None
@@ -133,8 +130,6 @@ class V1DataConnection(object):
             self.created_at = created_at
         if efs is not None:
             self.efs = efs
-        if efs_folder is not None:
-            self.efs_folder = efs_folder
         if gcp is not None:
             self.gcp = gcp
         if gcs_folder is not None:
@@ -276,47 +271,26 @@ class V1DataConnection(object):
         self._created_at = created_at
 
     @property
-    def efs(self) -> 'V1EfsDataConnection':
+    def efs(self) -> 'V1EfsConfig':
         """Gets the efs of this V1DataConnection.  # noqa: E501
 
 
         :return: The efs of this V1DataConnection.  # noqa: E501
-        :rtype: V1EfsDataConnection
+        :rtype: V1EfsConfig
         """
         return self._efs
 
     @efs.setter
-    def efs(self, efs: 'V1EfsDataConnection'):
+    def efs(self, efs: 'V1EfsConfig'):
         """Sets the efs of this V1DataConnection.
 
 
         :param efs: The efs of this V1DataConnection.  # noqa: E501
-        :type: V1EfsDataConnection
+        :type: V1EfsConfig
         """
 
         self._efs = efs
 
-    @property
-    def efs_folder(self) -> 'V1EFSFolderDataConnection':
-        """Gets the efs_folder of this V1DataConnection.  # noqa: E501
-
-
-        :return: The efs_folder of this V1DataConnection.  # noqa: E501
-        :rtype: V1EFSFolderDataConnection
-        """
-        return self._efs_folder
-
-    @efs_folder.setter
-    def efs_folder(self, efs_folder: 'V1EFSFolderDataConnection'):
-        """Sets the efs_folder of this V1DataConnection.
-
-
-        :param efs_folder: The efs_folder of this V1DataConnection.  # noqa: E501
-        :type: V1EFSFolderDataConnection
-        """
-
-        self._efs_folder = efs_folder
-
     @property
     def gcp(self) -> 'V1GcpDataConnection':
         """Gets the gcp of this V1DataConnection.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_delete_multi_machine_job_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 V1DeleteMultiMachineJobResponse(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
+        """V1DeleteMultiMachineJobResponse - 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(V1DeleteMultiMachineJobResponse, 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: 'V1DeleteMultiMachineJobResponse') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1DeleteMultiMachineJobResponse):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1DeleteMultiMachineJobResponse') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py

@@ -56,6 +56,7 @@ class V1Deployment(object):
         'spec': 'V1JobSpec',
         'status': 'V1DeploymentStatus',
         'strategy': 'V1DeploymentStrategy',
+        'template_id': 'str',
         'updated_at': 'datetime',
         'user_id': 'str'
     }
@@ -76,11 +77,12 @@ class V1Deployment(object):
         'spec': 'spec',
         'status': 'status',
         'strategy': 'strategy',
+        'template_id': 'templateId',
         'updated_at': 'updatedAt',
         'user_id': 'userId'
     }
 
-    def __init__(self, autoscaling: 'V1AutoscalingSpec' =None, cloudspace_id: 'str' =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
+    def __init__(self, autoscaling: 'V1AutoscalingSpec' =None, cloudspace_id: 'str' =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, template_id: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """V1Deployment - a model defined in Swagger"""  # noqa: E501
         self._autoscaling = None
         self._cloudspace_id = None
@@ -97,6 +99,7 @@ class V1Deployment(object):
         self._spec = None
         self._status = None
         self._strategy = None
+        self._template_id = None
         self._updated_at = None
         self._user_id = None
         self.discriminator = None
@@ -130,6 +133,8 @@ class V1Deployment(object):
             self.status = status
         if strategy is not None:
             self.strategy = strategy
+        if template_id is not None:
+            self.template_id = template_id
         if updated_at is not None:
             self.updated_at = updated_at
         if user_id is not None:
@@ -450,6 +455,27 @@ class V1Deployment(object):
 
         self._strategy = strategy
 
+    @property
+    def template_id(self) -> 'str':
+        """Gets the template_id of this V1Deployment.  # noqa: E501
+
+
+        :return: The template_id of this V1Deployment.  # noqa: E501
+        :rtype: str
+        """
+        return self._template_id
+
+    @template_id.setter
+    def template_id(self, template_id: 'str'):
+        """Sets the template_id of this V1Deployment.
+
+
+        :param template_id: The template_id of this V1Deployment.  # noqa: E501
+        :type: str
+        """
+
+        self._template_id = template_id
+
     @property
     def updated_at(self) -> 'datetime':
         """Gets the updated_at of this V1Deployment.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_deployment_metrics.py

@@ -41,40 +41,66 @@ class V1DeploymentMetrics(object):
                             and the value is json key in definition.
     """
     swagger_types = {
-        'url': 'str'
+        'performance': 'V1DeploymentPerformance',
+        'urls': 'list[str]'
     }
 
     attribute_map = {
-        'url': 'url'
+        'performance': 'performance',
+        'urls': 'urls'
     }
 
-    def __init__(self, url: 'str' =None):  # noqa: E501
+    def __init__(self, performance: 'V1DeploymentPerformance' =None, urls: 'list[str]' =None):  # noqa: E501
         """V1DeploymentMetrics - a model defined in Swagger"""  # noqa: E501
-        self._url = None
+        self._performance = None
+        self._urls = None
         self.discriminator = None
-        if url is not None:
-            self.url = url
+        if performance is not None:
+            self.performance = performance
+        if urls is not None:
+            self.urls = urls
 
     @property
-    def url(self) -> 'str':
-        """Gets the url of this V1DeploymentMetrics.  # noqa: E501
+    def performance(self) -> 'V1DeploymentPerformance':
+        """Gets the performance of this V1DeploymentMetrics.  # noqa: E501
 
 
-        :return: The url of this V1DeploymentMetrics.  # noqa: E501
-        :rtype: str
+        :return: The performance of this V1DeploymentMetrics.  # noqa: E501
+        :rtype: V1DeploymentPerformance
         """
-        return self._url
+        return self._performance
 
-    @url.setter
-    def url(self, url: 'str'):
-        """Sets the url of this V1DeploymentMetrics.
+    @performance.setter
+    def performance(self, performance: 'V1DeploymentPerformance'):
+        """Sets the performance of this V1DeploymentMetrics.
 
 
-        :param url: The url of this V1DeploymentMetrics.  # noqa: E501
-        :type: str
+        :param performance: The performance of this V1DeploymentMetrics.  # noqa: E501
+        :type: V1DeploymentPerformance
         """
 
-        self._url = url
+        self._performance = performance
+
+    @property
+    def urls(self) -> 'list[str]':
+        """Gets the urls of this V1DeploymentMetrics.  # noqa: E501
+
+
+        :return: The urls of this V1DeploymentMetrics.  # noqa: E501
+        :rtype: list[str]
+        """
+        return self._urls
+
+    @urls.setter
+    def urls(self, urls: 'list[str]'):
+        """Sets the urls of this V1DeploymentMetrics.
+
+
+        :param urls: The urls of this V1DeploymentMetrics.  # noqa: E501
+        :type: list[str]
+        """
+
+        self._urls = urls
 
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""