lightning-sdk 0.1.41__py3-none-any.whl → 0.1.43__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_status.py

@@ -46,6 +46,7 @@ class V1MultiMachineJobStatus(object):
         'message': 'str',
         'pending_machines': 'str',
         'ready_machines': 'str',
+        'retries': 'str',
         'started_at': 'datetime',
         'stopped_at': 'datetime'
     }
@@ -56,17 +57,19 @@ class V1MultiMachineJobStatus(object):
         'message': 'message',
         'pending_machines': 'pendingMachines',
         'ready_machines': 'readyMachines',
+        'retries': 'retries',
         '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
+    def __init__(self, deleting_machines: 'str' =None, failing_machines: 'str' =None, message: 'str' =None, pending_machines: 'str' =None, ready_machines: 'str' =None, retries: '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._retries = None
         self._started_at = None
         self._stopped_at = None
         self.discriminator = None
@@ -80,6 +83,8 @@ class V1MultiMachineJobStatus(object):
             self.pending_machines = pending_machines
         if ready_machines is not None:
             self.ready_machines = ready_machines
+        if retries is not None:
+            self.retries = retries
         if started_at is not None:
             self.started_at = started_at
         if stopped_at is not None:
@@ -190,6 +195,27 @@ class V1MultiMachineJobStatus(object):
 
         self._ready_machines = ready_machines
 
+    @property
+    def retries(self) -> 'str':
+        """Gets the retries of this V1MultiMachineJobStatus.  # noqa: E501
+
+
+        :return: The retries of this V1MultiMachineJobStatus.  # noqa: E501
+        :rtype: str
+        """
+        return self._retries
+
+    @retries.setter
+    def retries(self, retries: 'str'):
+        """Sets the retries of this V1MultiMachineJobStatus.
+
+
+        :param retries: The retries of this V1MultiMachineJobStatus.  # noqa: E501
+        :type: str
+        """
+
+        self._retries = retries
+
     @property
     def started_at(self) -> 'datetime':
         """Gets the started_at of this V1MultiMachineJobStatus.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py

@@ -73,6 +73,7 @@ class V1RuleResource(object):
     MODEL = "model"
     SLURMJOB = "slurmJob"
     MULTIMACHINEJOB = "multiMachineJob"
+    LITLOGGER = "litLogger"
     """
     Attributes:
       swagger_types (dict): The key is attribute name

lightning_sdk/lightning_cloud/openapi/models/v1_snowflake_data_connection.py

@@ -43,52 +43,42 @@ class V1SnowflakeDataConnection(object):
     swagger_types = {
         'bucket_name': 'str',
         'compress': 'bool',
+        'credential_id': 'str',
         'format': 'str',
         'include_header': 'bool',
-        'query': 'str',
-        'secret_account': 'str',
-        'secret_password': 'str',
-        'secret_username': 'str'
+        'query': 'str'
     }
 
     attribute_map = {
         'bucket_name': 'bucketName',
         'compress': 'compress',
+        'credential_id': 'credentialId',
         'format': 'format',
         'include_header': 'includeHeader',
-        'query': 'query',
-        'secret_account': 'secretAccount',
-        'secret_password': 'secretPassword',
-        'secret_username': 'secretUsername'
+        'query': 'query'
     }
 
-    def __init__(self, bucket_name: 'str' =None, compress: 'bool' =None, format: 'str' =None, include_header: 'bool' =None, query: 'str' =None, secret_account: 'str' =None, secret_password: 'str' =None, secret_username: 'str' =None):  # noqa: E501
+    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._secret_account = None
-        self._secret_password = None
-        self._secret_username = 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
-        if secret_account is not None:
-            self.secret_account = secret_account
-        if secret_password is not None:
-            self.secret_password = secret_password
-        if secret_username is not None:
-            self.secret_username = secret_username
 
     @property
     def bucket_name(self) -> 'str':
@@ -132,6 +122,27 @@ class V1SnowflakeDataConnection(object):
 
         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
@@ -195,69 +206,6 @@ class V1SnowflakeDataConnection(object):
 
         self._query = query
 
-    @property
-    def secret_account(self) -> 'str':
-        """Gets the secret_account of this V1SnowflakeDataConnection.  # noqa: E501
-
-
-        :return: The secret_account of this V1SnowflakeDataConnection.  # noqa: E501
-        :rtype: str
-        """
-        return self._secret_account
-
-    @secret_account.setter
-    def secret_account(self, secret_account: 'str'):
-        """Sets the secret_account of this V1SnowflakeDataConnection.
-
-
-        :param secret_account: The secret_account of this V1SnowflakeDataConnection.  # noqa: E501
-        :type: str
-        """
-
-        self._secret_account = secret_account
-
-    @property
-    def secret_password(self) -> 'str':
-        """Gets the secret_password of this V1SnowflakeDataConnection.  # noqa: E501
-
-
-        :return: The secret_password of this V1SnowflakeDataConnection.  # noqa: E501
-        :rtype: str
-        """
-        return self._secret_password
-
-    @secret_password.setter
-    def secret_password(self, secret_password: 'str'):
-        """Sets the secret_password of this V1SnowflakeDataConnection.
-
-
-        :param secret_password: The secret_password of this V1SnowflakeDataConnection.  # noqa: E501
-        :type: str
-        """
-
-        self._secret_password = secret_password
-
-    @property
-    def secret_username(self) -> 'str':
-        """Gets the secret_username of this V1SnowflakeDataConnection.  # noqa: E501
-
-
-        :return: The secret_username of this V1SnowflakeDataConnection.  # noqa: E501
-        :rtype: str
-        """
-        return self._secret_username
-
-    @secret_username.setter
-    def secret_username(self, secret_username: 'str'):
-        """Sets the secret_username of this V1SnowflakeDataConnection.
-
-
-        :param secret_username: The secret_username of this V1SnowflakeDataConnection.  # noqa: E501
-        :type: str
-        """
-
-        self._secret_username = secret_username
-
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/v1_system_metrics.py

@@ -44,22 +44,25 @@ class V1SystemMetrics(object):
         'cpu': 'V1CPUSystemMetrics',
         'gpu': 'list[V1GPUSystemMetrics]',
         'storage': 'V1StorageSystemMetrics',
-        'timestamp': 'datetime'
+        'timestamp': 'datetime',
+        'trainium': 'list[V1TrainiumSystemMetrics]'
     }
 
     attribute_map = {
         'cpu': 'cpu',
         'gpu': 'gpu',
         'storage': 'storage',
-        'timestamp': 'timestamp'
+        'timestamp': 'timestamp',
+        'trainium': 'trainium'
     }
 
-    def __init__(self, cpu: 'V1CPUSystemMetrics' =None, gpu: 'list[V1GPUSystemMetrics]' =None, storage: 'V1StorageSystemMetrics' =None, timestamp: 'datetime' =None):  # noqa: E501
+    def __init__(self, cpu: 'V1CPUSystemMetrics' =None, gpu: 'list[V1GPUSystemMetrics]' =None, storage: 'V1StorageSystemMetrics' =None, timestamp: 'datetime' =None, trainium: 'list[V1TrainiumSystemMetrics]' =None):  # noqa: E501
         """V1SystemMetrics - a model defined in Swagger"""  # noqa: E501
         self._cpu = None
         self._gpu = None
         self._storage = None
         self._timestamp = None
+        self._trainium = None
         self.discriminator = None
         if cpu is not None:
             self.cpu = cpu
@@ -69,6 +72,8 @@ class V1SystemMetrics(object):
             self.storage = storage
         if timestamp is not None:
             self.timestamp = timestamp
+        if trainium is not None:
+            self.trainium = trainium
 
     @property
     def cpu(self) -> 'V1CPUSystemMetrics':
@@ -154,6 +159,27 @@ class V1SystemMetrics(object):
 
         self._timestamp = timestamp
 
+    @property
+    def trainium(self) -> 'list[V1TrainiumSystemMetrics]':
+        """Gets the trainium of this V1SystemMetrics.  # noqa: E501
+
+
+        :return: The trainium of this V1SystemMetrics.  # noqa: E501
+        :rtype: list[V1TrainiumSystemMetrics]
+        """
+        return self._trainium
+
+    @trainium.setter
+    def trainium(self, trainium: 'list[V1TrainiumSystemMetrics]'):
+        """Sets the trainium of this V1SystemMetrics.
+
+
+        :param trainium: The trainium of this V1SystemMetrics.  # noqa: E501
+        :type: list[V1TrainiumSystemMetrics]
+        """
+
+        self._trainium = trainium
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/v1_trainium_system_metrics.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 V1TrainiumSystemMetrics(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 = {
+        'memory_total': 'int',
+        'memory_used': 'int',
+        'utilization': 'int'
+    }
+
+    attribute_map = {
+        'memory_total': 'memoryTotal',
+        'memory_used': 'memoryUsed',
+        'utilization': 'utilization'
+    }
+
+    def __init__(self, memory_total: 'int' =None, memory_used: 'int' =None, utilization: 'int' =None):  # noqa: E501
+        """V1TrainiumSystemMetrics - a model defined in Swagger"""  # noqa: E501
+        self._memory_total = None
+        self._memory_used = None
+        self._utilization = None
+        self.discriminator = None
+        if memory_total is not None:
+            self.memory_total = memory_total
+        if memory_used is not None:
+            self.memory_used = memory_used
+        if utilization is not None:
+            self.utilization = utilization
+
+    @property
+    def memory_total(self) -> 'int':
+        """Gets the memory_total of this V1TrainiumSystemMetrics.  # noqa: E501
+
+
+        :return: The memory_total of this V1TrainiumSystemMetrics.  # noqa: E501
+        :rtype: int
+        """
+        return self._memory_total
+
+    @memory_total.setter
+    def memory_total(self, memory_total: 'int'):
+        """Sets the memory_total of this V1TrainiumSystemMetrics.
+
+
+        :param memory_total: The memory_total of this V1TrainiumSystemMetrics.  # noqa: E501
+        :type: int
+        """
+
+        self._memory_total = memory_total
+
+    @property
+    def memory_used(self) -> 'int':
+        """Gets the memory_used of this V1TrainiumSystemMetrics.  # noqa: E501
+
+
+        :return: The memory_used of this V1TrainiumSystemMetrics.  # noqa: E501
+        :rtype: int
+        """
+        return self._memory_used
+
+    @memory_used.setter
+    def memory_used(self, memory_used: 'int'):
+        """Sets the memory_used of this V1TrainiumSystemMetrics.
+
+
+        :param memory_used: The memory_used of this V1TrainiumSystemMetrics.  # noqa: E501
+        :type: int
+        """
+
+        self._memory_used = memory_used
+
+    @property
+    def utilization(self) -> 'int':
+        """Gets the utilization of this V1TrainiumSystemMetrics.  # noqa: E501
+
+
+        :return: The utilization of this V1TrainiumSystemMetrics.  # noqa: E501
+        :rtype: int
+        """
+        return self._utilization
+
+    @utilization.setter
+    def utilization(self, utilization: 'int'):
+        """Sets the utilization of this V1TrainiumSystemMetrics.
+
+
+        :param utilization: The utilization of this V1TrainiumSystemMetrics.  # noqa: E501
+        :type: int
+        """
+
+        self._utilization = utilization
+
+    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(V1TrainiumSystemMetrics, 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: 'V1TrainiumSystemMetrics') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1TrainiumSystemMetrics):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1TrainiumSystemMetrics') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_update_metrics_stream_visibility_response.py

@@ -0,0 +1,97 @@
+# 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 V1UpdateMetricsStreamVisibilityResponse(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 = {
+    }
+
+    attribute_map = {
+    }
+
+    def __init__(self):  # noqa: E501
+        """V1UpdateMetricsStreamVisibilityResponse - a model defined in Swagger"""  # noqa: E501
+        self.discriminator = None
+
+    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(V1UpdateMetricsStreamVisibilityResponse, 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: 'V1UpdateMetricsStreamVisibilityResponse') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1UpdateMetricsStreamVisibilityResponse):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1UpdateMetricsStreamVisibilityResponse') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other