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

lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_compute_config.py

@@ -47,11 +47,11 @@ class V1UserRequestedComputeConfig(object):
         'disk_size': 'int',
         'gpu_image_override': 'str',
         'id': 'str',
-        'is_byom': 'bool',
         'name': 'str',
-        'preemptible': 'bool',
+        'requested_run_duration_seconds': 'str',
         'same_compute_on_resume': 'bool',
-        'shm_size': 'int'
+        'shm_size': 'int',
+        'spot': 'bool'
     }
 
     attribute_map = {
@@ -61,14 +61,14 @@ class V1UserRequestedComputeConfig(object):
         'disk_size': 'diskSize',
         'gpu_image_override': 'gpuImageOverride',
         'id': 'id',
-        'is_byom': 'isByom',
         'name': 'name',
-        'preemptible': 'preemptible',
+        'requested_run_duration_seconds': 'requestedRunDurationSeconds',
         'same_compute_on_resume': 'sameComputeOnResume',
-        'shm_size': 'shmSize'
+        'shm_size': 'shmSize',
+        'spot': 'spot'
     }
 
-    def __init__(self, affinity_identifier: 'str' =None, count: 'int' =None, cpu_image_override: 'str' =None, disk_size: 'int' =None, gpu_image_override: 'str' =None, id: 'str' =None, is_byom: 'bool' =None, name: 'str' =None, preemptible: 'bool' =None, same_compute_on_resume: 'bool' =None, shm_size: 'int' =None):  # noqa: E501
+    def __init__(self, affinity_identifier: 'str' =None, count: 'int' =None, cpu_image_override: 'str' =None, disk_size: 'int' =None, gpu_image_override: 'str' =None, id: 'str' =None, name: 'str' =None, requested_run_duration_seconds: 'str' =None, same_compute_on_resume: 'bool' =None, shm_size: 'int' =None, spot: 'bool' =None):  # noqa: E501
         """V1UserRequestedComputeConfig - a model defined in Swagger"""  # noqa: E501
         self._affinity_identifier = None
         self._count = None
@@ -76,11 +76,11 @@ class V1UserRequestedComputeConfig(object):
         self._disk_size = None
         self._gpu_image_override = None
         self._id = None
-        self._is_byom = None
         self._name = None
-        self._preemptible = None
+        self._requested_run_duration_seconds = None
         self._same_compute_on_resume = None
         self._shm_size = None
+        self._spot = None
         self.discriminator = None
         if affinity_identifier is not None:
             self.affinity_identifier = affinity_identifier
@@ -94,16 +94,16 @@ class V1UserRequestedComputeConfig(object):
             self.gpu_image_override = gpu_image_override
         if id is not None:
             self.id = id
-        if is_byom is not None:
-            self.is_byom = is_byom
         if name is not None:
             self.name = name
-        if preemptible is not None:
-            self.preemptible = preemptible
+        if requested_run_duration_seconds is not None:
+            self.requested_run_duration_seconds = requested_run_duration_seconds
         if same_compute_on_resume is not None:
             self.same_compute_on_resume = same_compute_on_resume
         if shm_size is not None:
             self.shm_size = shm_size
+        if spot is not None:
+            self.spot = spot
 
     @property
     def affinity_identifier(self) -> 'str':
@@ -237,27 +237,6 @@ class V1UserRequestedComputeConfig(object):
 
         self._id = id
 
-    @property
-    def is_byom(self) -> 'bool':
-        """Gets the is_byom of this V1UserRequestedComputeConfig.  # noqa: E501
-
-
-        :return: The is_byom of this V1UserRequestedComputeConfig.  # noqa: E501
-        :rtype: bool
-        """
-        return self._is_byom
-
-    @is_byom.setter
-    def is_byom(self, is_byom: 'bool'):
-        """Sets the is_byom of this V1UserRequestedComputeConfig.
-
-
-        :param is_byom: The is_byom of this V1UserRequestedComputeConfig.  # noqa: E501
-        :type: bool
-        """
-
-        self._is_byom = is_byom
-
     @property
     def name(self) -> 'str':
         """Gets the name of this V1UserRequestedComputeConfig.  # noqa: E501
@@ -280,25 +259,25 @@ class V1UserRequestedComputeConfig(object):
         self._name = name
 
     @property
-    def preemptible(self) -> 'bool':
-        """Gets the preemptible of this V1UserRequestedComputeConfig.  # noqa: E501
+    def requested_run_duration_seconds(self) -> 'str':
+        """Gets the requested_run_duration_seconds of this V1UserRequestedComputeConfig.  # noqa: E501
 
 
-        :return: The preemptible of this V1UserRequestedComputeConfig.  # noqa: E501
-        :rtype: bool
+        :return: The requested_run_duration_seconds of this V1UserRequestedComputeConfig.  # noqa: E501
+        :rtype: str
         """
-        return self._preemptible
+        return self._requested_run_duration_seconds
 
-    @preemptible.setter
-    def preemptible(self, preemptible: 'bool'):
-        """Sets the preemptible of this V1UserRequestedComputeConfig.
+    @requested_run_duration_seconds.setter
+    def requested_run_duration_seconds(self, requested_run_duration_seconds: 'str'):
+        """Sets the requested_run_duration_seconds of this V1UserRequestedComputeConfig.
 
 
-        :param preemptible: The preemptible of this V1UserRequestedComputeConfig.  # noqa: E501
-        :type: bool
+        :param requested_run_duration_seconds: The requested_run_duration_seconds of this V1UserRequestedComputeConfig.  # noqa: E501
+        :type: str
         """
 
-        self._preemptible = preemptible
+        self._requested_run_duration_seconds = requested_run_duration_seconds
 
     @property
     def same_compute_on_resume(self) -> 'bool':
@@ -342,6 +321,27 @@ class V1UserRequestedComputeConfig(object):
 
         self._shm_size = shm_size
 
+    @property
+    def spot(self) -> 'bool':
+        """Gets the spot of this V1UserRequestedComputeConfig.  # noqa: E501
+
+
+        :return: The spot of this V1UserRequestedComputeConfig.  # noqa: E501
+        :rtype: bool
+        """
+        return self._spot
+
+    @spot.setter
+    def spot(self, spot: 'bool'):
+        """Sets the spot of this V1UserRequestedComputeConfig.
+
+
+        :param spot: The spot of this V1UserRequestedComputeConfig.  # noqa: E501
+        :type: bool
+        """
+
+        self._spot = spot
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_flow_compute_config.py

@@ -42,28 +42,28 @@ class V1UserRequestedFlowComputeConfig(object):
     """
     swagger_types = {
         'name': 'str',
-        'preemptible': 'bool',
-        'shm_size': 'int'
+        'shm_size': 'int',
+        'spot': 'bool'
     }
 
     attribute_map = {
         'name': 'name',
-        'preemptible': 'preemptible',
-        'shm_size': 'shmSize'
+        'shm_size': 'shmSize',
+        'spot': 'spot'
     }
 
-    def __init__(self, name: 'str' =None, preemptible: 'bool' =None, shm_size: 'int' =None):  # noqa: E501
+    def __init__(self, name: 'str' =None, shm_size: 'int' =None, spot: 'bool' =None):  # noqa: E501
         """V1UserRequestedFlowComputeConfig - a model defined in Swagger"""  # noqa: E501
         self._name = None
-        self._preemptible = None
         self._shm_size = None
+        self._spot = None
         self.discriminator = None
         if name is not None:
             self.name = name
-        if preemptible is not None:
-            self.preemptible = preemptible
         if shm_size is not None:
             self.shm_size = shm_size
+        if spot is not None:
+            self.spot = spot
 
     @property
     def name(self) -> 'str':
@@ -86,27 +86,6 @@ class V1UserRequestedFlowComputeConfig(object):
 
         self._name = name
 
-    @property
-    def preemptible(self) -> 'bool':
-        """Gets the preemptible of this V1UserRequestedFlowComputeConfig.  # noqa: E501
-
-
-        :return: The preemptible of this V1UserRequestedFlowComputeConfig.  # noqa: E501
-        :rtype: bool
-        """
-        return self._preemptible
-
-    @preemptible.setter
-    def preemptible(self, preemptible: 'bool'):
-        """Sets the preemptible of this V1UserRequestedFlowComputeConfig.
-
-
-        :param preemptible: The preemptible of this V1UserRequestedFlowComputeConfig.  # noqa: E501
-        :type: bool
-        """
-
-        self._preemptible = preemptible
-
     @property
     def shm_size(self) -> 'int':
         """Gets the shm_size of this V1UserRequestedFlowComputeConfig.  # noqa: E501
@@ -128,6 +107,27 @@ class V1UserRequestedFlowComputeConfig(object):
 
         self._shm_size = shm_size
 
+    @property
+    def spot(self) -> 'bool':
+        """Gets the spot of this V1UserRequestedFlowComputeConfig.  # noqa: E501
+
+
+        :return: The spot of this V1UserRequestedFlowComputeConfig.  # noqa: E501
+        :rtype: bool
+        """
+        return self._spot
+
+    @spot.setter
+    def spot(self, spot: 'bool'):
+        """Sets the spot of this V1UserRequestedFlowComputeConfig.
+
+
+        :param spot: The spot of this V1UserRequestedFlowComputeConfig.  # noqa: E501
+        :type: bool
+        """
+
+        self._spot = spot
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

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

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

lightning_sdk/lightning_cloud/openapi/models/v1_validate_data_connection_response.py

@@ -41,19 +41,39 @@ class V1ValidateDataConnectionResponse(object):
                             and the value is json key in definition.
     """
     swagger_types = {
-        'aws': 'V1AwsDataConnection'
+        'aws': 'V1AwsDataConnection',
+        'efs': 'V1EfsConfig',
+        'gcp': 'V1GcpDataConnection',
+        'gcs_folder': 'V1GCSFolderDataConnection',
+        's3_folder': 'V1S3FolderDataConnection'
     }
 
     attribute_map = {
-        'aws': 'aws'
+        'aws': 'aws',
+        'efs': 'efs',
+        'gcp': 'gcp',
+        'gcs_folder': 'gcsFolder',
+        's3_folder': 's3Folder'
     }
 
-    def __init__(self, aws: 'V1AwsDataConnection' =None):  # noqa: E501
+    def __init__(self, aws: 'V1AwsDataConnection' =None, efs: 'V1EfsConfig' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, s3_folder: 'V1S3FolderDataConnection' =None):  # noqa: E501
         """V1ValidateDataConnectionResponse - a model defined in Swagger"""  # noqa: E501
         self._aws = None
+        self._efs = None
+        self._gcp = None
+        self._gcs_folder = None
+        self._s3_folder = None
         self.discriminator = None
         if aws is not None:
             self.aws = aws
+        if efs is not None:
+            self.efs = efs
+        if gcp is not None:
+            self.gcp = gcp
+        if gcs_folder is not None:
+            self.gcs_folder = gcs_folder
+        if s3_folder is not None:
+            self.s3_folder = s3_folder
 
     @property
     def aws(self) -> 'V1AwsDataConnection':
@@ -76,6 +96,90 @@ class V1ValidateDataConnectionResponse(object):
 
         self._aws = aws
 
+    @property
+    def efs(self) -> 'V1EfsConfig':
+        """Gets the efs of this V1ValidateDataConnectionResponse.  # noqa: E501
+
+
+        :return: The efs of this V1ValidateDataConnectionResponse.  # noqa: E501
+        :rtype: V1EfsConfig
+        """
+        return self._efs
+
+    @efs.setter
+    def efs(self, efs: 'V1EfsConfig'):
+        """Sets the efs of this V1ValidateDataConnectionResponse.
+
+
+        :param efs: The efs of this V1ValidateDataConnectionResponse.  # noqa: E501
+        :type: V1EfsConfig
+        """
+
+        self._efs = efs
+
+    @property
+    def gcp(self) -> 'V1GcpDataConnection':
+        """Gets the gcp of this V1ValidateDataConnectionResponse.  # noqa: E501
+
+
+        :return: The gcp of this V1ValidateDataConnectionResponse.  # noqa: E501
+        :rtype: V1GcpDataConnection
+        """
+        return self._gcp
+
+    @gcp.setter
+    def gcp(self, gcp: 'V1GcpDataConnection'):
+        """Sets the gcp of this V1ValidateDataConnectionResponse.
+
+
+        :param gcp: The gcp of this V1ValidateDataConnectionResponse.  # noqa: E501
+        :type: V1GcpDataConnection
+        """
+
+        self._gcp = gcp
+
+    @property
+    def gcs_folder(self) -> 'V1GCSFolderDataConnection':
+        """Gets the gcs_folder of this V1ValidateDataConnectionResponse.  # noqa: E501
+
+
+        :return: The gcs_folder of this V1ValidateDataConnectionResponse.  # noqa: E501
+        :rtype: V1GCSFolderDataConnection
+        """
+        return self._gcs_folder
+
+    @gcs_folder.setter
+    def gcs_folder(self, gcs_folder: 'V1GCSFolderDataConnection'):
+        """Sets the gcs_folder of this V1ValidateDataConnectionResponse.
+
+
+        :param gcs_folder: The gcs_folder of this V1ValidateDataConnectionResponse.  # noqa: E501
+        :type: V1GCSFolderDataConnection
+        """
+
+        self._gcs_folder = gcs_folder
+
+    @property
+    def s3_folder(self) -> 'V1S3FolderDataConnection':
+        """Gets the s3_folder of this V1ValidateDataConnectionResponse.  # noqa: E501
+
+
+        :return: The s3_folder of this V1ValidateDataConnectionResponse.  # noqa: E501
+        :rtype: V1S3FolderDataConnection
+        """
+        return self._s3_folder
+
+    @s3_folder.setter
+    def s3_folder(self, s3_folder: 'V1S3FolderDataConnection'):
+        """Sets the s3_folder of this V1ValidateDataConnectionResponse.
+
+
+        :param s3_folder: The s3_folder of this V1ValidateDataConnectionResponse.  # noqa: E501
+        :type: V1S3FolderDataConnection
+        """
+
+        self._s3_folder = s3_folder
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}