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

lightning_sdk/lightning_cloud/openapi/models/v1_get_cluster_accelerator_demand_response.py

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

lightning_sdk/lightning_cloud/openapi/models/v1_job.py

@@ -54,6 +54,7 @@ class V1Job(object):
         'ready_at': 'datetime',
         'release_id': 'str',
         'restart_count': 'int',
+        'restart_timings': 'list[V1RestartTiming]',
         'server_error': 'str',
         'spec': 'V1JobSpec',
         'started_at': 'datetime',
@@ -82,6 +83,7 @@ class V1Job(object):
         'ready_at': 'readyAt',
         'release_id': 'releaseId',
         'restart_count': 'restartCount',
+        'restart_timings': 'restartTimings',
         'server_error': 'serverError',
         'spec': 'spec',
         'started_at': 'startedAt',
@@ -96,7 +98,7 @@ class V1Job(object):
         'user_logs_started_at': 'userLogsStartedAt'
     }
 
-    def __init__(self, created_at: 'datetime' =None, deployment_id: 'str' =None, endpoint: 'V1Endpoint' =None, id: 'str' =None, is_healthy: 'bool' =None, message: 'str' =None, multi_machine_job_id: 'str' =None, name: 'str' =None, overprovisioned: 'bool' =None, project_id: 'str' =None, ready_at: 'datetime' =None, release_id: 'str' =None, restart_count: 'int' =None, server_error: 'str' =None, spec: 'V1JobSpec' =None, started_at: 'datetime' =None, state: 'str' =None, stopped_at: 'datetime' =None, timings: 'dict(str, V1JobTiming)' =None, total_cost: 'float' =None, total_size_bytes: 'str' =None, updated_at: 'datetime' =None, urls: 'list[str]' =None, user_id: 'str' =None, user_logs_started_at: 'datetime' =None):  # noqa: E501
+    def __init__(self, created_at: 'datetime' =None, deployment_id: 'str' =None, endpoint: 'V1Endpoint' =None, id: 'str' =None, is_healthy: 'bool' =None, message: 'str' =None, multi_machine_job_id: 'str' =None, name: 'str' =None, overprovisioned: 'bool' =None, project_id: 'str' =None, ready_at: 'datetime' =None, release_id: 'str' =None, restart_count: 'int' =None, restart_timings: 'list[V1RestartTiming]' =None, server_error: 'str' =None, spec: 'V1JobSpec' =None, started_at: 'datetime' =None, state: 'str' =None, stopped_at: 'datetime' =None, timings: 'dict(str, V1JobTiming)' =None, total_cost: 'float' =None, total_size_bytes: 'str' =None, updated_at: 'datetime' =None, urls: 'list[str]' =None, user_id: 'str' =None, user_logs_started_at: 'datetime' =None):  # noqa: E501
         """V1Job - a model defined in Swagger"""  # noqa: E501
         self._created_at = None
         self._deployment_id = None
@@ -111,6 +113,7 @@ class V1Job(object):
         self._ready_at = None
         self._release_id = None
         self._restart_count = None
+        self._restart_timings = None
         self._server_error = None
         self._spec = None
         self._started_at = None
@@ -150,6 +153,8 @@ class V1Job(object):
             self.release_id = release_id
         if restart_count is not None:
             self.restart_count = restart_count
+        if restart_timings is not None:
+            self.restart_timings = restart_timings
         if server_error is not None:
             self.server_error = server_error
         if spec is not None:
@@ -450,6 +455,27 @@ class V1Job(object):
 
         self._restart_count = restart_count
 
+    @property
+    def restart_timings(self) -> 'list[V1RestartTiming]':
+        """Gets the restart_timings of this V1Job.  # noqa: E501
+
+
+        :return: The restart_timings of this V1Job.  # noqa: E501
+        :rtype: list[V1RestartTiming]
+        """
+        return self._restart_timings
+
+    @restart_timings.setter
+    def restart_timings(self, restart_timings: 'list[V1RestartTiming]'):
+        """Sets the restart_timings of this V1Job.
+
+
+        :param restart_timings: The restart_timings of this V1Job.  # noqa: E501
+        :type: list[V1RestartTiming]
+        """
+
+        self._restart_timings = restart_timings
+
     @property
     def server_error(self) -> 'str':
         """Gets the server_error of this V1Job.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py

@@ -57,6 +57,7 @@ class V1JobSpec(object):
         'instance_type': 'str',
         'modified_volume': 'bool',
         'path_mappings': 'list[V1PathMapping]',
+        'quantity': 'int',
         'readiness_probe': 'V1JobHealthCheckConfig',
         'regions': 'list[str]',
         'requested_run_duration_seconds': 'str',
@@ -84,6 +85,7 @@ class V1JobSpec(object):
         'instance_type': 'instanceType',
         'modified_volume': 'modifiedVolume',
         'path_mappings': 'pathMappings',
+        'quantity': 'quantity',
         'readiness_probe': 'readinessProbe',
         'regions': 'regions',
         'requested_run_duration_seconds': 'requestedRunDurationSeconds',
@@ -94,7 +96,7 @@ class V1JobSpec(object):
         'volumes': 'volumes'
     }
 
-    def __init__(self, artifacts_destination: 'str' =None, artifacts_source: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, data_paths: 'list[V1DataPath]' =None, entrypoint: 'str' =None, env: 'list[V1EnvVar]' =None, image: 'str' =None, image_cluster_credentials: 'bool' =None, image_secret_ref: 'str' =None, include_credentials: 'bool' =None, instance_name: 'str' =None, instance_type: 'str' =None, modified_volume: 'bool' =None, path_mappings: 'list[V1PathMapping]' =None, readiness_probe: 'V1JobHealthCheckConfig' =None, regions: 'list[str]' =None, requested_run_duration_seconds: 'str' =None, resources: 'V1Resources' =None, restart_policy: 'str' =None, run_id: 'str' =None, spot: 'bool' =None, volumes: 'list[V1Volume]' =None):  # noqa: E501
+    def __init__(self, artifacts_destination: 'str' =None, artifacts_source: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, data_paths: 'list[V1DataPath]' =None, entrypoint: 'str' =None, env: 'list[V1EnvVar]' =None, image: 'str' =None, image_cluster_credentials: 'bool' =None, image_secret_ref: 'str' =None, include_credentials: 'bool' =None, instance_name: 'str' =None, instance_type: 'str' =None, modified_volume: 'bool' =None, path_mappings: 'list[V1PathMapping]' =None, quantity: 'int' =None, readiness_probe: 'V1JobHealthCheckConfig' =None, regions: 'list[str]' =None, requested_run_duration_seconds: 'str' =None, resources: 'V1Resources' =None, restart_policy: 'str' =None, run_id: 'str' =None, spot: 'bool' =None, volumes: 'list[V1Volume]' =None):  # noqa: E501
         """V1JobSpec - a model defined in Swagger"""  # noqa: E501
         self._artifacts_destination = None
         self._artifacts_source = None
@@ -112,6 +114,7 @@ class V1JobSpec(object):
         self._instance_type = None
         self._modified_volume = None
         self._path_mappings = None
+        self._quantity = None
         self._readiness_probe = None
         self._regions = None
         self._requested_run_duration_seconds = None
@@ -153,6 +156,8 @@ class V1JobSpec(object):
             self.modified_volume = modified_volume
         if path_mappings is not None:
             self.path_mappings = path_mappings
+        if quantity is not None:
+            self.quantity = quantity
         if readiness_probe is not None:
             self.readiness_probe = readiness_probe
         if regions is not None:
@@ -506,6 +511,27 @@ class V1JobSpec(object):
 
         self._path_mappings = path_mappings
 
+    @property
+    def quantity(self) -> 'int':
+        """Gets the quantity of this V1JobSpec.  # noqa: E501
+
+
+        :return: The quantity of this V1JobSpec.  # noqa: E501
+        :rtype: int
+        """
+        return self._quantity
+
+    @quantity.setter
+    def quantity(self, quantity: 'int'):
+        """Sets the quantity of this V1JobSpec.
+
+
+        :param quantity: The quantity of this V1JobSpec.  # noqa: E501
+        :type: int
+        """
+
+        self._quantity = quantity
+
     @property
     def readiness_probe(self) -> 'V1JobHealthCheckConfig':
         """Gets the readiness_probe of this V1JobSpec.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_artifact.py

@@ -42,6 +42,7 @@ class V1LitRegistryArtifact(object):
     """
     swagger_types = {
         'full_hash_digest': 'str',
+        'location': 'str',
         'pull_time': 'datetime',
         'push_time': 'datetime',
         'short_hash_digest': 'str',
@@ -51,6 +52,7 @@ class V1LitRegistryArtifact(object):
 
     attribute_map = {
         'full_hash_digest': 'fullHashDigest',
+        'location': 'location',
         'pull_time': 'pullTime',
         'push_time': 'pushTime',
         'short_hash_digest': 'shortHashDigest',
@@ -58,9 +60,10 @@ class V1LitRegistryArtifact(object):
         'tag_name': 'tagName'
     }
 
-    def __init__(self, full_hash_digest: 'str' =None, pull_time: 'datetime' =None, push_time: 'datetime' =None, short_hash_digest: 'str' =None, size: 'str' =None, tag_name: 'str' =None):  # noqa: E501
+    def __init__(self, full_hash_digest: 'str' =None, location: 'str' =None, pull_time: 'datetime' =None, push_time: 'datetime' =None, short_hash_digest: 'str' =None, size: 'str' =None, tag_name: 'str' =None):  # noqa: E501
         """V1LitRegistryArtifact - a model defined in Swagger"""  # noqa: E501
         self._full_hash_digest = None
+        self._location = None
         self._pull_time = None
         self._push_time = None
         self._short_hash_digest = None
@@ -69,6 +72,8 @@ class V1LitRegistryArtifact(object):
         self.discriminator = None
         if full_hash_digest is not None:
             self.full_hash_digest = full_hash_digest
+        if location is not None:
+            self.location = location
         if pull_time is not None:
             self.pull_time = pull_time
         if push_time is not None:
@@ -101,6 +106,27 @@ class V1LitRegistryArtifact(object):
 
         self._full_hash_digest = full_hash_digest
 
+    @property
+    def location(self) -> 'str':
+        """Gets the location of this V1LitRegistryArtifact.  # noqa: E501
+
+
+        :return: The location of this V1LitRegistryArtifact.  # noqa: E501
+        :rtype: str
+        """
+        return self._location
+
+    @location.setter
+    def location(self, location: 'str'):
+        """Sets the location of this V1LitRegistryArtifact.
+
+
+        :param location: The location of this V1LitRegistryArtifact.  # noqa: E501
+        :type: str
+        """
+
+        self._location = location
+
     @property
     def pull_time(self) -> 'datetime':
         """Gets the pull_time of this V1LitRegistryArtifact.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_project.py

@@ -79,6 +79,7 @@ class V1LitRegistryProject(object):
     def id(self) -> 'int':
         """Gets the id of this V1LitRegistryProject.  # noqa: E501
 
+        This is the litcr project ID.  # noqa: E501
 
         :return: The id of this V1LitRegistryProject.  # noqa: E501
         :rtype: int
@@ -89,6 +90,7 @@ class V1LitRegistryProject(object):
     def id(self, id: 'int'):
         """Sets the id of this V1LitRegistryProject.
 
+        This is the litcr project ID.  # noqa: E501
 
         :param id: The id of this V1LitRegistryProject.  # noqa: E501
         :type: int
@@ -121,6 +123,7 @@ class V1LitRegistryProject(object):
     def project_id(self) -> 'str':
         """Gets the project_id of this V1LitRegistryProject.  # noqa: E501
 
+        This is the lightning AI teamspace project ID.  # noqa: E501
 
         :return: The project_id of this V1LitRegistryProject.  # noqa: E501
         :rtype: str
@@ -131,6 +134,7 @@ class V1LitRegistryProject(object):
     def project_id(self, project_id: 'str'):
         """Sets the project_id of this V1LitRegistryProject.
 
+        This is the lightning AI teamspace project ID.  # noqa: E501
 
         :param project_id: The project_id of this V1LitRegistryProject.  # noqa: E501
         :type: str
@@ -142,6 +146,7 @@ class V1LitRegistryProject(object):
     def repo_count(self) -> 'str':
         """Gets the repo_count of this V1LitRegistryProject.  # noqa: E501
 
+        This is how many repositories exist in a project. This will be the same as the number of repository objects we return.  # noqa: E501
 
         :return: The repo_count of this V1LitRegistryProject.  # noqa: E501
         :rtype: str
@@ -152,6 +157,7 @@ class V1LitRegistryProject(object):
     def repo_count(self, repo_count: 'str'):
         """Sets the repo_count of this V1LitRegistryProject.
 
+        This is how many repositories exist in a project. This will be the same as the number of repository objects we return.  # noqa: E501
 
         :param repo_count: The repo_count of this V1LitRegistryProject.  # noqa: E501
         :type: str
@@ -163,6 +169,7 @@ class V1LitRegistryProject(object):
     def repositories(self) -> 'list[V1LitRepository]':
         """Gets the repositories of this V1LitRegistryProject.  # noqa: E501
 
+        This is the list of repositories that belong to a project in our registry.  # noqa: E501
 
         :return: The repositories of this V1LitRegistryProject.  # noqa: E501
         :rtype: list[V1LitRepository]
@@ -173,6 +180,7 @@ class V1LitRegistryProject(object):
     def repositories(self, repositories: 'list[V1LitRepository]'):
         """Sets the repositories of this V1LitRegistryProject.
 
+        This is the list of repositories that belong to a project in our registry.  # noqa: E501
 
         :param repositories: The repositories of this V1LitRegistryProject.  # noqa: E501
         :type: list[V1LitRepository]

lightning_sdk/lightning_cloud/openapi/models/v1_lit_repository.py

@@ -45,6 +45,7 @@ class V1LitRepository(object):
         'creation_time': 'datetime',
         'description': 'str',
         'id': 'str',
+        'latest_artifact': 'V1LitRegistryArtifact',
         'name': 'str',
         'pull_count': 'str',
         'update_time': 'datetime'
@@ -55,17 +56,19 @@ class V1LitRepository(object):
         'creation_time': 'creationTime',
         'description': 'description',
         'id': 'id',
+        'latest_artifact': 'latestArtifact',
         'name': 'name',
         'pull_count': 'pullCount',
         'update_time': 'updateTime'
     }
 
-    def __init__(self, artifact_count: 'str' =None, creation_time: 'datetime' =None, description: 'str' =None, id: 'str' =None, name: 'str' =None, pull_count: 'str' =None, update_time: 'datetime' =None):  # noqa: E501
+    def __init__(self, artifact_count: 'str' =None, creation_time: 'datetime' =None, description: 'str' =None, id: 'str' =None, latest_artifact: 'V1LitRegistryArtifact' =None, name: 'str' =None, pull_count: 'str' =None, update_time: 'datetime' =None):  # noqa: E501
         """V1LitRepository - a model defined in Swagger"""  # noqa: E501
         self._artifact_count = None
         self._creation_time = None
         self._description = None
         self._id = None
+        self._latest_artifact = None
         self._name = None
         self._pull_count = None
         self._update_time = None
@@ -78,6 +81,8 @@ class V1LitRepository(object):
             self.description = description
         if id is not None:
             self.id = id
+        if latest_artifact is not None:
+            self.latest_artifact = latest_artifact
         if name is not None:
             self.name = name
         if pull_count is not None:
@@ -169,6 +174,27 @@ class V1LitRepository(object):
 
         self._id = id
 
+    @property
+    def latest_artifact(self) -> 'V1LitRegistryArtifact':
+        """Gets the latest_artifact of this V1LitRepository.  # noqa: E501
+
+
+        :return: The latest_artifact of this V1LitRepository.  # noqa: E501
+        :rtype: V1LitRegistryArtifact
+        """
+        return self._latest_artifact
+
+    @latest_artifact.setter
+    def latest_artifact(self, latest_artifact: 'V1LitRegistryArtifact'):
+        """Sets the latest_artifact of this V1LitRepository.
+
+
+        :param latest_artifact: The latest_artifact of this V1LitRepository.  # noqa: E501
+        :type: V1LitRegistryArtifact
+        """
+
+        self._latest_artifact = latest_artifact
+
     @property
     def name(self) -> 'str':
         """Gets the name of this V1LitRepository.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_report_restart_timings_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 V1ReportRestartTimingsResponse(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
+        """V1ReportRestartTimingsResponse - 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(V1ReportRestartTimingsResponse, 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: 'V1ReportRestartTimingsResponse') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1ReportRestartTimingsResponse):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1ReportRestartTimingsResponse') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other