lightning-sdk 0.1.37__py3-none-any.whl → 0.1.39__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/{v1_efs_data_connection.py → v1_efs_config.py}

@@ -28,7 +28,7 @@ if TYPE_CHECKING:
     from datetime import datetime
     from lightning_sdk.lightning_cloud.openapi.models import *
 
-class V1EfsDataConnection(object):
+class V1EfsConfig(object):
     """NOTE: This class is auto generated by the swagger code generator program.
 
     Do not edit the class manually.
@@ -55,7 +55,7 @@ class V1EfsDataConnection(object):
     }
 
     def __init__(self, file_system_id: 'str' =None, mount_targets: 'list[V1MountTarget]' =None, region: 'str' =None, source: 'str' =None):  # noqa: E501
-        """V1EfsDataConnection - a model defined in Swagger"""  # noqa: E501
+        """V1EfsConfig - a model defined in Swagger"""  # noqa: E501
         self._file_system_id = None
         self._mount_targets = None
         self._region = None
@@ -72,20 +72,20 @@ class V1EfsDataConnection(object):
 
     @property
     def file_system_id(self) -> 'str':
-        """Gets the file_system_id of this V1EfsDataConnection.  # noqa: E501
+        """Gets the file_system_id of this V1EfsConfig.  # noqa: E501
 
 
-        :return: The file_system_id of this V1EfsDataConnection.  # noqa: E501
+        :return: The file_system_id of this V1EfsConfig.  # noqa: E501
         :rtype: str
         """
         return self._file_system_id
 
     @file_system_id.setter
     def file_system_id(self, file_system_id: 'str'):
-        """Sets the file_system_id of this V1EfsDataConnection.
+        """Sets the file_system_id of this V1EfsConfig.
 
 
-        :param file_system_id: The file_system_id of this V1EfsDataConnection.  # noqa: E501
+        :param file_system_id: The file_system_id of this V1EfsConfig.  # noqa: E501
         :type: str
         """
 
@@ -93,20 +93,20 @@ class V1EfsDataConnection(object):
 
     @property
     def mount_targets(self) -> 'list[V1MountTarget]':
-        """Gets the mount_targets of this V1EfsDataConnection.  # noqa: E501
+        """Gets the mount_targets of this V1EfsConfig.  # noqa: E501
 
 
-        :return: The mount_targets of this V1EfsDataConnection.  # noqa: E501
+        :return: The mount_targets of this V1EfsConfig.  # noqa: E501
         :rtype: list[V1MountTarget]
         """
         return self._mount_targets
 
     @mount_targets.setter
     def mount_targets(self, mount_targets: 'list[V1MountTarget]'):
-        """Sets the mount_targets of this V1EfsDataConnection.
+        """Sets the mount_targets of this V1EfsConfig.
 
 
-        :param mount_targets: The mount_targets of this V1EfsDataConnection.  # noqa: E501
+        :param mount_targets: The mount_targets of this V1EfsConfig.  # noqa: E501
         :type: list[V1MountTarget]
         """
 
@@ -114,20 +114,20 @@ class V1EfsDataConnection(object):
 
     @property
     def region(self) -> 'str':
-        """Gets the region of this V1EfsDataConnection.  # noqa: E501
+        """Gets the region of this V1EfsConfig.  # noqa: E501
 
 
-        :return: The region of this V1EfsDataConnection.  # noqa: E501
+        :return: The region of this V1EfsConfig.  # noqa: E501
         :rtype: str
         """
         return self._region
 
     @region.setter
     def region(self, region: 'str'):
-        """Sets the region of this V1EfsDataConnection.
+        """Sets the region of this V1EfsConfig.
 
 
-        :param region: The region of this V1EfsDataConnection.  # noqa: E501
+        :param region: The region of this V1EfsConfig.  # noqa: E501
         :type: str
         """
 
@@ -135,20 +135,20 @@ class V1EfsDataConnection(object):
 
     @property
     def source(self) -> 'str':
-        """Gets the source of this V1EfsDataConnection.  # noqa: E501
+        """Gets the source of this V1EfsConfig.  # noqa: E501
 
 
-        :return: The source of this V1EfsDataConnection.  # noqa: E501
+        :return: The source of this V1EfsConfig.  # noqa: E501
         :rtype: str
         """
         return self._source
 
     @source.setter
     def source(self, source: 'str'):
-        """Sets the source of this V1EfsDataConnection.
+        """Sets the source of this V1EfsConfig.
 
 
-        :param source: The source of this V1EfsDataConnection.  # noqa: E501
+        :param source: The source of this V1EfsConfig.  # noqa: E501
         :type: str
         """
 
@@ -175,7 +175,7 @@ class V1EfsDataConnection(object):
                 ))
             else:
                 result[attr] = value
-        if issubclass(V1EfsDataConnection, dict):
+        if issubclass(V1EfsConfig, dict):
             for key, value in self.items():
                 result[key] = value
 
@@ -189,13 +189,13 @@ class V1EfsDataConnection(object):
         """For `print` and `pprint`"""
         return self.to_str()
 
-    def __eq__(self, other: 'V1EfsDataConnection') -> bool:
+    def __eq__(self, other: 'V1EfsConfig') -> bool:
         """Returns true if both objects are equal"""
-        if not isinstance(other, V1EfsDataConnection):
+        if not isinstance(other, V1EfsConfig):
             return False
 
         return self.__dict__ == other.__dict__
 
-    def __ne__(self, other: 'V1EfsDataConnection') -> bool:
+    def __ne__(self, other: 'V1EfsConfig') -> bool:
         """Returns true if both objects are not equal"""
         return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_response.py

@@ -45,6 +45,7 @@ class V1GetModelFilesResponse(object):
         'model_id': 'str',
         'project_id': 'str',
         'size_bytes': 'str',
+        'upload_complete': 'bool',
         'version': 'str'
     }
 
@@ -53,15 +54,17 @@ class V1GetModelFilesResponse(object):
         'model_id': 'modelId',
         'project_id': 'projectId',
         'size_bytes': 'sizeBytes',
+        'upload_complete': 'uploadComplete',
         'version': 'version'
     }
 
-    def __init__(self, filepaths: 'list[str]' =None, model_id: 'str' =None, project_id: 'str' =None, size_bytes: 'str' =None, version: 'str' =None):  # noqa: E501
+    def __init__(self, filepaths: 'list[str]' =None, model_id: 'str' =None, project_id: 'str' =None, size_bytes: 'str' =None, upload_complete: 'bool' =None, version: 'str' =None):  # noqa: E501
         """V1GetModelFilesResponse - a model defined in Swagger"""  # noqa: E501
         self._filepaths = None
         self._model_id = None
         self._project_id = None
         self._size_bytes = None
+        self._upload_complete = None
         self._version = None
         self.discriminator = None
         if filepaths is not None:
@@ -72,6 +75,8 @@ class V1GetModelFilesResponse(object):
             self.project_id = project_id
         if size_bytes is not None:
             self.size_bytes = size_bytes
+        if upload_complete is not None:
+            self.upload_complete = upload_complete
         if version is not None:
             self.version = version
 
@@ -159,6 +164,27 @@ class V1GetModelFilesResponse(object):
 
         self._size_bytes = size_bytes
 
+    @property
+    def upload_complete(self) -> 'bool':
+        """Gets the upload_complete of this V1GetModelFilesResponse.  # noqa: E501
+
+
+        :return: The upload_complete of this V1GetModelFilesResponse.  # noqa: E501
+        :rtype: bool
+        """
+        return self._upload_complete
+
+    @upload_complete.setter
+    def upload_complete(self, upload_complete: 'bool'):
+        """Sets the upload_complete of this V1GetModelFilesResponse.
+
+
+        :param upload_complete: The upload_complete of this V1GetModelFilesResponse.  # noqa: E501
+        :type: bool
+        """
+
+        self._upload_complete = upload_complete
+
     @property
     def version(self) -> 'str':
         """Gets the version of this V1GetModelFilesResponse.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py

@@ -42,6 +42,8 @@ class V1GoogleCloudDirectV1(object):
     """
     swagger_types = {
         'bucket_name': 'str',
+        'compute_list_bucket_project_role': 'str',
+        'compute_project_role': 'str',
         'credentials_secret_id': 'str',
         'credentials_service_account_email': 'str',
         'primary_region': 'str',
@@ -53,6 +55,8 @@ class V1GoogleCloudDirectV1(object):
 
     attribute_map = {
         'bucket_name': 'bucketName',
+        'compute_list_bucket_project_role': 'computeListBucketProjectRole',
+        'compute_project_role': 'computeProjectRole',
         'credentials_secret_id': 'credentialsSecretId',
         'credentials_service_account_email': 'credentialsServiceAccountEmail',
         'primary_region': 'primaryRegion',
@@ -62,9 +66,11 @@ class V1GoogleCloudDirectV1(object):
         'source_cidr_ips': 'sourceCidrIps'
     }
 
-    def __init__(self, bucket_name: 'str' =None, credentials_secret_id: 'str' =None, credentials_service_account_email: 'str' =None, primary_region: 'str' =None, project_id: 'str' =None, regions: 'list[str]' =None, service_account_email: 'str' =None, source_cidr_ips: 'list[str]' =None):  # noqa: E501
+    def __init__(self, bucket_name: 'str' =None, compute_list_bucket_project_role: 'str' =None, compute_project_role: 'str' =None, credentials_secret_id: 'str' =None, credentials_service_account_email: 'str' =None, primary_region: 'str' =None, project_id: 'str' =None, regions: 'list[str]' =None, service_account_email: 'str' =None, source_cidr_ips: 'list[str]' =None):  # noqa: E501
         """V1GoogleCloudDirectV1 - a model defined in Swagger"""  # noqa: E501
         self._bucket_name = None
+        self._compute_list_bucket_project_role = None
+        self._compute_project_role = None
         self._credentials_secret_id = None
         self._credentials_service_account_email = None
         self._primary_region = None
@@ -75,6 +81,10 @@ class V1GoogleCloudDirectV1(object):
         self.discriminator = None
         if bucket_name is not None:
             self.bucket_name = bucket_name
+        if compute_list_bucket_project_role is not None:
+            self.compute_list_bucket_project_role = compute_list_bucket_project_role
+        if compute_project_role is not None:
+            self.compute_project_role = compute_project_role
         if credentials_secret_id is not None:
             self.credentials_secret_id = credentials_secret_id
         if credentials_service_account_email is not None:
@@ -111,6 +121,48 @@ class V1GoogleCloudDirectV1(object):
 
         self._bucket_name = bucket_name
 
+    @property
+    def compute_list_bucket_project_role(self) -> 'str':
+        """Gets the compute_list_bucket_project_role of this V1GoogleCloudDirectV1.  # noqa: E501
+
+
+        :return: The compute_list_bucket_project_role of this V1GoogleCloudDirectV1.  # noqa: E501
+        :rtype: str
+        """
+        return self._compute_list_bucket_project_role
+
+    @compute_list_bucket_project_role.setter
+    def compute_list_bucket_project_role(self, compute_list_bucket_project_role: 'str'):
+        """Sets the compute_list_bucket_project_role of this V1GoogleCloudDirectV1.
+
+
+        :param compute_list_bucket_project_role: The compute_list_bucket_project_role of this V1GoogleCloudDirectV1.  # noqa: E501
+        :type: str
+        """
+
+        self._compute_list_bucket_project_role = compute_list_bucket_project_role
+
+    @property
+    def compute_project_role(self) -> 'str':
+        """Gets the compute_project_role of this V1GoogleCloudDirectV1.  # noqa: E501
+
+
+        :return: The compute_project_role of this V1GoogleCloudDirectV1.  # noqa: E501
+        :rtype: str
+        """
+        return self._compute_project_role
+
+    @compute_project_role.setter
+    def compute_project_role(self, compute_project_role: 'str'):
+        """Sets the compute_project_role of this V1GoogleCloudDirectV1.
+
+
+        :param compute_project_role: The compute_project_role of this V1GoogleCloudDirectV1.  # noqa: E501
+        :type: str
+        """
+
+        self._compute_project_role = compute_project_role
+
     @property
     def credentials_secret_id(self) -> 'str':
         """Gets the credentials_secret_id of this V1GoogleCloudDirectV1.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py

@@ -41,6 +41,8 @@ class V1JobSpec(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'artifacts_destination': 'str',
+        'artifacts_source': 'str',
         'cloudspace_id': 'str',
         'cluster_id': 'str',
         'command': 'str',
@@ -58,13 +60,13 @@ class V1JobSpec(object):
         'resources': 'V1Resources',
         'restart_policy': 'str',
         'run_id': 'str',
-        'skip_data_connections_setup': 'bool',
-        'skip_filesystem_setup': 'bool',
         'spot': 'bool',
         'volumes': 'list[V1Volume]'
     }
 
     attribute_map = {
+        'artifacts_destination': 'artifactsDestination',
+        'artifacts_source': 'artifactsSource',
         'cloudspace_id': 'cloudspaceId',
         'cluster_id': 'clusterId',
         'command': 'command',
@@ -82,14 +84,14 @@ class V1JobSpec(object):
         'resources': 'resources',
         'restart_policy': 'restartPolicy',
         'run_id': 'runId',
-        'skip_data_connections_setup': 'skipDataConnectionsSetup',
-        'skip_filesystem_setup': 'skipFilesystemSetup',
         'spot': 'spot',
         'volumes': 'volumes'
     }
 
-    def __init__(self, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, data_paths: 'list[V1DataPath]' =None, entrypoint: 'str' =None, env: 'list[V1EnvVar]' =None, image: 'str' =None, image_cluster_credentials: 'bool' =None, image_secret_ref: 'str' =None, include_credentials: 'bool' =None, instance_name: 'str' =None, instance_type: 'str' =None, modified_volume: 'bool' =None, readiness_probe: 'V1JobHealthCheckConfig' =None, resources: 'V1Resources' =None, restart_policy: 'str' =None, run_id: 'str' =None, skip_data_connections_setup: 'bool' =None, skip_filesystem_setup: 'bool' =None, spot: 'bool' =None, volumes: 'list[V1Volume]' =None):  # noqa: E501
+    def __init__(self, artifacts_destination: 'str' =None, artifacts_source: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, data_paths: 'list[V1DataPath]' =None, entrypoint: 'str' =None, env: 'list[V1EnvVar]' =None, image: 'str' =None, image_cluster_credentials: 'bool' =None, image_secret_ref: 'str' =None, include_credentials: 'bool' =None, instance_name: 'str' =None, instance_type: 'str' =None, modified_volume: 'bool' =None, readiness_probe: 'V1JobHealthCheckConfig' =None, resources: 'V1Resources' =None, restart_policy: 'str' =None, run_id: 'str' =None, spot: 'bool' =None, volumes: 'list[V1Volume]' =None):  # noqa: E501
         """V1JobSpec - a model defined in Swagger"""  # noqa: E501
+        self._artifacts_destination = None
+        self._artifacts_source = None
         self._cloudspace_id = None
         self._cluster_id = None
         self._command = None
@@ -107,11 +109,13 @@ class V1JobSpec(object):
         self._resources = None
         self._restart_policy = None
         self._run_id = None
-        self._skip_data_connections_setup = None
-        self._skip_filesystem_setup = None
         self._spot = None
         self._volumes = None
         self.discriminator = None
+        if artifacts_destination is not None:
+            self.artifacts_destination = artifacts_destination
+        if artifacts_source is not None:
+            self.artifacts_source = artifacts_source
         if cloudspace_id is not None:
             self.cloudspace_id = cloudspace_id
         if cluster_id is not None:
@@ -146,15 +150,53 @@ class V1JobSpec(object):
             self.restart_policy = restart_policy
         if run_id is not None:
             self.run_id = run_id
-        if skip_data_connections_setup is not None:
-            self.skip_data_connections_setup = skip_data_connections_setup
-        if skip_filesystem_setup is not None:
-            self.skip_filesystem_setup = skip_filesystem_setup
         if spot is not None:
             self.spot = spot
         if volumes is not None:
             self.volumes = volumes
 
+    @property
+    def artifacts_destination(self) -> 'str':
+        """Gets the artifacts_destination of this V1JobSpec.  # noqa: E501
+
+
+        :return: The artifacts_destination of this V1JobSpec.  # noqa: E501
+        :rtype: str
+        """
+        return self._artifacts_destination
+
+    @artifacts_destination.setter
+    def artifacts_destination(self, artifacts_destination: 'str'):
+        """Sets the artifacts_destination of this V1JobSpec.
+
+
+        :param artifacts_destination: The artifacts_destination of this V1JobSpec.  # noqa: E501
+        :type: str
+        """
+
+        self._artifacts_destination = artifacts_destination
+
+    @property
+    def artifacts_source(self) -> 'str':
+        """Gets the artifacts_source of this V1JobSpec.  # noqa: E501
+
+
+        :return: The artifacts_source of this V1JobSpec.  # noqa: E501
+        :rtype: str
+        """
+        return self._artifacts_source
+
+    @artifacts_source.setter
+    def artifacts_source(self, artifacts_source: 'str'):
+        """Sets the artifacts_source of this V1JobSpec.
+
+
+        :param artifacts_source: The artifacts_source of this V1JobSpec.  # noqa: E501
+        :type: str
+        """
+
+        self._artifacts_source = artifacts_source
+
     @property
     def cloudspace_id(self) -> 'str':
         """Gets the cloudspace_id of this V1JobSpec.  # noqa: E501
@@ -512,48 +554,6 @@ class V1JobSpec(object):
 
         self._run_id = run_id
 
-    @property
-    def skip_data_connections_setup(self) -> 'bool':
-        """Gets the skip_data_connections_setup of this V1JobSpec.  # noqa: E501
-
-
-        :return: The skip_data_connections_setup of this V1JobSpec.  # noqa: E501
-        :rtype: bool
-        """
-        return self._skip_data_connections_setup
-
-    @skip_data_connections_setup.setter
-    def skip_data_connections_setup(self, skip_data_connections_setup: 'bool'):
-        """Sets the skip_data_connections_setup of this V1JobSpec.
-
-
-        :param skip_data_connections_setup: The skip_data_connections_setup of this V1JobSpec.  # noqa: E501
-        :type: bool
-        """
-
-        self._skip_data_connections_setup = skip_data_connections_setup
-
-    @property
-    def skip_filesystem_setup(self) -> 'bool':
-        """Gets the skip_filesystem_setup of this V1JobSpec.  # noqa: E501
-
-
-        :return: The skip_filesystem_setup of this V1JobSpec.  # noqa: E501
-        :rtype: bool
-        """
-        return self._skip_filesystem_setup
-
-    @skip_filesystem_setup.setter
-    def skip_filesystem_setup(self, skip_filesystem_setup: 'bool'):
-        """Sets the skip_filesystem_setup of this V1JobSpec.
-
-
-        :param skip_filesystem_setup: The skip_filesystem_setup of this V1JobSpec.  # noqa: E501
-        :type: bool
-        """
-
-        self._skip_filesystem_setup = skip_filesystem_setup
-
     @property
     def spot(self) -> 'bool':
         """Gets the spot of this V1JobSpec.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_lambda_labs_direct_v1.py

@@ -0,0 +1,125 @@
+# 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 V1LambdaLabsDirectV1(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 = {
+        'credentials_secret_id': 'str'
+    }
+
+    attribute_map = {
+        'credentials_secret_id': 'credentialsSecretId'
+    }
+
+    def __init__(self, credentials_secret_id: 'str' =None):  # noqa: E501
+        """V1LambdaLabsDirectV1 - a model defined in Swagger"""  # noqa: E501
+        self._credentials_secret_id = None
+        self.discriminator = None
+        if credentials_secret_id is not None:
+            self.credentials_secret_id = credentials_secret_id
+
+    @property
+    def credentials_secret_id(self) -> 'str':
+        """Gets the credentials_secret_id of this V1LambdaLabsDirectV1.  # noqa: E501
+
+        ID of the secret that contains the LambdaLabs API key.  # noqa: E501
+
+        :return: The credentials_secret_id of this V1LambdaLabsDirectV1.  # noqa: E501
+        :rtype: str
+        """
+        return self._credentials_secret_id
+
+    @credentials_secret_id.setter
+    def credentials_secret_id(self, credentials_secret_id: 'str'):
+        """Sets the credentials_secret_id of this V1LambdaLabsDirectV1.
+
+        ID of the secret that contains the LambdaLabs API key.  # noqa: E501
+
+        :param credentials_secret_id: The credentials_secret_id of this V1LambdaLabsDirectV1.  # noqa: E501
+        :type: str
+        """
+
+        self._credentials_secret_id = credentials_secret_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(V1LambdaLabsDirectV1, 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: 'V1LambdaLabsDirectV1') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1LambdaLabsDirectV1):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1LambdaLabsDirectV1') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other