lightning-sdk 0.1.40__py3-none-any.whl → 0.1.42__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_metrics_stream.py

@@ -66,6 +66,7 @@ class V1MetricsStream(object):
         'user_id': 'str',
         'version': 'datetime',
         'version_number': 'int',
+        'visibility': 'V1ResourceVisibility',
         'work_id': 'str'
     }
 
@@ -95,10 +96,11 @@ class V1MetricsStream(object):
         'user_id': 'userId',
         'version': 'version',
         'version_number': 'versionNumber',
+        'visibility': 'visibility',
         'work_id': 'workId'
     }
 
-    def __init__(self, app_id: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, dark_color: 'str' =None, id: 'str' =None, job_name: 'str' =None, light_color: 'str' =None, name: 'str' =None, persisted: 'bool' =None, phase: 'V1PhaseType' =None, plugin_id: 'str' =None, project_id: 'str' =None, source_id: 'str' =None, stopped_at: 'datetime' =None, store_created_at: 'bool' =None, store_step: 'bool' =None, system_info: 'V1SystemInfo' =None, tags: 'list[V1MetricsTags]' =None, trackers: 'dict(str, V1MetricsTracker)' =None, updated_at: 'datetime' =None, updates_counter: 'int' =None, user_id: 'str' =None, version: 'datetime' =None, version_number: 'int' =None, work_id: 'str' =None):  # noqa: E501
+    def __init__(self, app_id: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, dark_color: 'str' =None, id: 'str' =None, job_name: 'str' =None, light_color: 'str' =None, name: 'str' =None, persisted: 'bool' =None, phase: 'V1PhaseType' =None, plugin_id: 'str' =None, project_id: 'str' =None, source_id: 'str' =None, stopped_at: 'datetime' =None, store_created_at: 'bool' =None, store_step: 'bool' =None, system_info: 'V1SystemInfo' =None, tags: 'list[V1MetricsTags]' =None, trackers: 'dict(str, V1MetricsTracker)' =None, updated_at: 'datetime' =None, updates_counter: 'int' =None, user_id: 'str' =None, version: 'datetime' =None, version_number: 'int' =None, visibility: 'V1ResourceVisibility' =None, work_id: 'str' =None):  # noqa: E501
         """V1MetricsStream - a model defined in Swagger"""  # noqa: E501
         self._app_id = None
         self._cloudspace_id = None
@@ -125,6 +127,7 @@ class V1MetricsStream(object):
         self._user_id = None
         self._version = None
         self._version_number = None
+        self._visibility = None
         self._work_id = None
         self.discriminator = None
         if app_id is not None:
@@ -177,6 +180,8 @@ class V1MetricsStream(object):
             self.version = version
         if version_number is not None:
             self.version_number = version_number
+        if visibility is not None:
+            self.visibility = visibility
         if work_id is not None:
             self.work_id = work_id
 
@@ -705,6 +710,27 @@ class V1MetricsStream(object):
 
         self._version_number = version_number
 
+    @property
+    def visibility(self) -> 'V1ResourceVisibility':
+        """Gets the visibility of this V1MetricsStream.  # noqa: E501
+
+
+        :return: The visibility of this V1MetricsStream.  # noqa: E501
+        :rtype: V1ResourceVisibility
+        """
+        return self._visibility
+
+    @visibility.setter
+    def visibility(self, visibility: 'V1ResourceVisibility'):
+        """Sets the visibility of this V1MetricsStream.
+
+
+        :param visibility: The visibility of this V1MetricsStream.  # noqa: E501
+        :type: V1ResourceVisibility
+        """
+
+        self._visibility = visibility
+
     @property
     def work_id(self) -> 'str':
         """Gets the work_id of this V1MetricsStream.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_model_file.py

@@ -0,0 +1,175 @@
+# 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 V1ModelFile(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 = {
+        'filepath': 'str',
+        'size': 'str',
+        'url': 'str'
+    }
+
+    attribute_map = {
+        'filepath': 'filepath',
+        'size': 'size',
+        'url': 'url'
+    }
+
+    def __init__(self, filepath: 'str' =None, size: 'str' =None, url: 'str' =None):  # noqa: E501
+        """V1ModelFile - a model defined in Swagger"""  # noqa: E501
+        self._filepath = None
+        self._size = None
+        self._url = None
+        self.discriminator = None
+        if filepath is not None:
+            self.filepath = filepath
+        if size is not None:
+            self.size = size
+        if url is not None:
+            self.url = url
+
+    @property
+    def filepath(self) -> 'str':
+        """Gets the filepath of this V1ModelFile.  # noqa: E501
+
+
+        :return: The filepath of this V1ModelFile.  # noqa: E501
+        :rtype: str
+        """
+        return self._filepath
+
+    @filepath.setter
+    def filepath(self, filepath: 'str'):
+        """Sets the filepath of this V1ModelFile.
+
+
+        :param filepath: The filepath of this V1ModelFile.  # noqa: E501
+        :type: str
+        """
+
+        self._filepath = filepath
+
+    @property
+    def size(self) -> 'str':
+        """Gets the size of this V1ModelFile.  # noqa: E501
+
+
+        :return: The size of this V1ModelFile.  # noqa: E501
+        :rtype: str
+        """
+        return self._size
+
+    @size.setter
+    def size(self, size: 'str'):
+        """Sets the size of this V1ModelFile.
+
+
+        :param size: The size of this V1ModelFile.  # noqa: E501
+        :type: str
+        """
+
+        self._size = size
+
+    @property
+    def url(self) -> 'str':
+        """Gets the url of this V1ModelFile.  # noqa: E501
+
+
+        :return: The url of this V1ModelFile.  # noqa: E501
+        :rtype: str
+        """
+        return self._url
+
+    @url.setter
+    def url(self, url: 'str'):
+        """Sets the url of this V1ModelFile.
+
+
+        :param url: The url of this V1ModelFile.  # noqa: E501
+        :type: str
+        """
+
+        self._url = url
+
+    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(V1ModelFile, 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: 'V1ModelFile') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1ModelFile):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1ModelFile') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py

@@ -44,6 +44,7 @@ class V1MultiMachineJob(object):
         'cloudspace_id': 'str',
         'created_at': 'datetime',
         'desired_state': 'V1MultiMachineJobState',
+        'fault_tolerance': 'V1MultiMachineJobFaultTolerance',
         'id': 'str',
         'machines': 'int',
         'message': 'str',
@@ -60,6 +61,7 @@ class V1MultiMachineJob(object):
         'cloudspace_id': 'cloudspaceId',
         'created_at': 'createdAt',
         'desired_state': 'desiredState',
+        'fault_tolerance': 'faultTolerance',
         'id': 'id',
         'machines': 'machines',
         'message': 'message',
@@ -72,11 +74,12 @@ class V1MultiMachineJob(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, cloudspace_id: 'str' =None, created_at: 'datetime' =None, desired_state: 'V1MultiMachineJobState' =None, id: 'str' =None, machines: 'int' =None, message: 'str' =None, name: 'str' =None, project_id: 'str' =None, spec: 'V1JobSpec' =None, state: 'V1MultiMachineJobState' =None, status: 'V1MultiMachineJobStatus' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, cloudspace_id: 'str' =None, created_at: 'datetime' =None, desired_state: 'V1MultiMachineJobState' =None, fault_tolerance: 'V1MultiMachineJobFaultTolerance' =None, id: 'str' =None, machines: 'int' =None, message: 'str' =None, name: 'str' =None, project_id: 'str' =None, spec: 'V1JobSpec' =None, state: 'V1MultiMachineJobState' =None, status: 'V1MultiMachineJobStatus' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """V1MultiMachineJob - a model defined in Swagger"""  # noqa: E501
         self._cloudspace_id = None
         self._created_at = None
         self._desired_state = None
+        self._fault_tolerance = None
         self._id = None
         self._machines = None
         self._message = None
@@ -94,6 +97,8 @@ class V1MultiMachineJob(object):
             self.created_at = created_at
         if desired_state is not None:
             self.desired_state = desired_state
+        if fault_tolerance is not None:
+            self.fault_tolerance = fault_tolerance
         if id is not None:
             self.id = id
         if machines is not None:
@@ -178,6 +183,27 @@ class V1MultiMachineJob(object):
 
         self._desired_state = desired_state
 
+    @property
+    def fault_tolerance(self) -> 'V1MultiMachineJobFaultTolerance':
+        """Gets the fault_tolerance of this V1MultiMachineJob.  # noqa: E501
+
+
+        :return: The fault_tolerance of this V1MultiMachineJob.  # noqa: E501
+        :rtype: V1MultiMachineJobFaultTolerance
+        """
+        return self._fault_tolerance
+
+    @fault_tolerance.setter
+    def fault_tolerance(self, fault_tolerance: 'V1MultiMachineJobFaultTolerance'):
+        """Sets the fault_tolerance of this V1MultiMachineJob.
+
+
+        :param fault_tolerance: The fault_tolerance of this V1MultiMachineJob.  # noqa: E501
+        :type: V1MultiMachineJobFaultTolerance
+        """
+
+        self._fault_tolerance = fault_tolerance
+
     @property
     def id(self) -> 'str':
         """Gets the id of this V1MultiMachineJob.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_event.py

@@ -0,0 +1,331 @@
+# 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 V1MultiMachineJobEvent(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 = {
+        'action': 'str',
+        'id': 'str',
+        'job_id': 'str',
+        'message': 'str',
+        'multi_machine_job_id': 'str',
+        'project_id': 'str',
+        'reason': 'str',
+        'timestamp': 'datetime',
+        'type': 'V1MultiMachineJobEventType'
+    }
+
+    attribute_map = {
+        'action': 'action',
+        'id': 'id',
+        'job_id': 'jobId',
+        'message': 'message',
+        'multi_machine_job_id': 'multiMachineJobId',
+        'project_id': 'projectId',
+        'reason': 'reason',
+        'timestamp': 'timestamp',
+        'type': 'type'
+    }
+
+    def __init__(self, action: 'str' =None, id: 'str' =None, job_id: 'str' =None, message: 'str' =None, multi_machine_job_id: 'str' =None, project_id: 'str' =None, reason: 'str' =None, timestamp: 'datetime' =None, type: 'V1MultiMachineJobEventType' =None):  # noqa: E501
+        """V1MultiMachineJobEvent - a model defined in Swagger"""  # noqa: E501
+        self._action = None
+        self._id = None
+        self._job_id = None
+        self._message = None
+        self._multi_machine_job_id = None
+        self._project_id = None
+        self._reason = None
+        self._timestamp = None
+        self._type = None
+        self.discriminator = None
+        if action is not None:
+            self.action = action
+        if id is not None:
+            self.id = id
+        if job_id is not None:
+            self.job_id = job_id
+        if message is not None:
+            self.message = message
+        if multi_machine_job_id is not None:
+            self.multi_machine_job_id = multi_machine_job_id
+        if project_id is not None:
+            self.project_id = project_id
+        if reason is not None:
+            self.reason = reason
+        if timestamp is not None:
+            self.timestamp = timestamp
+        if type is not None:
+            self.type = type
+
+    @property
+    def action(self) -> 'str':
+        """Gets the action of this V1MultiMachineJobEvent.  # noqa: E501
+
+
+        :return: The action of this V1MultiMachineJobEvent.  # noqa: E501
+        :rtype: str
+        """
+        return self._action
+
+    @action.setter
+    def action(self, action: 'str'):
+        """Sets the action of this V1MultiMachineJobEvent.
+
+
+        :param action: The action of this V1MultiMachineJobEvent.  # noqa: E501
+        :type: str
+        """
+
+        self._action = action
+
+    @property
+    def id(self) -> 'str':
+        """Gets the id of this V1MultiMachineJobEvent.  # noqa: E501
+
+
+        :return: The id of this V1MultiMachineJobEvent.  # noqa: E501
+        :rtype: str
+        """
+        return self._id
+
+    @id.setter
+    def id(self, id: 'str'):
+        """Sets the id of this V1MultiMachineJobEvent.
+
+
+        :param id: The id of this V1MultiMachineJobEvent.  # noqa: E501
+        :type: str
+        """
+
+        self._id = id
+
+    @property
+    def job_id(self) -> 'str':
+        """Gets the job_id of this V1MultiMachineJobEvent.  # noqa: E501
+
+
+        :return: The job_id of this V1MultiMachineJobEvent.  # noqa: E501
+        :rtype: str
+        """
+        return self._job_id
+
+    @job_id.setter
+    def job_id(self, job_id: 'str'):
+        """Sets the job_id of this V1MultiMachineJobEvent.
+
+
+        :param job_id: The job_id of this V1MultiMachineJobEvent.  # noqa: E501
+        :type: str
+        """
+
+        self._job_id = job_id
+
+    @property
+    def message(self) -> 'str':
+        """Gets the message of this V1MultiMachineJobEvent.  # noqa: E501
+
+
+        :return: The message of this V1MultiMachineJobEvent.  # noqa: E501
+        :rtype: str
+        """
+        return self._message
+
+    @message.setter
+    def message(self, message: 'str'):
+        """Sets the message of this V1MultiMachineJobEvent.
+
+
+        :param message: The message of this V1MultiMachineJobEvent.  # noqa: E501
+        :type: str
+        """
+
+        self._message = message
+
+    @property
+    def multi_machine_job_id(self) -> 'str':
+        """Gets the multi_machine_job_id of this V1MultiMachineJobEvent.  # noqa: E501
+
+
+        :return: The multi_machine_job_id of this V1MultiMachineJobEvent.  # noqa: E501
+        :rtype: str
+        """
+        return self._multi_machine_job_id
+
+    @multi_machine_job_id.setter
+    def multi_machine_job_id(self, multi_machine_job_id: 'str'):
+        """Sets the multi_machine_job_id of this V1MultiMachineJobEvent.
+
+
+        :param multi_machine_job_id: The multi_machine_job_id of this V1MultiMachineJobEvent.  # noqa: E501
+        :type: str
+        """
+
+        self._multi_machine_job_id = multi_machine_job_id
+
+    @property
+    def project_id(self) -> 'str':
+        """Gets the project_id of this V1MultiMachineJobEvent.  # noqa: E501
+
+
+        :return: The project_id of this V1MultiMachineJobEvent.  # noqa: E501
+        :rtype: str
+        """
+        return self._project_id
+
+    @project_id.setter
+    def project_id(self, project_id: 'str'):
+        """Sets the project_id of this V1MultiMachineJobEvent.
+
+
+        :param project_id: The project_id of this V1MultiMachineJobEvent.  # noqa: E501
+        :type: str
+        """
+
+        self._project_id = project_id
+
+    @property
+    def reason(self) -> 'str':
+        """Gets the reason of this V1MultiMachineJobEvent.  # noqa: E501
+
+
+        :return: The reason of this V1MultiMachineJobEvent.  # noqa: E501
+        :rtype: str
+        """
+        return self._reason
+
+    @reason.setter
+    def reason(self, reason: 'str'):
+        """Sets the reason of this V1MultiMachineJobEvent.
+
+
+        :param reason: The reason of this V1MultiMachineJobEvent.  # noqa: E501
+        :type: str
+        """
+
+        self._reason = reason
+
+    @property
+    def timestamp(self) -> 'datetime':
+        """Gets the timestamp of this V1MultiMachineJobEvent.  # noqa: E501
+
+
+        :return: The timestamp of this V1MultiMachineJobEvent.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._timestamp
+
+    @timestamp.setter
+    def timestamp(self, timestamp: 'datetime'):
+        """Sets the timestamp of this V1MultiMachineJobEvent.
+
+
+        :param timestamp: The timestamp of this V1MultiMachineJobEvent.  # noqa: E501
+        :type: datetime
+        """
+
+        self._timestamp = timestamp
+
+    @property
+    def type(self) -> 'V1MultiMachineJobEventType':
+        """Gets the type of this V1MultiMachineJobEvent.  # noqa: E501
+
+
+        :return: The type of this V1MultiMachineJobEvent.  # noqa: E501
+        :rtype: V1MultiMachineJobEventType
+        """
+        return self._type
+
+    @type.setter
+    def type(self, type: 'V1MultiMachineJobEventType'):
+        """Sets the type of this V1MultiMachineJobEvent.
+
+
+        :param type: The type of this V1MultiMachineJobEvent.  # noqa: E501
+        :type: V1MultiMachineJobEventType
+        """
+
+        self._type = type
+
+    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(V1MultiMachineJobEvent, 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: 'V1MultiMachineJobEvent') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1MultiMachineJobEvent):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1MultiMachineJobEvent') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other