lightning-sdk 0.1.36__py3-none-any.whl → 0.1.38__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_status.py

@@ -0,0 +1,279 @@
+# 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 V1MultiMachineJobStatus(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 = {
+        'deleting_machines': 'str',
+        'failing_machines': 'str',
+        'message': 'str',
+        'pending_machines': 'str',
+        'ready_machines': 'str',
+        'started_at': 'datetime',
+        'stopped_at': 'datetime'
+    }
+
+    attribute_map = {
+        'deleting_machines': 'deletingMachines',
+        'failing_machines': 'failingMachines',
+        'message': 'message',
+        'pending_machines': 'pendingMachines',
+        'ready_machines': 'readyMachines',
+        'started_at': 'startedAt',
+        'stopped_at': 'stoppedAt'
+    }
+
+    def __init__(self, deleting_machines: 'str' =None, failing_machines: 'str' =None, message: 'str' =None, pending_machines: 'str' =None, ready_machines: 'str' =None, started_at: 'datetime' =None, stopped_at: 'datetime' =None):  # noqa: E501
+        """V1MultiMachineJobStatus - a model defined in Swagger"""  # noqa: E501
+        self._deleting_machines = None
+        self._failing_machines = None
+        self._message = None
+        self._pending_machines = None
+        self._ready_machines = None
+        self._started_at = None
+        self._stopped_at = None
+        self.discriminator = None
+        if deleting_machines is not None:
+            self.deleting_machines = deleting_machines
+        if failing_machines is not None:
+            self.failing_machines = failing_machines
+        if message is not None:
+            self.message = message
+        if pending_machines is not None:
+            self.pending_machines = pending_machines
+        if ready_machines is not None:
+            self.ready_machines = ready_machines
+        if started_at is not None:
+            self.started_at = started_at
+        if stopped_at is not None:
+            self.stopped_at = stopped_at
+
+    @property
+    def deleting_machines(self) -> 'str':
+        """Gets the deleting_machines of this V1MultiMachineJobStatus.  # noqa: E501
+
+
+        :return: The deleting_machines of this V1MultiMachineJobStatus.  # noqa: E501
+        :rtype: str
+        """
+        return self._deleting_machines
+
+    @deleting_machines.setter
+    def deleting_machines(self, deleting_machines: 'str'):
+        """Sets the deleting_machines of this V1MultiMachineJobStatus.
+
+
+        :param deleting_machines: The deleting_machines of this V1MultiMachineJobStatus.  # noqa: E501
+        :type: str
+        """
+
+        self._deleting_machines = deleting_machines
+
+    @property
+    def failing_machines(self) -> 'str':
+        """Gets the failing_machines of this V1MultiMachineJobStatus.  # noqa: E501
+
+
+        :return: The failing_machines of this V1MultiMachineJobStatus.  # noqa: E501
+        :rtype: str
+        """
+        return self._failing_machines
+
+    @failing_machines.setter
+    def failing_machines(self, failing_machines: 'str'):
+        """Sets the failing_machines of this V1MultiMachineJobStatus.
+
+
+        :param failing_machines: The failing_machines of this V1MultiMachineJobStatus.  # noqa: E501
+        :type: str
+        """
+
+        self._failing_machines = failing_machines
+
+    @property
+    def message(self) -> 'str':
+        """Gets the message of this V1MultiMachineJobStatus.  # noqa: E501
+
+
+        :return: The message of this V1MultiMachineJobStatus.  # noqa: E501
+        :rtype: str
+        """
+        return self._message
+
+    @message.setter
+    def message(self, message: 'str'):
+        """Sets the message of this V1MultiMachineJobStatus.
+
+
+        :param message: The message of this V1MultiMachineJobStatus.  # noqa: E501
+        :type: str
+        """
+
+        self._message = message
+
+    @property
+    def pending_machines(self) -> 'str':
+        """Gets the pending_machines of this V1MultiMachineJobStatus.  # noqa: E501
+
+
+        :return: The pending_machines of this V1MultiMachineJobStatus.  # noqa: E501
+        :rtype: str
+        """
+        return self._pending_machines
+
+    @pending_machines.setter
+    def pending_machines(self, pending_machines: 'str'):
+        """Sets the pending_machines of this V1MultiMachineJobStatus.
+
+
+        :param pending_machines: The pending_machines of this V1MultiMachineJobStatus.  # noqa: E501
+        :type: str
+        """
+
+        self._pending_machines = pending_machines
+
+    @property
+    def ready_machines(self) -> 'str':
+        """Gets the ready_machines of this V1MultiMachineJobStatus.  # noqa: E501
+
+
+        :return: The ready_machines of this V1MultiMachineJobStatus.  # noqa: E501
+        :rtype: str
+        """
+        return self._ready_machines
+
+    @ready_machines.setter
+    def ready_machines(self, ready_machines: 'str'):
+        """Sets the ready_machines of this V1MultiMachineJobStatus.
+
+
+        :param ready_machines: The ready_machines of this V1MultiMachineJobStatus.  # noqa: E501
+        :type: str
+        """
+
+        self._ready_machines = ready_machines
+
+    @property
+    def started_at(self) -> 'datetime':
+        """Gets the started_at of this V1MultiMachineJobStatus.  # noqa: E501
+
+
+        :return: The started_at of this V1MultiMachineJobStatus.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._started_at
+
+    @started_at.setter
+    def started_at(self, started_at: 'datetime'):
+        """Sets the started_at of this V1MultiMachineJobStatus.
+
+
+        :param started_at: The started_at of this V1MultiMachineJobStatus.  # noqa: E501
+        :type: datetime
+        """
+
+        self._started_at = started_at
+
+    @property
+    def stopped_at(self) -> 'datetime':
+        """Gets the stopped_at of this V1MultiMachineJobStatus.  # noqa: E501
+
+
+        :return: The stopped_at of this V1MultiMachineJobStatus.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._stopped_at
+
+    @stopped_at.setter
+    def stopped_at(self, stopped_at: 'datetime'):
+        """Sets the stopped_at of this V1MultiMachineJobStatus.
+
+
+        :param stopped_at: The stopped_at of this V1MultiMachineJobStatus.  # noqa: E501
+        :type: datetime
+        """
+
+        self._stopped_at = stopped_at
+
+    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(V1MultiMachineJobStatus, 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: 'V1MultiMachineJobStatus') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1MultiMachineJobStatus):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1MultiMachineJobStatus') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py

@@ -70,6 +70,8 @@ class V1RuleResource(object):
     MANAGEDENDPOINT = "managedEndpoint"
     JOB = "job"
     DEPLOYMENT = "deployment"
+    MODEL = "model"
+    SLURMJOB = "slurmJob"
     """
     Attributes:
       swagger_types (dict): The key is attribute name

lightning_sdk/lightning_cloud/openapi/models/v1_select.py

@@ -41,19 +41,24 @@ class V1Select(object):
                             and the value is json key in definition.
     """
     swagger_types = {
-        'options': 'list[str]'
+        'options': 'list[str]',
+        'value': 'str'
     }
 
     attribute_map = {
-        'options': 'options'
+        'options': 'options',
+        'value': 'value'
     }
 
-    def __init__(self, options: 'list[str]' =None):  # noqa: E501
+    def __init__(self, options: 'list[str]' =None, value: 'str' =None):  # noqa: E501
         """V1Select - a model defined in Swagger"""  # noqa: E501
         self._options = None
+        self._value = None
         self.discriminator = None
         if options is not None:
             self.options = options
+        if value is not None:
+            self.value = value
 
     @property
     def options(self) -> 'list[str]':
@@ -76,6 +81,27 @@ class V1Select(object):
 
         self._options = options
 
+    @property
+    def value(self) -> 'str':
+        """Gets the value of this V1Select.  # noqa: E501
+
+
+        :return: The value of this V1Select.  # noqa: E501
+        :rtype: str
+        """
+        return self._value
+
+    @value.setter
+    def value(self, value: 'str'):
+        """Sets the value of this V1Select.
+
+
+        :param value: The value of this V1Select.  # noqa: E501
+        :type: str
+        """
+
+        self._value = value
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/v1_system_info.py

@@ -54,6 +54,7 @@ class V1SystemInfo(object):
         'gpu_count': 'int',
         'gpu_memory_gb': 'int',
         'gpu_name': 'str',
+        'hostname': 'str',
         'hyperparameters': 'str',
         'installed_packages': 'str',
         'litlogger_version': 'str',
@@ -76,6 +77,7 @@ class V1SystemInfo(object):
         'gpu_count': 'gpuCount',
         'gpu_memory_gb': 'gpuMemoryGb',
         'gpu_name': 'gpuName',
+        'hostname': 'hostname',
         'hyperparameters': 'hyperparameters',
         'installed_packages': 'installedPackages',
         'litlogger_version': 'litloggerVersion',
@@ -84,7 +86,7 @@ class V1SystemInfo(object):
         'system_memory_gb': 'systemMemoryGb'
     }
 
-    def __init__(self, cli_args: 'str' =None, cpu_count_logical: 'int' =None, cpu_count_physical: 'int' =None, cpu_name: 'str' =None, cuda_version: 'str' =None, cudnn_version: 'str' =None, execution_command: 'str' =None, git_branch: 'str' =None, git_commit_hash: 'str' =None, git_repo_name: 'str' =None, gpu_count: 'int' =None, gpu_memory_gb: 'int' =None, gpu_name: 'str' =None, hyperparameters: 'str' =None, installed_packages: 'str' =None, litlogger_version: 'str' =None, os_name: 'str' =None, python_version: 'str' =None, system_memory_gb: 'int' =None):  # noqa: E501
+    def __init__(self, cli_args: 'str' =None, cpu_count_logical: 'int' =None, cpu_count_physical: 'int' =None, cpu_name: 'str' =None, cuda_version: 'str' =None, cudnn_version: 'str' =None, execution_command: 'str' =None, git_branch: 'str' =None, git_commit_hash: 'str' =None, git_repo_name: 'str' =None, gpu_count: 'int' =None, gpu_memory_gb: 'int' =None, gpu_name: 'str' =None, hostname: 'str' =None, hyperparameters: 'str' =None, installed_packages: 'str' =None, litlogger_version: 'str' =None, os_name: 'str' =None, python_version: 'str' =None, system_memory_gb: 'int' =None):  # noqa: E501
         """V1SystemInfo - a model defined in Swagger"""  # noqa: E501
         self._cli_args = None
         self._cpu_count_logical = None
@@ -99,6 +101,7 @@ class V1SystemInfo(object):
         self._gpu_count = None
         self._gpu_memory_gb = None
         self._gpu_name = None
+        self._hostname = None
         self._hyperparameters = None
         self._installed_packages = None
         self._litlogger_version = None
@@ -132,6 +135,8 @@ class V1SystemInfo(object):
             self.gpu_memory_gb = gpu_memory_gb
         if gpu_name is not None:
             self.gpu_name = gpu_name
+        if hostname is not None:
+            self.hostname = hostname
         if hyperparameters is not None:
             self.hyperparameters = hyperparameters
         if installed_packages is not None:
@@ -418,6 +423,27 @@ class V1SystemInfo(object):
 
         self._gpu_name = gpu_name
 
+    @property
+    def hostname(self) -> 'str':
+        """Gets the hostname of this V1SystemInfo.  # noqa: E501
+
+
+        :return: The hostname of this V1SystemInfo.  # noqa: E501
+        :rtype: str
+        """
+        return self._hostname
+
+    @hostname.setter
+    def hostname(self, hostname: 'str'):
+        """Sets the hostname of this V1SystemInfo.
+
+
+        :param hostname: The hostname of this V1SystemInfo.  # noqa: E501
+        :type: str
+        """
+
+        self._hostname = hostname
+
     @property
     def hyperparameters(self) -> 'str':
         """Gets the hyperparameters of this V1SystemInfo.  # noqa: E501