lightning-sdk 0.1.57__py3-none-any.whl → 0.2.0__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_create_job_request.py

@@ -0,0 +1,201 @@
+# 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 V1CreateJobRequest(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 = {
+        'endpoint': 'V1Endpoint',
+        'name': 'str',
+        'project_id': 'str',
+        'spec': 'V1JobSpec'
+    }
+
+    attribute_map = {
+        'endpoint': 'endpoint',
+        'name': 'name',
+        'project_id': 'projectId',
+        'spec': 'spec'
+    }
+
+    def __init__(self, endpoint: 'V1Endpoint' =None, name: 'str' =None, project_id: 'str' =None, spec: 'V1JobSpec' =None):  # noqa: E501
+        """V1CreateJobRequest - a model defined in Swagger"""  # noqa: E501
+        self._endpoint = None
+        self._name = None
+        self._project_id = None
+        self._spec = None
+        self.discriminator = None
+        if endpoint is not None:
+            self.endpoint = endpoint
+        if name is not None:
+            self.name = name
+        if project_id is not None:
+            self.project_id = project_id
+        if spec is not None:
+            self.spec = spec
+
+    @property
+    def endpoint(self) -> 'V1Endpoint':
+        """Gets the endpoint of this V1CreateJobRequest.  # noqa: E501
+
+
+        :return: The endpoint of this V1CreateJobRequest.  # noqa: E501
+        :rtype: V1Endpoint
+        """
+        return self._endpoint
+
+    @endpoint.setter
+    def endpoint(self, endpoint: 'V1Endpoint'):
+        """Sets the endpoint of this V1CreateJobRequest.
+
+
+        :param endpoint: The endpoint of this V1CreateJobRequest.  # noqa: E501
+        :type: V1Endpoint
+        """
+
+        self._endpoint = endpoint
+
+    @property
+    def name(self) -> 'str':
+        """Gets the name of this V1CreateJobRequest.  # noqa: E501
+
+
+        :return: The name of this V1CreateJobRequest.  # noqa: E501
+        :rtype: str
+        """
+        return self._name
+
+    @name.setter
+    def name(self, name: 'str'):
+        """Sets the name of this V1CreateJobRequest.
+
+
+        :param name: The name of this V1CreateJobRequest.  # noqa: E501
+        :type: str
+        """
+
+        self._name = name
+
+    @property
+    def project_id(self) -> 'str':
+        """Gets the project_id of this V1CreateJobRequest.  # noqa: E501
+
+
+        :return: The project_id of this V1CreateJobRequest.  # noqa: E501
+        :rtype: str
+        """
+        return self._project_id
+
+    @project_id.setter
+    def project_id(self, project_id: 'str'):
+        """Sets the project_id of this V1CreateJobRequest.
+
+
+        :param project_id: The project_id of this V1CreateJobRequest.  # noqa: E501
+        :type: str
+        """
+
+        self._project_id = project_id
+
+    @property
+    def spec(self) -> 'V1JobSpec':
+        """Gets the spec of this V1CreateJobRequest.  # noqa: E501
+
+
+        :return: The spec of this V1CreateJobRequest.  # noqa: E501
+        :rtype: V1JobSpec
+        """
+        return self._spec
+
+    @spec.setter
+    def spec(self, spec: 'V1JobSpec'):
+        """Sets the spec of this V1CreateJobRequest.
+
+
+        :param spec: The spec of this V1CreateJobRequest.  # noqa: E501
+        :type: V1JobSpec
+        """
+
+        self._spec = spec
+
+    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(V1CreateJobRequest, 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: 'V1CreateJobRequest') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1CreateJobRequest):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1CreateJobRequest') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_create_managed_endpoint_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 V1CreateManagedEndpointResponse(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 = {
+        'assistants': 'list[V1Assistant]',
+        'endpoint': 'V1ManagedEndpoint'
+    }
+
+    attribute_map = {
+        'assistants': 'assistants',
+        'endpoint': 'endpoint'
+    }
+
+    def __init__(self, assistants: 'list[V1Assistant]' =None, endpoint: 'V1ManagedEndpoint' =None):  # noqa: E501
+        """V1CreateManagedEndpointResponse - a model defined in Swagger"""  # noqa: E501
+        self._assistants = None
+        self._endpoint = None
+        self.discriminator = None
+        if assistants is not None:
+            self.assistants = assistants
+        if endpoint is not None:
+            self.endpoint = endpoint
+
+    @property
+    def assistants(self) -> 'list[V1Assistant]':
+        """Gets the assistants of this V1CreateManagedEndpointResponse.  # noqa: E501
+
+
+        :return: The assistants of this V1CreateManagedEndpointResponse.  # noqa: E501
+        :rtype: list[V1Assistant]
+        """
+        return self._assistants
+
+    @assistants.setter
+    def assistants(self, assistants: 'list[V1Assistant]'):
+        """Sets the assistants of this V1CreateManagedEndpointResponse.
+
+
+        :param assistants: The assistants of this V1CreateManagedEndpointResponse.  # noqa: E501
+        :type: list[V1Assistant]
+        """
+
+        self._assistants = assistants
+
+    @property
+    def endpoint(self) -> 'V1ManagedEndpoint':
+        """Gets the endpoint of this V1CreateManagedEndpointResponse.  # noqa: E501
+
+
+        :return: The endpoint of this V1CreateManagedEndpointResponse.  # noqa: E501
+        :rtype: V1ManagedEndpoint
+        """
+        return self._endpoint
+
+    @endpoint.setter
+    def endpoint(self, endpoint: 'V1ManagedEndpoint'):
+        """Sets the endpoint of this V1CreateManagedEndpointResponse.
+
+
+        :param endpoint: The endpoint of this V1CreateManagedEndpointResponse.  # noqa: E501
+        :type: V1ManagedEndpoint
+        """
+
+        self._endpoint = endpoint
+
+    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(V1CreateManagedEndpointResponse, 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: 'V1CreateManagedEndpointResponse') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1CreateManagedEndpointResponse):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1CreateManagedEndpointResponse') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_create_multi_machine_job_request.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 V1CreateMultiMachineJobRequest(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',
+        'fault_tolerance': 'V1MultiMachineJobFaultTolerance',
+        'machines': 'int',
+        'name': 'str',
+        'project_id': 'str',
+        'spec': 'V1JobSpec'
+    }
+
+    attribute_map = {
+        'cluster_id': 'clusterId',
+        'fault_tolerance': 'faultTolerance',
+        'machines': 'machines',
+        'name': 'name',
+        'project_id': 'projectId',
+        'spec': 'spec'
+    }
+
+    def __init__(self, cluster_id: 'str' =None, fault_tolerance: 'V1MultiMachineJobFaultTolerance' =None, machines: 'int' =None, name: 'str' =None, project_id: 'str' =None, spec: 'V1JobSpec' =None):  # noqa: E501
+        """V1CreateMultiMachineJobRequest - a model defined in Swagger"""  # noqa: E501
+        self._cluster_id = None
+        self._fault_tolerance = None
+        self._machines = None
+        self._name = None
+        self._project_id = None
+        self._spec = None
+        self.discriminator = None
+        if cluster_id is not None:
+            self.cluster_id = cluster_id
+        if fault_tolerance is not None:
+            self.fault_tolerance = fault_tolerance
+        if machines is not None:
+            self.machines = machines
+        if name is not None:
+            self.name = name
+        if project_id is not None:
+            self.project_id = project_id
+        if spec is not None:
+            self.spec = spec
+
+    @property
+    def cluster_id(self) -> 'str':
+        """Gets the cluster_id of this V1CreateMultiMachineJobRequest.  # noqa: E501
+
+
+        :return: The cluster_id of this V1CreateMultiMachineJobRequest.  # noqa: E501
+        :rtype: str
+        """
+        return self._cluster_id
+
+    @cluster_id.setter
+    def cluster_id(self, cluster_id: 'str'):
+        """Sets the cluster_id of this V1CreateMultiMachineJobRequest.
+
+
+        :param cluster_id: The cluster_id of this V1CreateMultiMachineJobRequest.  # noqa: E501
+        :type: str
+        """
+
+        self._cluster_id = cluster_id
+
+    @property
+    def fault_tolerance(self) -> 'V1MultiMachineJobFaultTolerance':
+        """Gets the fault_tolerance of this V1CreateMultiMachineJobRequest.  # noqa: E501
+
+
+        :return: The fault_tolerance of this V1CreateMultiMachineJobRequest.  # noqa: E501
+        :rtype: V1MultiMachineJobFaultTolerance
+        """
+        return self._fault_tolerance
+
+    @fault_tolerance.setter
+    def fault_tolerance(self, fault_tolerance: 'V1MultiMachineJobFaultTolerance'):
+        """Sets the fault_tolerance of this V1CreateMultiMachineJobRequest.
+
+
+        :param fault_tolerance: The fault_tolerance of this V1CreateMultiMachineJobRequest.  # noqa: E501
+        :type: V1MultiMachineJobFaultTolerance
+        """
+
+        self._fault_tolerance = fault_tolerance
+
+    @property
+    def machines(self) -> 'int':
+        """Gets the machines of this V1CreateMultiMachineJobRequest.  # noqa: E501
+
+
+        :return: The machines of this V1CreateMultiMachineJobRequest.  # noqa: E501
+        :rtype: int
+        """
+        return self._machines
+
+    @machines.setter
+    def machines(self, machines: 'int'):
+        """Sets the machines of this V1CreateMultiMachineJobRequest.
+
+
+        :param machines: The machines of this V1CreateMultiMachineJobRequest.  # noqa: E501
+        :type: int
+        """
+
+        self._machines = machines
+
+    @property
+    def name(self) -> 'str':
+        """Gets the name of this V1CreateMultiMachineJobRequest.  # noqa: E501
+
+
+        :return: The name of this V1CreateMultiMachineJobRequest.  # noqa: E501
+        :rtype: str
+        """
+        return self._name
+
+    @name.setter
+    def name(self, name: 'str'):
+        """Sets the name of this V1CreateMultiMachineJobRequest.
+
+
+        :param name: The name of this V1CreateMultiMachineJobRequest.  # noqa: E501
+        :type: str
+        """
+
+        self._name = name
+
+    @property
+    def project_id(self) -> 'str':
+        """Gets the project_id of this V1CreateMultiMachineJobRequest.  # noqa: E501
+
+
+        :return: The project_id of this V1CreateMultiMachineJobRequest.  # noqa: E501
+        :rtype: str
+        """
+        return self._project_id
+
+    @project_id.setter
+    def project_id(self, project_id: 'str'):
+        """Sets the project_id of this V1CreateMultiMachineJobRequest.
+
+
+        :param project_id: The project_id of this V1CreateMultiMachineJobRequest.  # noqa: E501
+        :type: str
+        """
+
+        self._project_id = project_id
+
+    @property
+    def spec(self) -> 'V1JobSpec':
+        """Gets the spec of this V1CreateMultiMachineJobRequest.  # noqa: E501
+
+
+        :return: The spec of this V1CreateMultiMachineJobRequest.  # noqa: E501
+        :rtype: V1JobSpec
+        """
+        return self._spec
+
+    @spec.setter
+    def spec(self, spec: 'V1JobSpec'):
+        """Sets the spec of this V1CreateMultiMachineJobRequest.
+
+
+        :param spec: The spec of this V1CreateMultiMachineJobRequest.  # noqa: E501
+        :type: V1JobSpec
+        """
+
+        self._spec = spec
+
+    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(V1CreateMultiMachineJobRequest, 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: 'V1CreateMultiMachineJobRequest') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1CreateMultiMachineJobRequest):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1CreateMultiMachineJobRequest') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py

@@ -47,6 +47,7 @@ class V1DataConnection(object):
         'cluster_id': 'str',
         'created_at': 'datetime',
         'efs': 'V1EfsConfig',
+        'filestore': 'V1FilestoreDataConnection',
         'gcp': 'V1GcpDataConnection',
         'gcs_folder': 'V1GCSFolderDataConnection',
         'id': 'str',
@@ -73,6 +74,7 @@ class V1DataConnection(object):
         'cluster_id': 'clusterId',
         'created_at': 'createdAt',
         'efs': 'efs',
+        'filestore': 'filestore',
         'gcp': 'gcp',
         'gcs_folder': 'gcsFolder',
         'id': 'id',
@@ -92,7 +94,7 @@ class V1DataConnection(object):
         'writable': 'writable'
     }
 
-    def __init__(self, access_cluster_ids: 'list[str]' =None, accessible: 'bool' =None, aws: 'V1AwsDataConnection' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, efs: 'V1EfsConfig' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, id: 'str' =None, index: 'V1Index' =None, is_managed: 'bool' =None, lock_out: 'bool' =None, name: 'str' =None, number_of_files: 'str' =None, project_id: 'str' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, state: 'V1DataConnectionState' =None, total_size_bytes: 'str' =None, type: 'str' =None, updated_at: 'datetime' =None, writable: 'bool' =None):  # noqa: E501
+    def __init__(self, access_cluster_ids: 'list[str]' =None, accessible: 'bool' =None, aws: 'V1AwsDataConnection' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, efs: 'V1EfsConfig' =None, filestore: 'V1FilestoreDataConnection' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, id: 'str' =None, index: 'V1Index' =None, is_managed: 'bool' =None, lock_out: 'bool' =None, name: 'str' =None, number_of_files: 'str' =None, project_id: 'str' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, state: 'V1DataConnectionState' =None, total_size_bytes: 'str' =None, type: 'str' =None, updated_at: 'datetime' =None, writable: 'bool' =None):  # noqa: E501
         """V1DataConnection - a model defined in Swagger"""  # noqa: E501
         self._access_cluster_ids = None
         self._accessible = None
@@ -100,6 +102,7 @@ class V1DataConnection(object):
         self._cluster_id = None
         self._created_at = None
         self._efs = None
+        self._filestore = None
         self._gcp = None
         self._gcs_folder = None
         self._id = None
@@ -130,6 +133,8 @@ class V1DataConnection(object):
             self.created_at = created_at
         if efs is not None:
             self.efs = efs
+        if filestore is not None:
+            self.filestore = filestore
         if gcp is not None:
             self.gcp = gcp
         if gcs_folder is not None:
@@ -291,6 +296,27 @@ class V1DataConnection(object):
 
         self._efs = efs
 
+    @property
+    def filestore(self) -> 'V1FilestoreDataConnection':
+        """Gets the filestore of this V1DataConnection.  # noqa: E501
+
+
+        :return: The filestore of this V1DataConnection.  # noqa: E501
+        :rtype: V1FilestoreDataConnection
+        """
+        return self._filestore
+
+    @filestore.setter
+    def filestore(self, filestore: 'V1FilestoreDataConnection'):
+        """Sets the filestore of this V1DataConnection.
+
+
+        :param filestore: The filestore of this V1DataConnection.  # noqa: E501
+        :type: V1FilestoreDataConnection
+        """
+
+        self._filestore = filestore
+
     @property
     def gcp(self) -> 'V1GcpDataConnection':
         """Gets the gcp of this V1DataConnection.  # noqa: E501