lightning-sdk 0.1.49__py3-none-any.whl → 2025.11.5__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_job_resource.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 V1JobResource(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 = {
+        'agent_job': 'V1AgentJob',
+        'app_instance': 'Externalv1LightningappInstance',
+        'batch_job': 'V1Job',
+        'deployment': 'V1Deployment',
+        'multi_machine_job': 'V1MultiMachineJob',
+        'pipeline': 'V1Pipeline',
+        'project_id': 'str'
+    }
+
+    attribute_map = {
+        'agent_job': 'agentJob',
+        'app_instance': 'appInstance',
+        'batch_job': 'batchJob',
+        'deployment': 'deployment',
+        'multi_machine_job': 'multiMachineJob',
+        'pipeline': 'pipeline',
+        'project_id': 'projectId'
+    }
+
+    def __init__(self, agent_job: 'V1AgentJob' =None, app_instance: 'Externalv1LightningappInstance' =None, batch_job: 'V1Job' =None, deployment: 'V1Deployment' =None, multi_machine_job: 'V1MultiMachineJob' =None, pipeline: 'V1Pipeline' =None, project_id: 'str' =None):  # noqa: E501
+        """V1JobResource - a model defined in Swagger"""  # noqa: E501
+        self._agent_job = None
+        self._app_instance = None
+        self._batch_job = None
+        self._deployment = None
+        self._multi_machine_job = None
+        self._pipeline = None
+        self._project_id = None
+        self.discriminator = None
+        if agent_job is not None:
+            self.agent_job = agent_job
+        if app_instance is not None:
+            self.app_instance = app_instance
+        if batch_job is not None:
+            self.batch_job = batch_job
+        if deployment is not None:
+            self.deployment = deployment
+        if multi_machine_job is not None:
+            self.multi_machine_job = multi_machine_job
+        if pipeline is not None:
+            self.pipeline = pipeline
+        if project_id is not None:
+            self.project_id = project_id
+
+    @property
+    def agent_job(self) -> 'V1AgentJob':
+        """Gets the agent_job of this V1JobResource.  # noqa: E501
+
+
+        :return: The agent_job of this V1JobResource.  # noqa: E501
+        :rtype: V1AgentJob
+        """
+        return self._agent_job
+
+    @agent_job.setter
+    def agent_job(self, agent_job: 'V1AgentJob'):
+        """Sets the agent_job of this V1JobResource.
+
+
+        :param agent_job: The agent_job of this V1JobResource.  # noqa: E501
+        :type: V1AgentJob
+        """
+
+        self._agent_job = agent_job
+
+    @property
+    def app_instance(self) -> 'Externalv1LightningappInstance':
+        """Gets the app_instance of this V1JobResource.  # noqa: E501
+
+
+        :return: The app_instance of this V1JobResource.  # noqa: E501
+        :rtype: Externalv1LightningappInstance
+        """
+        return self._app_instance
+
+    @app_instance.setter
+    def app_instance(self, app_instance: 'Externalv1LightningappInstance'):
+        """Sets the app_instance of this V1JobResource.
+
+
+        :param app_instance: The app_instance of this V1JobResource.  # noqa: E501
+        :type: Externalv1LightningappInstance
+        """
+
+        self._app_instance = app_instance
+
+    @property
+    def batch_job(self) -> 'V1Job':
+        """Gets the batch_job of this V1JobResource.  # noqa: E501
+
+
+        :return: The batch_job of this V1JobResource.  # noqa: E501
+        :rtype: V1Job
+        """
+        return self._batch_job
+
+    @batch_job.setter
+    def batch_job(self, batch_job: 'V1Job'):
+        """Sets the batch_job of this V1JobResource.
+
+
+        :param batch_job: The batch_job of this V1JobResource.  # noqa: E501
+        :type: V1Job
+        """
+
+        self._batch_job = batch_job
+
+    @property
+    def deployment(self) -> 'V1Deployment':
+        """Gets the deployment of this V1JobResource.  # noqa: E501
+
+
+        :return: The deployment of this V1JobResource.  # noqa: E501
+        :rtype: V1Deployment
+        """
+        return self._deployment
+
+    @deployment.setter
+    def deployment(self, deployment: 'V1Deployment'):
+        """Sets the deployment of this V1JobResource.
+
+
+        :param deployment: The deployment of this V1JobResource.  # noqa: E501
+        :type: V1Deployment
+        """
+
+        self._deployment = deployment
+
+    @property
+    def multi_machine_job(self) -> 'V1MultiMachineJob':
+        """Gets the multi_machine_job of this V1JobResource.  # noqa: E501
+
+
+        :return: The multi_machine_job of this V1JobResource.  # noqa: E501
+        :rtype: V1MultiMachineJob
+        """
+        return self._multi_machine_job
+
+    @multi_machine_job.setter
+    def multi_machine_job(self, multi_machine_job: 'V1MultiMachineJob'):
+        """Sets the multi_machine_job of this V1JobResource.
+
+
+        :param multi_machine_job: The multi_machine_job of this V1JobResource.  # noqa: E501
+        :type: V1MultiMachineJob
+        """
+
+        self._multi_machine_job = multi_machine_job
+
+    @property
+    def pipeline(self) -> 'V1Pipeline':
+        """Gets the pipeline of this V1JobResource.  # noqa: E501
+
+
+        :return: The pipeline of this V1JobResource.  # noqa: E501
+        :rtype: V1Pipeline
+        """
+        return self._pipeline
+
+    @pipeline.setter
+    def pipeline(self, pipeline: 'V1Pipeline'):
+        """Sets the pipeline of this V1JobResource.
+
+
+        :param pipeline: The pipeline of this V1JobResource.  # noqa: E501
+        :type: V1Pipeline
+        """
+
+        self._pipeline = pipeline
+
+    @property
+    def project_id(self) -> 'str':
+        """Gets the project_id of this V1JobResource.  # noqa: E501
+
+
+        :return: The project_id of this V1JobResource.  # noqa: E501
+        :rtype: str
+        """
+        return self._project_id
+
+    @project_id.setter
+    def project_id(self, project_id: 'str'):
+        """Sets the project_id of this V1JobResource.
+
+
+        :param project_id: The project_id of this V1JobResource.  # noqa: E501
+        :type: str
+        """
+
+        self._project_id = project_id
+
+    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(V1JobResource, 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: 'V1JobResource') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1JobResource):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1JobResource') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py

@@ -47,6 +47,7 @@ class V1JobSpec(object):
         'cluster_id': 'str',
         'command': 'str',
         'data_paths': 'list[V1DataPath]',
+        'disable_artifacts_persistence': 'bool',
         'entrypoint': 'str',
         'env': 'list[V1EnvVar]',
         'image': 'str',
@@ -55,13 +56,20 @@ class V1JobSpec(object):
         'include_credentials': 'bool',
         'instance_name': 'str',
         'instance_type': 'str',
+        'machine_image_version': 'str',
         'modified_volume': 'bool',
+        'path_mappings': 'list[V1PathMapping]',
+        'pipeline_id': 'str',
+        'quantity': 'int',
+        'rank': 'int',
         'readiness_probe': 'V1JobHealthCheckConfig',
         'regions': 'list[str]',
+        'requested_run_duration_seconds': 'str',
         'resources': 'V1Resources',
         'restart_policy': 'str',
         'run_id': 'str',
         'spot': 'bool',
+        'use_tls': 'bool',
         'volumes': 'list[V1Volume]'
     }
 
@@ -72,6 +80,7 @@ class V1JobSpec(object):
         'cluster_id': 'clusterId',
         'command': 'command',
         'data_paths': 'dataPaths',
+        'disable_artifacts_persistence': 'disableArtifactsPersistence',
         'entrypoint': 'entrypoint',
         'env': 'env',
         'image': 'image',
@@ -80,17 +89,24 @@ class V1JobSpec(object):
         'include_credentials': 'includeCredentials',
         'instance_name': 'instanceName',
         'instance_type': 'instanceType',
+        'machine_image_version': 'machineImageVersion',
         'modified_volume': 'modifiedVolume',
+        'path_mappings': 'pathMappings',
+        'pipeline_id': 'pipelineId',
+        'quantity': 'quantity',
+        'rank': 'rank',
         'readiness_probe': 'readinessProbe',
         'regions': 'regions',
+        'requested_run_duration_seconds': 'requestedRunDurationSeconds',
         'resources': 'resources',
         'restart_policy': 'restartPolicy',
         'run_id': 'runId',
         'spot': 'spot',
+        'use_tls': 'useTls',
         '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, readiness_probe: 'V1JobHealthCheckConfig' =None, regions: 'list[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, disable_artifacts_persistence: 'bool' =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, machine_image_version: 'str' =None, modified_volume: 'bool' =None, path_mappings: 'list[V1PathMapping]' =None, pipeline_id: 'str' =None, quantity: 'int' =None, rank: '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, use_tls: 'bool' =None, volumes: 'list[V1Volume]' =None):  # noqa: E501
         """V1JobSpec - a model defined in Swagger"""  # noqa: E501
         self._artifacts_destination = None
         self._artifacts_source = None
@@ -98,6 +114,7 @@ class V1JobSpec(object):
         self._cluster_id = None
         self._command = None
         self._data_paths = None
+        self._disable_artifacts_persistence = None
         self._entrypoint = None
         self._env = None
         self._image = None
@@ -106,13 +123,20 @@ class V1JobSpec(object):
         self._include_credentials = None
         self._instance_name = None
         self._instance_type = None
+        self._machine_image_version = None
         self._modified_volume = None
+        self._path_mappings = None
+        self._pipeline_id = None
+        self._quantity = None
+        self._rank = None
         self._readiness_probe = None
         self._regions = None
+        self._requested_run_duration_seconds = None
         self._resources = None
         self._restart_policy = None
         self._run_id = None
         self._spot = None
+        self._use_tls = None
         self._volumes = None
         self.discriminator = None
         if artifacts_destination is not None:
@@ -127,6 +151,8 @@ class V1JobSpec(object):
             self.command = command
         if data_paths is not None:
             self.data_paths = data_paths
+        if disable_artifacts_persistence is not None:
+            self.disable_artifacts_persistence = disable_artifacts_persistence
         if entrypoint is not None:
             self.entrypoint = entrypoint
         if env is not None:
@@ -143,12 +169,24 @@ class V1JobSpec(object):
             self.instance_name = instance_name
         if instance_type is not None:
             self.instance_type = instance_type
+        if machine_image_version is not None:
+            self.machine_image_version = machine_image_version
         if modified_volume is not None:
             self.modified_volume = modified_volume
+        if path_mappings is not None:
+            self.path_mappings = path_mappings
+        if pipeline_id is not None:
+            self.pipeline_id = pipeline_id
+        if quantity is not None:
+            self.quantity = quantity
+        if rank is not None:
+            self.rank = rank
         if readiness_probe is not None:
             self.readiness_probe = readiness_probe
         if regions is not None:
             self.regions = regions
+        if requested_run_duration_seconds is not None:
+            self.requested_run_duration_seconds = requested_run_duration_seconds
         if resources is not None:
             self.resources = resources
         if restart_policy is not None:
@@ -157,6 +195,8 @@ class V1JobSpec(object):
             self.run_id = run_id
         if spot is not None:
             self.spot = spot
+        if use_tls is not None:
+            self.use_tls = use_tls
         if volumes is not None:
             self.volumes = volumes
 
@@ -286,6 +326,27 @@ class V1JobSpec(object):
 
         self._data_paths = data_paths
 
+    @property
+    def disable_artifacts_persistence(self) -> 'bool':
+        """Gets the disable_artifacts_persistence of this V1JobSpec.  # noqa: E501
+
+
+        :return: The disable_artifacts_persistence of this V1JobSpec.  # noqa: E501
+        :rtype: bool
+        """
+        return self._disable_artifacts_persistence
+
+    @disable_artifacts_persistence.setter
+    def disable_artifacts_persistence(self, disable_artifacts_persistence: 'bool'):
+        """Sets the disable_artifacts_persistence of this V1JobSpec.
+
+
+        :param disable_artifacts_persistence: The disable_artifacts_persistence of this V1JobSpec.  # noqa: E501
+        :type: bool
+        """
+
+        self._disable_artifacts_persistence = disable_artifacts_persistence
+
     @property
     def entrypoint(self) -> 'str':
         """Gets the entrypoint of this V1JobSpec.  # noqa: E501
@@ -454,6 +515,27 @@ class V1JobSpec(object):
 
         self._instance_type = instance_type
 
+    @property
+    def machine_image_version(self) -> 'str':
+        """Gets the machine_image_version of this V1JobSpec.  # noqa: E501
+
+
+        :return: The machine_image_version of this V1JobSpec.  # noqa: E501
+        :rtype: str
+        """
+        return self._machine_image_version
+
+    @machine_image_version.setter
+    def machine_image_version(self, machine_image_version: 'str'):
+        """Sets the machine_image_version of this V1JobSpec.
+
+
+        :param machine_image_version: The machine_image_version of this V1JobSpec.  # noqa: E501
+        :type: str
+        """
+
+        self._machine_image_version = machine_image_version
+
     @property
     def modified_volume(self) -> 'bool':
         """Gets the modified_volume of this V1JobSpec.  # noqa: E501
@@ -475,6 +557,90 @@ class V1JobSpec(object):
 
         self._modified_volume = modified_volume
 
+    @property
+    def path_mappings(self) -> 'list[V1PathMapping]':
+        """Gets the path_mappings of this V1JobSpec.  # noqa: E501
+
+
+        :return: The path_mappings of this V1JobSpec.  # noqa: E501
+        :rtype: list[V1PathMapping]
+        """
+        return self._path_mappings
+
+    @path_mappings.setter
+    def path_mappings(self, path_mappings: 'list[V1PathMapping]'):
+        """Sets the path_mappings of this V1JobSpec.
+
+
+        :param path_mappings: The path_mappings of this V1JobSpec.  # noqa: E501
+        :type: list[V1PathMapping]
+        """
+
+        self._path_mappings = path_mappings
+
+    @property
+    def pipeline_id(self) -> 'str':
+        """Gets the pipeline_id of this V1JobSpec.  # noqa: E501
+
+
+        :return: The pipeline_id of this V1JobSpec.  # noqa: E501
+        :rtype: str
+        """
+        return self._pipeline_id
+
+    @pipeline_id.setter
+    def pipeline_id(self, pipeline_id: 'str'):
+        """Sets the pipeline_id of this V1JobSpec.
+
+
+        :param pipeline_id: The pipeline_id of this V1JobSpec.  # noqa: E501
+        :type: str
+        """
+
+        self._pipeline_id = pipeline_id
+
+    @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 rank(self) -> 'int':
+        """Gets the rank of this V1JobSpec.  # noqa: E501
+
+
+        :return: The rank of this V1JobSpec.  # noqa: E501
+        :rtype: int
+        """
+        return self._rank
+
+    @rank.setter
+    def rank(self, rank: 'int'):
+        """Sets the rank of this V1JobSpec.
+
+
+        :param rank: The rank of this V1JobSpec.  # noqa: E501
+        :type: int
+        """
+
+        self._rank = rank
+
     @property
     def readiness_probe(self) -> 'V1JobHealthCheckConfig':
         """Gets the readiness_probe of this V1JobSpec.  # noqa: E501
@@ -517,6 +683,27 @@ class V1JobSpec(object):
 
         self._regions = regions
 
+    @property
+    def requested_run_duration_seconds(self) -> 'str':
+        """Gets the requested_run_duration_seconds of this V1JobSpec.  # noqa: E501
+
+
+        :return: The requested_run_duration_seconds of this V1JobSpec.  # noqa: E501
+        :rtype: str
+        """
+        return self._requested_run_duration_seconds
+
+    @requested_run_duration_seconds.setter
+    def requested_run_duration_seconds(self, requested_run_duration_seconds: 'str'):
+        """Sets the requested_run_duration_seconds of this V1JobSpec.
+
+
+        :param requested_run_duration_seconds: The requested_run_duration_seconds of this V1JobSpec.  # noqa: E501
+        :type: str
+        """
+
+        self._requested_run_duration_seconds = requested_run_duration_seconds
+
     @property
     def resources(self) -> 'V1Resources':
         """Gets the resources of this V1JobSpec.  # noqa: E501
@@ -601,6 +788,27 @@ class V1JobSpec(object):
 
         self._spot = spot
 
+    @property
+    def use_tls(self) -> 'bool':
+        """Gets the use_tls of this V1JobSpec.  # noqa: E501
+
+
+        :return: The use_tls of this V1JobSpec.  # noqa: E501
+        :rtype: bool
+        """
+        return self._use_tls
+
+    @use_tls.setter
+    def use_tls(self, use_tls: 'bool'):
+        """Sets the use_tls of this V1JobSpec.
+
+
+        :param use_tls: The use_tls of this V1JobSpec.  # noqa: E501
+        :type: bool
+        """
+
+        self._use_tls = use_tls
+
     @property
     def volumes(self) -> 'list[V1Volume]':
         """Gets the volumes of this V1JobSpec.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_job_timing.py

@@ -44,6 +44,7 @@ class V1JobTiming(object):
         'end': 'datetime',
         'name': 'str',
         'payload': 'str',
+        'rank': 'int',
         'start': 'datetime'
     }
 
@@ -51,14 +52,16 @@ class V1JobTiming(object):
         'end': 'end',
         'name': 'name',
         'payload': 'payload',
+        'rank': 'rank',
         'start': 'start'
     }
 
-    def __init__(self, end: 'datetime' =None, name: 'str' =None, payload: 'str' =None, start: 'datetime' =None):  # noqa: E501
+    def __init__(self, end: 'datetime' =None, name: 'str' =None, payload: 'str' =None, rank: 'int' =None, start: 'datetime' =None):  # noqa: E501
         """V1JobTiming - a model defined in Swagger"""  # noqa: E501
         self._end = None
         self._name = None
         self._payload = None
+        self._rank = None
         self._start = None
         self.discriminator = None
         if end is not None:
@@ -67,6 +70,8 @@ class V1JobTiming(object):
             self.name = name
         if payload is not None:
             self.payload = payload
+        if rank is not None:
+            self.rank = rank
         if start is not None:
             self.start = start
 
@@ -133,6 +138,27 @@ class V1JobTiming(object):
 
         self._payload = payload
 
+    @property
+    def rank(self) -> 'int':
+        """Gets the rank of this V1JobTiming.  # noqa: E501
+
+
+        :return: The rank of this V1JobTiming.  # noqa: E501
+        :rtype: int
+        """
+        return self._rank
+
+    @rank.setter
+    def rank(self, rank: 'int'):
+        """Sets the rank of this V1JobTiming.
+
+
+        :param rank: The rank of this V1JobTiming.  # noqa: E501
+        :type: int
+        """
+
+        self._rank = rank
+
     @property
     def start(self) -> 'datetime':
         """Gets the start of this V1JobTiming.  # noqa: E501