lightning-sdk 0.1.49__py3-none-any.whl → 2025.11.5__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/setup.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 Setup(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_ids': 'list[str]',
+        'gcp': 'V1GcpDataConnection'
+    }
+
+    attribute_map = {
+        'cluster_ids': 'clusterIds',
+        'gcp': 'gcp'
+    }
+
+    def __init__(self, cluster_ids: 'list[str]' =None, gcp: 'V1GcpDataConnection' =None):  # noqa: E501
+        """Setup - a model defined in Swagger"""  # noqa: E501
+        self._cluster_ids = None
+        self._gcp = None
+        self.discriminator = None
+        if cluster_ids is not None:
+            self.cluster_ids = cluster_ids
+        if gcp is not None:
+            self.gcp = gcp
+
+    @property
+    def cluster_ids(self) -> 'list[str]':
+        """Gets the cluster_ids of this Setup.  # noqa: E501
+
+
+        :return: The cluster_ids of this Setup.  # noqa: E501
+        :rtype: list[str]
+        """
+        return self._cluster_ids
+
+    @cluster_ids.setter
+    def cluster_ids(self, cluster_ids: 'list[str]'):
+        """Sets the cluster_ids of this Setup.
+
+
+        :param cluster_ids: The cluster_ids of this Setup.  # noqa: E501
+        :type: list[str]
+        """
+
+        self._cluster_ids = cluster_ids
+
+    @property
+    def gcp(self) -> 'V1GcpDataConnection':
+        """Gets the gcp of this Setup.  # noqa: E501
+
+
+        :return: The gcp of this Setup.  # noqa: E501
+        :rtype: V1GcpDataConnection
+        """
+        return self._gcp
+
+    @gcp.setter
+    def gcp(self, gcp: 'V1GcpDataConnection'):
+        """Sets the gcp of this Setup.
+
+
+        :param gcp: The gcp of this Setup.  # noqa: E501
+        :type: V1GcpDataConnection
+        """
+
+        self._gcp = gcp
+
+    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(Setup, 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: 'Setup') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, Setup):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'Setup') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/slurm_jobs_body.py

@@ -45,8 +45,11 @@ class SlurmJobsBody(object):
         'cloudspace_id': 'str',
         'cluster_id': 'str',
         'command': 'str',
+        'entrypoint': 'str',
+        'image': 'str',
+        'image_secret_ref': 'str',
         'name': 'str',
-        'num_gpus': 'str',
+        'partition': 'str',
         'service_id': 'str',
         'sync_env': 'bool',
         'work_dir': 'str'
@@ -57,21 +60,27 @@ class SlurmJobsBody(object):
         'cloudspace_id': 'cloudspaceId',
         'cluster_id': 'clusterId',
         'command': 'command',
+        'entrypoint': 'entrypoint',
+        'image': 'image',
+        'image_secret_ref': 'imageSecretRef',
         'name': 'name',
-        'num_gpus': 'numGpus',
+        'partition': 'partition',
         'service_id': 'serviceId',
         'sync_env': 'syncEnv',
         'work_dir': 'workDir'
     }
 
-    def __init__(self, cache_id: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, name: 'str' =None, num_gpus: 'str' =None, service_id: 'str' =None, sync_env: 'bool' =None, work_dir: 'str' =None):  # noqa: E501
+    def __init__(self, cache_id: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, entrypoint: 'str' =None, image: 'str' =None, image_secret_ref: 'str' =None, name: 'str' =None, partition: 'str' =None, service_id: 'str' =None, sync_env: 'bool' =None, work_dir: 'str' =None):  # noqa: E501
         """SlurmJobsBody - a model defined in Swagger"""  # noqa: E501
         self._cache_id = None
         self._cloudspace_id = None
         self._cluster_id = None
         self._command = None
+        self._entrypoint = None
+        self._image = None
+        self._image_secret_ref = None
         self._name = None
-        self._num_gpus = None
+        self._partition = None
         self._service_id = None
         self._sync_env = None
         self._work_dir = None
@@ -84,10 +93,16 @@ class SlurmJobsBody(object):
             self.cluster_id = cluster_id
         if command is not None:
             self.command = command
+        if entrypoint is not None:
+            self.entrypoint = entrypoint
+        if image is not None:
+            self.image = image
+        if image_secret_ref is not None:
+            self.image_secret_ref = image_secret_ref
         if name is not None:
             self.name = name
-        if num_gpus is not None:
-            self.num_gpus = num_gpus
+        if partition is not None:
+            self.partition = partition
         if service_id is not None:
             self.service_id = service_id
         if sync_env is not None:
@@ -179,6 +194,69 @@ class SlurmJobsBody(object):
 
         self._command = command
 
+    @property
+    def entrypoint(self) -> 'str':
+        """Gets the entrypoint of this SlurmJobsBody.  # noqa: E501
+
+
+        :return: The entrypoint of this SlurmJobsBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._entrypoint
+
+    @entrypoint.setter
+    def entrypoint(self, entrypoint: 'str'):
+        """Sets the entrypoint of this SlurmJobsBody.
+
+
+        :param entrypoint: The entrypoint of this SlurmJobsBody.  # noqa: E501
+        :type: str
+        """
+
+        self._entrypoint = entrypoint
+
+    @property
+    def image(self) -> 'str':
+        """Gets the image of this SlurmJobsBody.  # noqa: E501
+
+
+        :return: The image of this SlurmJobsBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._image
+
+    @image.setter
+    def image(self, image: 'str'):
+        """Sets the image of this SlurmJobsBody.
+
+
+        :param image: The image of this SlurmJobsBody.  # noqa: E501
+        :type: str
+        """
+
+        self._image = image
+
+    @property
+    def image_secret_ref(self) -> 'str':
+        """Gets the image_secret_ref of this SlurmJobsBody.  # noqa: E501
+
+
+        :return: The image_secret_ref of this SlurmJobsBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._image_secret_ref
+
+    @image_secret_ref.setter
+    def image_secret_ref(self, image_secret_ref: 'str'):
+        """Sets the image_secret_ref of this SlurmJobsBody.
+
+
+        :param image_secret_ref: The image_secret_ref of this SlurmJobsBody.  # noqa: E501
+        :type: str
+        """
+
+        self._image_secret_ref = image_secret_ref
+
     @property
     def name(self) -> 'str':
         """Gets the name of this SlurmJobsBody.  # noqa: E501
@@ -203,25 +281,25 @@ class SlurmJobsBody(object):
         self._name = name
 
     @property
-    def num_gpus(self) -> 'str':
-        """Gets the num_gpus of this SlurmJobsBody.  # noqa: E501
+    def partition(self) -> 'str':
+        """Gets the partition of this SlurmJobsBody.  # noqa: E501
 
 
-        :return: The num_gpus of this SlurmJobsBody.  # noqa: E501
+        :return: The partition of this SlurmJobsBody.  # noqa: E501
         :rtype: str
         """
-        return self._num_gpus
+        return self._partition
 
-    @num_gpus.setter
-    def num_gpus(self, num_gpus: 'str'):
-        """Sets the num_gpus of this SlurmJobsBody.
+    @partition.setter
+    def partition(self, partition: 'str'):
+        """Sets the partition of this SlurmJobsBody.
 
 
-        :param num_gpus: The num_gpus of this SlurmJobsBody.  # noqa: E501
+        :param partition: The partition of this SlurmJobsBody.  # noqa: E501
         :type: str
         """
 
-        self._num_gpus = num_gpus
+        self._partition = partition
 
     @property
     def service_id(self) -> 'str':

lightning_sdk/lightning_cloud/openapi/models/storage_complete_body.py

@@ -42,36 +42,41 @@ class StorageCompleteBody(object):
     """
     swagger_types = {
         'cluster_id': 'str',
+        'data_connection_id': 'str',
         'filename': 'str',
         'parts': 'list[V1CompleteUpload]',
-        'suppress_reindex': 'bool',
+        'reindex': 'bool',
         'upload_id': 'str'
     }
 
     attribute_map = {
         'cluster_id': 'clusterId',
+        'data_connection_id': 'dataConnectionId',
         'filename': 'filename',
         'parts': 'parts',
-        'suppress_reindex': 'suppressReindex',
+        'reindex': 'reindex',
         'upload_id': 'uploadId'
     }
 
-    def __init__(self, cluster_id: 'str' =None, filename: 'str' =None, parts: 'list[V1CompleteUpload]' =None, suppress_reindex: 'bool' =None, upload_id: 'str' =None):  # noqa: E501
+    def __init__(self, cluster_id: 'str' =None, data_connection_id: 'str' =None, filename: 'str' =None, parts: 'list[V1CompleteUpload]' =None, reindex: 'bool' =None, upload_id: 'str' =None):  # noqa: E501
         """StorageCompleteBody - a model defined in Swagger"""  # noqa: E501
         self._cluster_id = None
+        self._data_connection_id = None
         self._filename = None
         self._parts = None
-        self._suppress_reindex = None
+        self._reindex = None
         self._upload_id = None
         self.discriminator = None
         if cluster_id is not None:
             self.cluster_id = cluster_id
+        if data_connection_id is not None:
+            self.data_connection_id = data_connection_id
         if filename is not None:
             self.filename = filename
         if parts is not None:
             self.parts = parts
-        if suppress_reindex is not None:
-            self.suppress_reindex = suppress_reindex
+        if reindex is not None:
+            self.reindex = reindex
         if upload_id is not None:
             self.upload_id = upload_id
 
@@ -96,6 +101,27 @@ class StorageCompleteBody(object):
 
         self._cluster_id = cluster_id
 
+    @property
+    def data_connection_id(self) -> 'str':
+        """Gets the data_connection_id of this StorageCompleteBody.  # noqa: E501
+
+
+        :return: The data_connection_id of this StorageCompleteBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._data_connection_id
+
+    @data_connection_id.setter
+    def data_connection_id(self, data_connection_id: 'str'):
+        """Sets the data_connection_id of this StorageCompleteBody.
+
+
+        :param data_connection_id: The data_connection_id of this StorageCompleteBody.  # noqa: E501
+        :type: str
+        """
+
+        self._data_connection_id = data_connection_id
+
     @property
     def filename(self) -> 'str':
         """Gets the filename of this StorageCompleteBody.  # noqa: E501
@@ -139,25 +165,25 @@ class StorageCompleteBody(object):
         self._parts = parts
 
     @property
-    def suppress_reindex(self) -> 'bool':
-        """Gets the suppress_reindex of this StorageCompleteBody.  # noqa: E501
+    def reindex(self) -> 'bool':
+        """Gets the reindex of this StorageCompleteBody.  # noqa: E501
 
 
-        :return: The suppress_reindex of this StorageCompleteBody.  # noqa: E501
+        :return: The reindex of this StorageCompleteBody.  # noqa: E501
         :rtype: bool
         """
-        return self._suppress_reindex
+        return self._reindex
 
-    @suppress_reindex.setter
-    def suppress_reindex(self, suppress_reindex: 'bool'):
-        """Sets the suppress_reindex of this StorageCompleteBody.
+    @reindex.setter
+    def reindex(self, reindex: 'bool'):
+        """Sets the reindex of this StorageCompleteBody.
 
 
-        :param suppress_reindex: The suppress_reindex of this StorageCompleteBody.  # noqa: E501
+        :param reindex: The reindex of this StorageCompleteBody.  # noqa: E501
         :type: bool
         """
 
-        self._suppress_reindex = suppress_reindex
+        self._reindex = reindex
 
     @property
     def upload_id(self) -> 'str':

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