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

lightning_sdk/lightning_cloud/openapi/models/v1_slurm_job.py

@@ -41,15 +41,20 @@ class V1SLURMJob(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'cache_id': 'str',
         'cloudspace_id': 'str',
         'cluster_id': 'str',
         'command': 'str',
         'created_at': 'datetime',
         'id': 'str',
+        'linked_project_id': 'str',
+        'linked_user_id': 'str',
         'message': 'str',
         'name': 'str',
+        'num_gpus': 'str',
         'project_id': 'str',
         'run_id': 'str',
+        'service_id': 'str',
         'slurm_v1_status': 'V1SlurmV1JobStatus',
         'state': 'str',
         'updated_at': 'datetime',
@@ -61,15 +66,20 @@ class V1SLURMJob(object):
     }
 
     attribute_map = {
+        'cache_id': 'cacheId',
         'cloudspace_id': 'cloudspaceId',
         'cluster_id': 'clusterId',
         'command': 'command',
         'created_at': 'createdAt',
         'id': 'id',
+        'linked_project_id': 'linkedProjectId',
+        'linked_user_id': 'linkedUserId',
         'message': 'message',
         'name': 'name',
+        'num_gpus': 'numGpus',
         'project_id': 'projectId',
         'run_id': 'runId',
+        'service_id': 'serviceId',
         'slurm_v1_status': 'slurmV1Status',
         'state': 'state',
         'updated_at': 'updatedAt',
@@ -80,17 +90,22 @@ class V1SLURMJob(object):
         'work_dir': 'workDir'
     }
 
-    def __init__(self, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, created_at: 'datetime' =None, id: 'str' =None, message: 'str' =None, name: 'str' =None, project_id: 'str' =None, run_id: 'str' =None, slurm_v1_status: 'V1SlurmV1JobStatus' =None, state: 'str' =None, updated_at: 'datetime' =None, upload_eta_seconds: 'str' =None, upload_in_progress: 'bool' =None, upload_percentage: 'str' =None, user_id: 'str' =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, created_at: 'datetime' =None, id: 'str' =None, linked_project_id: 'str' =None, linked_user_id: 'str' =None, message: 'str' =None, name: 'str' =None, num_gpus: 'str' =None, project_id: 'str' =None, run_id: 'str' =None, service_id: 'str' =None, slurm_v1_status: 'V1SlurmV1JobStatus' =None, state: 'str' =None, updated_at: 'datetime' =None, upload_eta_seconds: 'str' =None, upload_in_progress: 'bool' =None, upload_percentage: 'str' =None, user_id: 'str' =None, work_dir: 'str' =None):  # noqa: E501
         """V1SLURMJob - a model defined in Swagger"""  # noqa: E501
+        self._cache_id = None
         self._cloudspace_id = None
         self._cluster_id = None
         self._command = None
         self._created_at = None
         self._id = None
+        self._linked_project_id = None
+        self._linked_user_id = None
         self._message = None
         self._name = None
+        self._num_gpus = None
         self._project_id = None
         self._run_id = None
+        self._service_id = None
         self._slurm_v1_status = None
         self._state = None
         self._updated_at = None
@@ -100,6 +115,8 @@ class V1SLURMJob(object):
         self._user_id = None
         self._work_dir = None
         self.discriminator = None
+        if cache_id is not None:
+            self.cache_id = cache_id
         if cloudspace_id is not None:
             self.cloudspace_id = cloudspace_id
         if cluster_id is not None:
@@ -110,14 +127,22 @@ class V1SLURMJob(object):
             self.created_at = created_at
         if id is not None:
             self.id = id
+        if linked_project_id is not None:
+            self.linked_project_id = linked_project_id
+        if linked_user_id is not None:
+            self.linked_user_id = linked_user_id
         if message is not None:
             self.message = message
         if name is not None:
             self.name = name
+        if num_gpus is not None:
+            self.num_gpus = num_gpus
         if project_id is not None:
             self.project_id = project_id
         if run_id is not None:
             self.run_id = run_id
+        if service_id is not None:
+            self.service_id = service_id
         if slurm_v1_status is not None:
             self.slurm_v1_status = slurm_v1_status
         if state is not None:
@@ -135,6 +160,27 @@ class V1SLURMJob(object):
         if work_dir is not None:
             self.work_dir = work_dir
 
+    @property
+    def cache_id(self) -> 'str':
+        """Gets the cache_id of this V1SLURMJob.  # noqa: E501
+
+
+        :return: The cache_id of this V1SLURMJob.  # noqa: E501
+        :rtype: str
+        """
+        return self._cache_id
+
+    @cache_id.setter
+    def cache_id(self, cache_id: 'str'):
+        """Sets the cache_id of this V1SLURMJob.
+
+
+        :param cache_id: The cache_id of this V1SLURMJob.  # noqa: E501
+        :type: str
+        """
+
+        self._cache_id = cache_id
+
     @property
     def cloudspace_id(self) -> 'str':
         """Gets the cloudspace_id of this V1SLURMJob.  # noqa: E501
@@ -240,6 +286,48 @@ class V1SLURMJob(object):
 
         self._id = id
 
+    @property
+    def linked_project_id(self) -> 'str':
+        """Gets the linked_project_id of this V1SLURMJob.  # noqa: E501
+
+
+        :return: The linked_project_id of this V1SLURMJob.  # noqa: E501
+        :rtype: str
+        """
+        return self._linked_project_id
+
+    @linked_project_id.setter
+    def linked_project_id(self, linked_project_id: 'str'):
+        """Sets the linked_project_id of this V1SLURMJob.
+
+
+        :param linked_project_id: The linked_project_id of this V1SLURMJob.  # noqa: E501
+        :type: str
+        """
+
+        self._linked_project_id = linked_project_id
+
+    @property
+    def linked_user_id(self) -> 'str':
+        """Gets the linked_user_id of this V1SLURMJob.  # noqa: E501
+
+
+        :return: The linked_user_id of this V1SLURMJob.  # noqa: E501
+        :rtype: str
+        """
+        return self._linked_user_id
+
+    @linked_user_id.setter
+    def linked_user_id(self, linked_user_id: 'str'):
+        """Sets the linked_user_id of this V1SLURMJob.
+
+
+        :param linked_user_id: The linked_user_id of this V1SLURMJob.  # noqa: E501
+        :type: str
+        """
+
+        self._linked_user_id = linked_user_id
+
     @property
     def message(self) -> 'str':
         """Gets the message of this V1SLURMJob.  # noqa: E501
@@ -282,6 +370,27 @@ class V1SLURMJob(object):
 
         self._name = name
 
+    @property
+    def num_gpus(self) -> 'str':
+        """Gets the num_gpus of this V1SLURMJob.  # noqa: E501
+
+
+        :return: The num_gpus of this V1SLURMJob.  # noqa: E501
+        :rtype: str
+        """
+        return self._num_gpus
+
+    @num_gpus.setter
+    def num_gpus(self, num_gpus: 'str'):
+        """Sets the num_gpus of this V1SLURMJob.
+
+
+        :param num_gpus: The num_gpus of this V1SLURMJob.  # noqa: E501
+        :type: str
+        """
+
+        self._num_gpus = num_gpus
+
     @property
     def project_id(self) -> 'str':
         """Gets the project_id of this V1SLURMJob.  # noqa: E501
@@ -324,6 +433,27 @@ class V1SLURMJob(object):
 
         self._run_id = run_id
 
+    @property
+    def service_id(self) -> 'str':
+        """Gets the service_id of this V1SLURMJob.  # noqa: E501
+
+
+        :return: The service_id of this V1SLURMJob.  # noqa: E501
+        :rtype: str
+        """
+        return self._service_id
+
+    @service_id.setter
+    def service_id(self, service_id: 'str'):
+        """Sets the service_id of this V1SLURMJob.
+
+
+        :param service_id: The service_id of this V1SLURMJob.  # noqa: E501
+        :type: str
+        """
+
+        self._service_id = service_id
+
     @property
     def slurm_v1_status(self) -> 'V1SlurmV1JobStatus':
         """Gets the slurm_v1_status of this V1SLURMJob.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_slurm_node.py

@@ -48,6 +48,8 @@ class V1SlurmNode(object):
         'cores': 'int',
         'cpus': 'int',
         'free_memory': 'str',
+        'gpu_type': 'str',
+        'gpus': 'int',
         'hostname': 'str',
         'idle_cpus': 'int',
         'name': 'str',
@@ -64,6 +66,8 @@ class V1SlurmNode(object):
         'cores': 'cores',
         'cpus': 'cpus',
         'free_memory': 'freeMemory',
+        'gpu_type': 'gpuType',
+        'gpus': 'gpus',
         'hostname': 'hostname',
         'idle_cpus': 'idleCpus',
         'name': 'name',
@@ -72,7 +76,7 @@ class V1SlurmNode(object):
         'state': 'state'
     }
 
-    def __init__(self, address: 'str' =None, alloc_cpus: 'int' =None, alloc_memory: 'str' =None, architecture: 'str' =None, cores: 'int' =None, cpus: 'int' =None, free_memory: 'str' =None, hostname: 'str' =None, idle_cpus: 'int' =None, name: 'str' =None, operating_system: 'str' =None, slurmd_version: 'str' =None, state: 'str' =None):  # noqa: E501
+    def __init__(self, address: 'str' =None, alloc_cpus: 'int' =None, alloc_memory: 'str' =None, architecture: 'str' =None, cores: 'int' =None, cpus: 'int' =None, free_memory: 'str' =None, gpu_type: 'str' =None, gpus: 'int' =None, hostname: 'str' =None, idle_cpus: 'int' =None, name: 'str' =None, operating_system: 'str' =None, slurmd_version: 'str' =None, state: 'str' =None):  # noqa: E501
         """V1SlurmNode - a model defined in Swagger"""  # noqa: E501
         self._address = None
         self._alloc_cpus = None
@@ -81,6 +85,8 @@ class V1SlurmNode(object):
         self._cores = None
         self._cpus = None
         self._free_memory = None
+        self._gpu_type = None
+        self._gpus = None
         self._hostname = None
         self._idle_cpus = None
         self._name = None
@@ -102,6 +108,10 @@ class V1SlurmNode(object):
             self.cpus = cpus
         if free_memory is not None:
             self.free_memory = free_memory
+        if gpu_type is not None:
+            self.gpu_type = gpu_type
+        if gpus is not None:
+            self.gpus = gpus
         if hostname is not None:
             self.hostname = hostname
         if idle_cpus is not None:
@@ -262,6 +272,48 @@ class V1SlurmNode(object):
 
         self._free_memory = free_memory
 
+    @property
+    def gpu_type(self) -> 'str':
+        """Gets the gpu_type of this V1SlurmNode.  # noqa: E501
+
+
+        :return: The gpu_type of this V1SlurmNode.  # noqa: E501
+        :rtype: str
+        """
+        return self._gpu_type
+
+    @gpu_type.setter
+    def gpu_type(self, gpu_type: 'str'):
+        """Sets the gpu_type of this V1SlurmNode.
+
+
+        :param gpu_type: The gpu_type of this V1SlurmNode.  # noqa: E501
+        :type: str
+        """
+
+        self._gpu_type = gpu_type
+
+    @property
+    def gpus(self) -> 'int':
+        """Gets the gpus of this V1SlurmNode.  # noqa: E501
+
+
+        :return: The gpus of this V1SlurmNode.  # noqa: E501
+        :rtype: int
+        """
+        return self._gpus
+
+    @gpus.setter
+    def gpus(self, gpus: 'int'):
+        """Sets the gpus of this V1SlurmNode.
+
+
+        :param gpus: The gpus of this V1SlurmNode.  # noqa: E501
+        :type: int
+        """
+
+        self._gpus = gpus
+
     @property
     def hostname(self) -> 'str':
         """Gets the hostname of this V1SlurmNode.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_slurm_v1.py

@@ -41,14 +41,40 @@ class V1SlurmV1(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'work_dir': 'str'
     }
 
     attribute_map = {
+        'work_dir': 'workDir'
     }
 
-    def __init__(self):  # noqa: E501
+    def __init__(self, work_dir: 'str' =None):  # noqa: E501
         """V1SlurmV1 - a model defined in Swagger"""  # noqa: E501
+        self._work_dir = None
         self.discriminator = None
+        if work_dir is not None:
+            self.work_dir = work_dir
+
+    @property
+    def work_dir(self) -> 'str':
+        """Gets the work_dir of this V1SlurmV1.  # noqa: E501
+
+
+        :return: The work_dir of this V1SlurmV1.  # noqa: E501
+        :rtype: str
+        """
+        return self._work_dir
+
+    @work_dir.setter
+    def work_dir(self, work_dir: 'str'):
+        """Sets the work_dir of this V1SlurmV1.
+
+
+        :param work_dir: The work_dir of this V1SlurmV1.  # noqa: E501
+        :type: str
+        """
+
+        self._work_dir = work_dir
 
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""

lightning_sdk/lightning_cloud/openapi/models/v1_snowflake_data_connection.py

@@ -0,0 +1,253 @@
+# 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 V1SnowflakeDataConnection(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 = {
+        'bucket_name': 'str',
+        'compress': 'bool',
+        'credential_id': 'str',
+        'format': 'str',
+        'include_header': 'bool',
+        'query': 'str'
+    }
+
+    attribute_map = {
+        'bucket_name': 'bucketName',
+        'compress': 'compress',
+        'credential_id': 'credentialId',
+        'format': 'format',
+        'include_header': 'includeHeader',
+        'query': 'query'
+    }
+
+    def __init__(self, bucket_name: 'str' =None, compress: 'bool' =None, credential_id: 'str' =None, format: 'str' =None, include_header: 'bool' =None, query: 'str' =None):  # noqa: E501
+        """V1SnowflakeDataConnection - a model defined in Swagger"""  # noqa: E501
+        self._bucket_name = None
+        self._compress = None
+        self._credential_id = None
+        self._format = None
+        self._include_header = None
+        self._query = None
+        self.discriminator = None
+        if bucket_name is not None:
+            self.bucket_name = bucket_name
+        if compress is not None:
+            self.compress = compress
+        if credential_id is not None:
+            self.credential_id = credential_id
+        if format is not None:
+            self.format = format
+        if include_header is not None:
+            self.include_header = include_header
+        if query is not None:
+            self.query = query
+
+    @property
+    def bucket_name(self) -> 'str':
+        """Gets the bucket_name of this V1SnowflakeDataConnection.  # noqa: E501
+
+
+        :return: The bucket_name of this V1SnowflakeDataConnection.  # noqa: E501
+        :rtype: str
+        """
+        return self._bucket_name
+
+    @bucket_name.setter
+    def bucket_name(self, bucket_name: 'str'):
+        """Sets the bucket_name of this V1SnowflakeDataConnection.
+
+
+        :param bucket_name: The bucket_name of this V1SnowflakeDataConnection.  # noqa: E501
+        :type: str
+        """
+
+        self._bucket_name = bucket_name
+
+    @property
+    def compress(self) -> 'bool':
+        """Gets the compress of this V1SnowflakeDataConnection.  # noqa: E501
+
+
+        :return: The compress of this V1SnowflakeDataConnection.  # noqa: E501
+        :rtype: bool
+        """
+        return self._compress
+
+    @compress.setter
+    def compress(self, compress: 'bool'):
+        """Sets the compress of this V1SnowflakeDataConnection.
+
+
+        :param compress: The compress of this V1SnowflakeDataConnection.  # noqa: E501
+        :type: bool
+        """
+
+        self._compress = compress
+
+    @property
+    def credential_id(self) -> 'str':
+        """Gets the credential_id of this V1SnowflakeDataConnection.  # noqa: E501
+
+
+        :return: The credential_id of this V1SnowflakeDataConnection.  # noqa: E501
+        :rtype: str
+        """
+        return self._credential_id
+
+    @credential_id.setter
+    def credential_id(self, credential_id: 'str'):
+        """Sets the credential_id of this V1SnowflakeDataConnection.
+
+
+        :param credential_id: The credential_id of this V1SnowflakeDataConnection.  # noqa: E501
+        :type: str
+        """
+
+        self._credential_id = credential_id
+
+    @property
+    def format(self) -> 'str':
+        """Gets the format of this V1SnowflakeDataConnection.  # noqa: E501
+
+
+        :return: The format of this V1SnowflakeDataConnection.  # noqa: E501
+        :rtype: str
+        """
+        return self._format
+
+    @format.setter
+    def format(self, format: 'str'):
+        """Sets the format of this V1SnowflakeDataConnection.
+
+
+        :param format: The format of this V1SnowflakeDataConnection.  # noqa: E501
+        :type: str
+        """
+
+        self._format = format
+
+    @property
+    def include_header(self) -> 'bool':
+        """Gets the include_header of this V1SnowflakeDataConnection.  # noqa: E501
+
+
+        :return: The include_header of this V1SnowflakeDataConnection.  # noqa: E501
+        :rtype: bool
+        """
+        return self._include_header
+
+    @include_header.setter
+    def include_header(self, include_header: 'bool'):
+        """Sets the include_header of this V1SnowflakeDataConnection.
+
+
+        :param include_header: The include_header of this V1SnowflakeDataConnection.  # noqa: E501
+        :type: bool
+        """
+
+        self._include_header = include_header
+
+    @property
+    def query(self) -> 'str':
+        """Gets the query of this V1SnowflakeDataConnection.  # noqa: E501
+
+
+        :return: The query of this V1SnowflakeDataConnection.  # noqa: E501
+        :rtype: str
+        """
+        return self._query
+
+    @query.setter
+    def query(self, query: 'str'):
+        """Sets the query of this V1SnowflakeDataConnection.
+
+
+        :param query: The query of this V1SnowflakeDataConnection.  # noqa: E501
+        :type: str
+        """
+
+        self._query = query
+
+    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(V1SnowflakeDataConnection, 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: 'V1SnowflakeDataConnection') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1SnowflakeDataConnection):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1SnowflakeDataConnection') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other