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

lightning_sdk/lightning_cloud/openapi/models/model_id_versions_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 ModelIdVersionsBody(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 = {
+        'cluster_id': 'str'
+    }
+
+    attribute_map = {
+        'cluster_id': 'clusterId'
+    }
+
+    def __init__(self, cluster_id: 'str' =None):  # noqa: E501
+        """ModelIdVersionsBody - a model defined in Swagger"""  # noqa: E501
+        self._cluster_id = None
+        self.discriminator = None
+        if cluster_id is not None:
+            self.cluster_id = cluster_id
+
+    @property
+    def cluster_id(self) -> 'str':
+        """Gets the cluster_id of this ModelIdVersionsBody.  # noqa: E501
+
+
+        :return: The cluster_id of this ModelIdVersionsBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._cluster_id
+
+    @cluster_id.setter
+    def cluster_id(self, cluster_id: 'str'):
+        """Sets the cluster_id of this ModelIdVersionsBody.
+
+
+        :param cluster_id: The cluster_id of this ModelIdVersionsBody.  # noqa: E501
+        :type: str
+        """
+
+        self._cluster_id = cluster_id
+
+    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(ModelIdVersionsBody, 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: 'ModelIdVersionsBody') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, ModelIdVersionsBody):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'ModelIdVersionsBody') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/project_id_models_body.py

@@ -41,6 +41,7 @@ class ProjectIdModelsBody(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'always_create': 'bool',
         'cluster_id': 'str',
         'metadata': 'dict(str, str)',
         'name': 'str',
@@ -48,19 +49,23 @@ class ProjectIdModelsBody(object):
     }
 
     attribute_map = {
+        'always_create': 'alwaysCreate',
         'cluster_id': 'clusterId',
         'metadata': 'metadata',
         'name': 'name',
         'private': 'private'
     }
 
-    def __init__(self, cluster_id: 'str' =None, metadata: 'dict(str, str)' =None, name: 'str' =None, private: 'bool' =None):  # noqa: E501
+    def __init__(self, always_create: 'bool' =None, cluster_id: 'str' =None, metadata: 'dict(str, str)' =None, name: 'str' =None, private: 'bool' =None):  # noqa: E501
         """ProjectIdModelsBody - a model defined in Swagger"""  # noqa: E501
+        self._always_create = None
         self._cluster_id = None
         self._metadata = None
         self._name = None
         self._private = None
         self.discriminator = None
+        if always_create is not None:
+            self.always_create = always_create
         if cluster_id is not None:
             self.cluster_id = cluster_id
         if metadata is not None:
@@ -70,6 +75,27 @@ class ProjectIdModelsBody(object):
         if private is not None:
             self.private = private
 
+    @property
+    def always_create(self) -> 'bool':
+        """Gets the always_create of this ProjectIdModelsBody.  # noqa: E501
+
+
+        :return: The always_create of this ProjectIdModelsBody.  # noqa: E501
+        :rtype: bool
+        """
+        return self._always_create
+
+    @always_create.setter
+    def always_create(self, always_create: 'bool'):
+        """Sets the always_create of this ProjectIdModelsBody.
+
+
+        :param always_create: The always_create of this ProjectIdModelsBody.  # noqa: E501
+        :type: bool
+        """
+
+        self._always_create = always_create
+
     @property
     def cluster_id(self) -> 'str':
         """Gets the cluster_id of this ProjectIdModelsBody.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/update.py

@@ -42,39 +42,59 @@ class Update(object):
     """
     swagger_types = {
         'aws': 'V1AwsDataConnection',
+        'efs': 'V1EfsDataConnection',
+        'efs_folder': 'V1EFSFolderDataConnection',
         'gcp': 'V1GcpDataConnection',
+        'gcs_folder': 'V1GCSFolderDataConnection',
         'name': 'str',
         'run_cmds': 'list[str]',
+        's3_folder': 'V1S3FolderDataConnection',
         'snowflake': 'V1SnowflakeDataConnection',
         'writable': 'bool'
     }
 
     attribute_map = {
         'aws': 'aws',
+        'efs': 'efs',
+        'efs_folder': 'efsFolder',
         'gcp': 'gcp',
+        'gcs_folder': 'gcsFolder',
         'name': 'name',
         'run_cmds': 'runCmds',
+        's3_folder': 's3Folder',
         'snowflake': 'snowflake',
         'writable': 'writable'
     }
 
-    def __init__(self, aws: 'V1AwsDataConnection' =None, gcp: 'V1GcpDataConnection' =None, name: 'str' =None, run_cmds: 'list[str]' =None, snowflake: 'V1SnowflakeDataConnection' =None, writable: 'bool' =None):  # noqa: E501
+    def __init__(self, aws: 'V1AwsDataConnection' =None, efs: 'V1EfsDataConnection' =None, efs_folder: 'V1EFSFolderDataConnection' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, name: 'str' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, writable: 'bool' =None):  # noqa: E501
         """Update - a model defined in Swagger"""  # noqa: E501
         self._aws = None
+        self._efs = None
+        self._efs_folder = None
         self._gcp = None
+        self._gcs_folder = None
         self._name = None
         self._run_cmds = None
+        self._s3_folder = None
         self._snowflake = None
         self._writable = None
         self.discriminator = None
         if aws is not None:
             self.aws = aws
+        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:
+            self.gcs_folder = gcs_folder
         if name is not None:
             self.name = name
         if run_cmds is not None:
             self.run_cmds = run_cmds
+        if s3_folder is not None:
+            self.s3_folder = s3_folder
         if snowflake is not None:
             self.snowflake = snowflake
         if writable is not None:
@@ -101,6 +121,48 @@ class Update(object):
 
         self._aws = aws
 
+    @property
+    def efs(self) -> 'V1EfsDataConnection':
+        """Gets the efs of this Update.  # noqa: E501
+
+
+        :return: The efs of this Update.  # noqa: E501
+        :rtype: V1EfsDataConnection
+        """
+        return self._efs
+
+    @efs.setter
+    def efs(self, efs: 'V1EfsDataConnection'):
+        """Sets the efs of this Update.
+
+
+        :param efs: The efs of this Update.  # noqa: E501
+        :type: V1EfsDataConnection
+        """
+
+        self._efs = efs
+
+    @property
+    def efs_folder(self) -> 'V1EFSFolderDataConnection':
+        """Gets the efs_folder of this Update.  # noqa: E501
+
+
+        :return: The efs_folder of this Update.  # noqa: E501
+        :rtype: V1EFSFolderDataConnection
+        """
+        return self._efs_folder
+
+    @efs_folder.setter
+    def efs_folder(self, efs_folder: 'V1EFSFolderDataConnection'):
+        """Sets the efs_folder of this Update.
+
+
+        :param efs_folder: The efs_folder of this Update.  # noqa: E501
+        :type: V1EFSFolderDataConnection
+        """
+
+        self._efs_folder = efs_folder
+
     @property
     def gcp(self) -> 'V1GcpDataConnection':
         """Gets the gcp of this Update.  # noqa: E501
@@ -122,6 +184,27 @@ class Update(object):
 
         self._gcp = gcp
 
+    @property
+    def gcs_folder(self) -> 'V1GCSFolderDataConnection':
+        """Gets the gcs_folder of this Update.  # noqa: E501
+
+
+        :return: The gcs_folder of this Update.  # noqa: E501
+        :rtype: V1GCSFolderDataConnection
+        """
+        return self._gcs_folder
+
+    @gcs_folder.setter
+    def gcs_folder(self, gcs_folder: 'V1GCSFolderDataConnection'):
+        """Sets the gcs_folder of this Update.
+
+
+        :param gcs_folder: The gcs_folder of this Update.  # noqa: E501
+        :type: V1GCSFolderDataConnection
+        """
+
+        self._gcs_folder = gcs_folder
+
     @property
     def name(self) -> 'str':
         """Gets the name of this Update.  # noqa: E501
@@ -164,6 +247,27 @@ class Update(object):
 
         self._run_cmds = run_cmds
 
+    @property
+    def s3_folder(self) -> 'V1S3FolderDataConnection':
+        """Gets the s3_folder of this Update.  # noqa: E501
+
+
+        :return: The s3_folder of this Update.  # noqa: E501
+        :rtype: V1S3FolderDataConnection
+        """
+        return self._s3_folder
+
+    @s3_folder.setter
+    def s3_folder(self, s3_folder: 'V1S3FolderDataConnection'):
+        """Sets the s3_folder of this Update.
+
+
+        :param s3_folder: The s3_folder of this Update.  # noqa: E501
+        :type: V1S3FolderDataConnection
+        """
+
+        self._s3_folder = s3_folder
+
     @property
     def snowflake(self) -> 'V1SnowflakeDataConnection':
         """Gets the snowflake of this Update.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_affiliate_link.py

@@ -47,6 +47,9 @@ class V1AffiliateLink(object):
         'id': 'str',
         'max_uses': 'int',
         'name': 'str',
+        'referred_incentives': 'dict(str, str)',
+        'skip_waitlist': 'bool',
+        'type': 'str',
         'updated_at': 'datetime',
         'url': 'str',
         'usage_count': 'int',
@@ -60,13 +63,16 @@ class V1AffiliateLink(object):
         'id': 'id',
         'max_uses': 'maxUses',
         'name': 'name',
+        'referred_incentives': 'referredIncentives',
+        'skip_waitlist': 'skipWaitlist',
+        'type': 'type',
         'updated_at': 'updatedAt',
         'url': 'url',
         'usage_count': 'usageCount',
         'user_id': 'userId'
     }
 
-    def __init__(self, code: 'str' =None, created_at: 'datetime' =None, expires_at: 'datetime' =None, id: 'str' =None, max_uses: 'int' =None, name: 'str' =None, updated_at: 'datetime' =None, url: 'str' =None, usage_count: 'int' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, code: 'str' =None, created_at: 'datetime' =None, expires_at: 'datetime' =None, id: 'str' =None, max_uses: 'int' =None, name: 'str' =None, referred_incentives: 'dict(str, str)' =None, skip_waitlist: 'bool' =None, type: 'str' =None, updated_at: 'datetime' =None, url: 'str' =None, usage_count: 'int' =None, user_id: 'str' =None):  # noqa: E501
         """V1AffiliateLink - a model defined in Swagger"""  # noqa: E501
         self._code = None
         self._created_at = None
@@ -74,6 +80,9 @@ class V1AffiliateLink(object):
         self._id = None
         self._max_uses = None
         self._name = None
+        self._referred_incentives = None
+        self._skip_waitlist = None
+        self._type = None
         self._updated_at = None
         self._url = None
         self._usage_count = None
@@ -91,6 +100,12 @@ class V1AffiliateLink(object):
             self.max_uses = max_uses
         if name is not None:
             self.name = name
+        if referred_incentives is not None:
+            self.referred_incentives = referred_incentives
+        if skip_waitlist is not None:
+            self.skip_waitlist = skip_waitlist
+        if type is not None:
+            self.type = type
         if updated_at is not None:
             self.updated_at = updated_at
         if url is not None:
@@ -226,6 +241,69 @@ class V1AffiliateLink(object):
 
         self._name = name
 
+    @property
+    def referred_incentives(self) -> 'dict(str, str)':
+        """Gets the referred_incentives of this V1AffiliateLink.  # noqa: E501
+
+
+        :return: The referred_incentives of this V1AffiliateLink.  # noqa: E501
+        :rtype: dict(str, str)
+        """
+        return self._referred_incentives
+
+    @referred_incentives.setter
+    def referred_incentives(self, referred_incentives: 'dict(str, str)'):
+        """Sets the referred_incentives of this V1AffiliateLink.
+
+
+        :param referred_incentives: The referred_incentives of this V1AffiliateLink.  # noqa: E501
+        :type: dict(str, str)
+        """
+
+        self._referred_incentives = referred_incentives
+
+    @property
+    def skip_waitlist(self) -> 'bool':
+        """Gets the skip_waitlist of this V1AffiliateLink.  # noqa: E501
+
+
+        :return: The skip_waitlist of this V1AffiliateLink.  # noqa: E501
+        :rtype: bool
+        """
+        return self._skip_waitlist
+
+    @skip_waitlist.setter
+    def skip_waitlist(self, skip_waitlist: 'bool'):
+        """Sets the skip_waitlist of this V1AffiliateLink.
+
+
+        :param skip_waitlist: The skip_waitlist of this V1AffiliateLink.  # noqa: E501
+        :type: bool
+        """
+
+        self._skip_waitlist = skip_waitlist
+
+    @property
+    def type(self) -> 'str':
+        """Gets the type of this V1AffiliateLink.  # noqa: E501
+
+
+        :return: The type of this V1AffiliateLink.  # noqa: E501
+        :rtype: str
+        """
+        return self._type
+
+    @type.setter
+    def type(self, type: 'str'):
+        """Sets the type of this V1AffiliateLink.
+
+
+        :param type: The type of this V1AffiliateLink.  # noqa: E501
+        :type: str
+        """
+
+        self._type = type
+
     @property
     def updated_at(self) -> 'datetime':
         """Gets the updated_at of this V1AffiliateLink.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_app_type.py

@@ -0,0 +1,104 @@
+# 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 V1AppType(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+
+    """
+    allowed enum values
+    """
+    JOB = "APP_TYPE_JOB"
+    MMT = "APP_TYPE_MMT"
+    DATA_PREP = "APP_TYPE_DATA_PREP"
+    """
+    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
+        """V1AppType - 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(V1AppType, 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: 'V1AppType') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1AppType):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1AppType') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py

@@ -63,6 +63,7 @@ class V1ClusterAccelerator(object):
         'is_tier_restricted': 'bool',
         'local_disk_size': 'str',
         'local_disk_supported': 'bool',
+        'max_available_quota': 'str',
         'non_spot': 'bool',
         'quota_checked_at': 'datetime',
         'quota_code': 'str',
@@ -101,6 +102,7 @@ class V1ClusterAccelerator(object):
         'is_tier_restricted': 'isTierRestricted',
         'local_disk_size': 'localDiskSize',
         'local_disk_supported': 'localDiskSupported',
+        'max_available_quota': 'maxAvailableQuota',
         'non_spot': 'nonSpot',
         'quota_checked_at': 'quotaCheckedAt',
         'quota_code': 'quotaCode',
@@ -116,7 +118,7 @@ class V1ClusterAccelerator(object):
         'spot_price': 'spotPrice'
     }
 
-    def __init__(self, accelerator_type: 'str' =None, allowed_resources: 'list[str]' =None, available_in_seconds: 'str' =None, available_in_seconds_spot: 'str' =None, available_zones: 'list[str]' =None, byoc_only: 'bool' =None, capacity_block_only: 'bool' =None, capacity_block_price: 'float' =None, capacity_blocks_available: 'list[V1ClusterCapacityReservation]' =None, cluster_id: 'str' =None, cost: 'float' =None, detailed_quotas_info: 'list[V1AcceleratorQuotaInfo]' =None, device_card: 'str' =None, device_info: 'str' =None, display_name: 'str' =None, enabled: 'bool' =None, family: 'str' =None, instance_id: 'str' =None, is_custom: 'bool' =None, is_tier_restricted: 'bool' =None, local_disk_size: 'str' =None, local_disk_supported: 'bool' =None, non_spot: 'bool' =None, quota_checked_at: 'datetime' =None, quota_code: 'str' =None, quota_name: 'str' =None, quota_page_url: 'str' =None, quota_service_code: 'str' =None, quota_utilization: 'str' =None, quota_value: 'str' =None, reservable: 'bool' =None, reservation_available_zones: 'list[str]' =None, resources: 'V1Resources' =None, slug: 'str' =None, spot_price: 'float' =None):  # noqa: E501
+    def __init__(self, accelerator_type: 'str' =None, allowed_resources: 'list[str]' =None, available_in_seconds: 'str' =None, available_in_seconds_spot: 'str' =None, available_zones: 'list[str]' =None, byoc_only: 'bool' =None, capacity_block_only: 'bool' =None, capacity_block_price: 'float' =None, capacity_blocks_available: 'list[V1ClusterCapacityReservation]' =None, cluster_id: 'str' =None, cost: 'float' =None, detailed_quotas_info: 'list[V1AcceleratorQuotaInfo]' =None, device_card: 'str' =None, device_info: 'str' =None, display_name: 'str' =None, enabled: 'bool' =None, family: 'str' =None, instance_id: 'str' =None, is_custom: 'bool' =None, is_tier_restricted: 'bool' =None, local_disk_size: 'str' =None, local_disk_supported: 'bool' =None, max_available_quota: 'str' =None, non_spot: 'bool' =None, quota_checked_at: 'datetime' =None, quota_code: 'str' =None, quota_name: 'str' =None, quota_page_url: 'str' =None, quota_service_code: 'str' =None, quota_utilization: 'str' =None, quota_value: 'str' =None, reservable: 'bool' =None, reservation_available_zones: 'list[str]' =None, resources: 'V1Resources' =None, slug: 'str' =None, spot_price: 'float' =None):  # noqa: E501
         """V1ClusterAccelerator - a model defined in Swagger"""  # noqa: E501
         self._accelerator_type = None
         self._allowed_resources = None
@@ -140,6 +142,7 @@ class V1ClusterAccelerator(object):
         self._is_tier_restricted = None
         self._local_disk_size = None
         self._local_disk_supported = None
+        self._max_available_quota = None
         self._non_spot = None
         self._quota_checked_at = None
         self._quota_code = None
@@ -198,6 +201,8 @@ class V1ClusterAccelerator(object):
             self.local_disk_size = local_disk_size
         if local_disk_supported is not None:
             self.local_disk_supported = local_disk_supported
+        if max_available_quota is not None:
+            self.max_available_quota = max_available_quota
         if non_spot is not None:
             self.non_spot = non_spot
         if quota_checked_at is not None:
@@ -687,6 +692,27 @@ class V1ClusterAccelerator(object):
 
         self._local_disk_supported = local_disk_supported
 
+    @property
+    def max_available_quota(self) -> 'str':
+        """Gets the max_available_quota of this V1ClusterAccelerator.  # noqa: E501
+
+
+        :return: The max_available_quota of this V1ClusterAccelerator.  # noqa: E501
+        :rtype: str
+        """
+        return self._max_available_quota
+
+    @max_available_quota.setter
+    def max_available_quota(self, max_available_quota: 'str'):
+        """Sets the max_available_quota of this V1ClusterAccelerator.
+
+
+        :param max_available_quota: The max_available_quota of this V1ClusterAccelerator.  # noqa: E501
+        :type: str
+        """
+
+        self._max_available_quota = max_available_quota
+
     @property
     def non_spot(self) -> 'bool':
         """Gets the non_spot of this V1ClusterAccelerator.  # noqa: E501