lightning-sdk 0.1.58__py3-none-any.whl → 0.2.0__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py

@@ -41,30 +41,144 @@ class V1ProjectSettings(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'allow_aws_saas': 'bool',
+        'allow_gcp_saas': 'bool',
+        'allow_lambda_saas': 'bool',
+        'allow_vultr_saas': 'bool',
         'preferred_cluster': 'str',
+        'preferred_deployment_provider': 'str',
+        'preferred_studio_provider': 'str',
         'same_compute_on_resume': 'bool',
         'start_studio_on_spot_instance': 'bool'
     }
 
     attribute_map = {
+        'allow_aws_saas': 'allowAwsSaas',
+        'allow_gcp_saas': 'allowGcpSaas',
+        'allow_lambda_saas': 'allowLambdaSaas',
+        'allow_vultr_saas': 'allowVultrSaas',
         'preferred_cluster': 'preferredCluster',
+        'preferred_deployment_provider': 'preferredDeploymentProvider',
+        'preferred_studio_provider': 'preferredStudioProvider',
         'same_compute_on_resume': 'sameComputeOnResume',
         'start_studio_on_spot_instance': 'startStudioOnSpotInstance'
     }
 
-    def __init__(self, preferred_cluster: 'str' =None, same_compute_on_resume: 'bool' =None, start_studio_on_spot_instance: 'bool' =None):  # noqa: E501
+    def __init__(self, allow_aws_saas: 'bool' =None, allow_gcp_saas: 'bool' =None, allow_lambda_saas: 'bool' =None, allow_vultr_saas: 'bool' =None, preferred_cluster: 'str' =None, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, same_compute_on_resume: 'bool' =None, start_studio_on_spot_instance: 'bool' =None):  # noqa: E501
         """V1ProjectSettings - a model defined in Swagger"""  # noqa: E501
+        self._allow_aws_saas = None
+        self._allow_gcp_saas = None
+        self._allow_lambda_saas = None
+        self._allow_vultr_saas = None
         self._preferred_cluster = None
+        self._preferred_deployment_provider = None
+        self._preferred_studio_provider = None
         self._same_compute_on_resume = None
         self._start_studio_on_spot_instance = None
         self.discriminator = None
+        if allow_aws_saas is not None:
+            self.allow_aws_saas = allow_aws_saas
+        if allow_gcp_saas is not None:
+            self.allow_gcp_saas = allow_gcp_saas
+        if allow_lambda_saas is not None:
+            self.allow_lambda_saas = allow_lambda_saas
+        if allow_vultr_saas is not None:
+            self.allow_vultr_saas = allow_vultr_saas
         if preferred_cluster is not None:
             self.preferred_cluster = preferred_cluster
+        if preferred_deployment_provider is not None:
+            self.preferred_deployment_provider = preferred_deployment_provider
+        if preferred_studio_provider is not None:
+            self.preferred_studio_provider = preferred_studio_provider
         if same_compute_on_resume is not None:
             self.same_compute_on_resume = same_compute_on_resume
         if start_studio_on_spot_instance is not None:
             self.start_studio_on_spot_instance = start_studio_on_spot_instance
 
+    @property
+    def allow_aws_saas(self) -> 'bool':
+        """Gets the allow_aws_saas of this V1ProjectSettings.  # noqa: E501
+
+
+        :return: The allow_aws_saas of this V1ProjectSettings.  # noqa: E501
+        :rtype: bool
+        """
+        return self._allow_aws_saas
+
+    @allow_aws_saas.setter
+    def allow_aws_saas(self, allow_aws_saas: 'bool'):
+        """Sets the allow_aws_saas of this V1ProjectSettings.
+
+
+        :param allow_aws_saas: The allow_aws_saas of this V1ProjectSettings.  # noqa: E501
+        :type: bool
+        """
+
+        self._allow_aws_saas = allow_aws_saas
+
+    @property
+    def allow_gcp_saas(self) -> 'bool':
+        """Gets the allow_gcp_saas of this V1ProjectSettings.  # noqa: E501
+
+
+        :return: The allow_gcp_saas of this V1ProjectSettings.  # noqa: E501
+        :rtype: bool
+        """
+        return self._allow_gcp_saas
+
+    @allow_gcp_saas.setter
+    def allow_gcp_saas(self, allow_gcp_saas: 'bool'):
+        """Sets the allow_gcp_saas of this V1ProjectSettings.
+
+
+        :param allow_gcp_saas: The allow_gcp_saas of this V1ProjectSettings.  # noqa: E501
+        :type: bool
+        """
+
+        self._allow_gcp_saas = allow_gcp_saas
+
+    @property
+    def allow_lambda_saas(self) -> 'bool':
+        """Gets the allow_lambda_saas of this V1ProjectSettings.  # noqa: E501
+
+
+        :return: The allow_lambda_saas of this V1ProjectSettings.  # noqa: E501
+        :rtype: bool
+        """
+        return self._allow_lambda_saas
+
+    @allow_lambda_saas.setter
+    def allow_lambda_saas(self, allow_lambda_saas: 'bool'):
+        """Sets the allow_lambda_saas of this V1ProjectSettings.
+
+
+        :param allow_lambda_saas: The allow_lambda_saas of this V1ProjectSettings.  # noqa: E501
+        :type: bool
+        """
+
+        self._allow_lambda_saas = allow_lambda_saas
+
+    @property
+    def allow_vultr_saas(self) -> 'bool':
+        """Gets the allow_vultr_saas of this V1ProjectSettings.  # noqa: E501
+
+
+        :return: The allow_vultr_saas of this V1ProjectSettings.  # noqa: E501
+        :rtype: bool
+        """
+        return self._allow_vultr_saas
+
+    @allow_vultr_saas.setter
+    def allow_vultr_saas(self, allow_vultr_saas: 'bool'):
+        """Sets the allow_vultr_saas of this V1ProjectSettings.
+
+
+        :param allow_vultr_saas: The allow_vultr_saas of this V1ProjectSettings.  # noqa: E501
+        :type: bool
+        """
+
+        self._allow_vultr_saas = allow_vultr_saas
+
     @property
     def preferred_cluster(self) -> 'str':
         """Gets the preferred_cluster of this V1ProjectSettings.  # noqa: E501
@@ -86,6 +200,48 @@ class V1ProjectSettings(object):
 
         self._preferred_cluster = preferred_cluster
 
+    @property
+    def preferred_deployment_provider(self) -> 'str':
+        """Gets the preferred_deployment_provider of this V1ProjectSettings.  # noqa: E501
+
+
+        :return: The preferred_deployment_provider of this V1ProjectSettings.  # noqa: E501
+        :rtype: str
+        """
+        return self._preferred_deployment_provider
+
+    @preferred_deployment_provider.setter
+    def preferred_deployment_provider(self, preferred_deployment_provider: 'str'):
+        """Sets the preferred_deployment_provider of this V1ProjectSettings.
+
+
+        :param preferred_deployment_provider: The preferred_deployment_provider of this V1ProjectSettings.  # noqa: E501
+        :type: str
+        """
+
+        self._preferred_deployment_provider = preferred_deployment_provider
+
+    @property
+    def preferred_studio_provider(self) -> 'str':
+        """Gets the preferred_studio_provider of this V1ProjectSettings.  # noqa: E501
+
+
+        :return: The preferred_studio_provider of this V1ProjectSettings.  # noqa: E501
+        :rtype: str
+        """
+        return self._preferred_studio_provider
+
+    @preferred_studio_provider.setter
+    def preferred_studio_provider(self, preferred_studio_provider: 'str'):
+        """Sets the preferred_studio_provider of this V1ProjectSettings.
+
+
+        :param preferred_studio_provider: The preferred_studio_provider of this V1ProjectSettings.  # noqa: E501
+        :type: str
+        """
+
+        self._preferred_studio_provider = preferred_studio_provider
+
     @property
     def same_compute_on_resume(self) -> 'bool':
         """Gets the same_compute_on_resume of this V1ProjectSettings.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_restart_timing.py

@@ -43,25 +43,30 @@ class V1RestartTiming(object):
     swagger_types = {
         'counter': 'int',
         'end': 'datetime',
+        'rank': 'int',
         'start': 'datetime'
     }
 
     attribute_map = {
         'counter': 'counter',
         'end': 'end',
+        'rank': 'rank',
         'start': 'start'
     }
 
-    def __init__(self, counter: 'int' =None, end: 'datetime' =None, start: 'datetime' =None):  # noqa: E501
+    def __init__(self, counter: 'int' =None, end: 'datetime' =None, rank: 'int' =None, start: 'datetime' =None):  # noqa: E501
         """V1RestartTiming - a model defined in Swagger"""  # noqa: E501
         self._counter = None
         self._end = None
+        self._rank = None
         self._start = None
         self.discriminator = None
         if counter is not None:
             self.counter = counter
         if end is not None:
             self.end = end
+        if rank is not None:
+            self.rank = rank
         if start is not None:
             self.start = start
 
@@ -107,6 +112,27 @@ class V1RestartTiming(object):
 
         self._end = end
 
+    @property
+    def rank(self) -> 'int':
+        """Gets the rank of this V1RestartTiming.  # noqa: E501
+
+
+        :return: The rank of this V1RestartTiming.  # noqa: E501
+        :rtype: int
+        """
+        return self._rank
+
+    @rank.setter
+    def rank(self, rank: 'int'):
+        """Sets the rank of this V1RestartTiming.
+
+
+        :param rank: The rank of this V1RestartTiming.  # noqa: E501
+        :type: int
+        """
+
+        self._rank = rank
+
     @property
     def start(self) -> 'datetime':
         """Gets the start of this V1RestartTiming.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py

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

lightning_sdk/lightning_cloud/openapi/models/v1_shared_filesystem.py

@@ -0,0 +1,201 @@
+# 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 V1SharedFilesystem(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 = {
+        'enabled': 'bool',
+        'resource_id': 'str',
+        'resource_name': 'str',
+        'state': 'str'
+    }
+
+    attribute_map = {
+        'enabled': 'enabled',
+        'resource_id': 'resourceId',
+        'resource_name': 'resourceName',
+        'state': 'state'
+    }
+
+    def __init__(self, enabled: 'bool' =None, resource_id: 'str' =None, resource_name: 'str' =None, state: 'str' =None):  # noqa: E501
+        """V1SharedFilesystem - a model defined in Swagger"""  # noqa: E501
+        self._enabled = None
+        self._resource_id = None
+        self._resource_name = None
+        self._state = None
+        self.discriminator = None
+        if enabled is not None:
+            self.enabled = enabled
+        if resource_id is not None:
+            self.resource_id = resource_id
+        if resource_name is not None:
+            self.resource_name = resource_name
+        if state is not None:
+            self.state = state
+
+    @property
+    def enabled(self) -> 'bool':
+        """Gets the enabled of this V1SharedFilesystem.  # noqa: E501
+
+
+        :return: The enabled of this V1SharedFilesystem.  # noqa: E501
+        :rtype: bool
+        """
+        return self._enabled
+
+    @enabled.setter
+    def enabled(self, enabled: 'bool'):
+        """Sets the enabled of this V1SharedFilesystem.
+
+
+        :param enabled: The enabled of this V1SharedFilesystem.  # noqa: E501
+        :type: bool
+        """
+
+        self._enabled = enabled
+
+    @property
+    def resource_id(self) -> 'str':
+        """Gets the resource_id of this V1SharedFilesystem.  # noqa: E501
+
+
+        :return: The resource_id of this V1SharedFilesystem.  # noqa: E501
+        :rtype: str
+        """
+        return self._resource_id
+
+    @resource_id.setter
+    def resource_id(self, resource_id: 'str'):
+        """Sets the resource_id of this V1SharedFilesystem.
+
+
+        :param resource_id: The resource_id of this V1SharedFilesystem.  # noqa: E501
+        :type: str
+        """
+
+        self._resource_id = resource_id
+
+    @property
+    def resource_name(self) -> 'str':
+        """Gets the resource_name of this V1SharedFilesystem.  # noqa: E501
+
+
+        :return: The resource_name of this V1SharedFilesystem.  # noqa: E501
+        :rtype: str
+        """
+        return self._resource_name
+
+    @resource_name.setter
+    def resource_name(self, resource_name: 'str'):
+        """Sets the resource_name of this V1SharedFilesystem.
+
+
+        :param resource_name: The resource_name of this V1SharedFilesystem.  # noqa: E501
+        :type: str
+        """
+
+        self._resource_name = resource_name
+
+    @property
+    def state(self) -> 'str':
+        """Gets the state of this V1SharedFilesystem.  # noqa: E501
+
+
+        :return: The state of this V1SharedFilesystem.  # noqa: E501
+        :rtype: str
+        """
+        return self._state
+
+    @state.setter
+    def state(self, state: 'str'):
+        """Sets the state of this V1SharedFilesystem.
+
+
+        :param state: The state of this V1SharedFilesystem.  # noqa: E501
+        :type: str
+        """
+
+        self._state = state
+
+    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(V1SharedFilesystem, 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: 'V1SharedFilesystem') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1SharedFilesystem):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1SharedFilesystem') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_slurm_job.py

@@ -47,6 +47,7 @@ class V1SLURMJob(object):
         'command': 'str',
         'created_at': 'datetime',
         'id': 'str',
+        'image': 'str',
         'linked_project_id': 'str',
         'linked_user_id': 'str',
         'message': 'str',
@@ -72,6 +73,7 @@ class V1SLURMJob(object):
         'command': 'command',
         'created_at': 'createdAt',
         'id': 'id',
+        'image': 'image',
         'linked_project_id': 'linkedProjectId',
         'linked_user_id': 'linkedUserId',
         'message': 'message',
@@ -90,7 +92,7 @@ class V1SLURMJob(object):
         'work_dir': 'workDir'
     }
 
-    def __init__(self, cache_id: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, created_at: 'datetime' =None, id: 'str' =None, linked_project_id: 'str' =None, linked_user_id: 'str' =None, message: 'str' =None, name: 'str' =None, num_gpus: 'str' =None, project_id: 'str' =None, run_id: 'str' =None, service_id: 'str' =None, slurm_v1_status: 'V1SlurmV1JobStatus' =None, state: 'str' =None, updated_at: 'datetime' =None, upload_eta_seconds: 'str' =None, upload_in_progress: 'bool' =None, upload_percentage: 'str' =None, user_id: 'str' =None, work_dir: 'str' =None):  # noqa: E501
+    def __init__(self, cache_id: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, created_at: 'datetime' =None, id: 'str' =None, image: 'str' =None, linked_project_id: 'str' =None, linked_user_id: 'str' =None, message: 'str' =None, name: 'str' =None, num_gpus: 'str' =None, project_id: 'str' =None, run_id: 'str' =None, service_id: 'str' =None, slurm_v1_status: 'V1SlurmV1JobStatus' =None, state: 'str' =None, updated_at: 'datetime' =None, upload_eta_seconds: 'str' =None, upload_in_progress: 'bool' =None, upload_percentage: 'str' =None, user_id: 'str' =None, work_dir: 'str' =None):  # noqa: E501
         """V1SLURMJob - a model defined in Swagger"""  # noqa: E501
         self._cache_id = None
         self._cloudspace_id = None
@@ -98,6 +100,7 @@ class V1SLURMJob(object):
         self._command = None
         self._created_at = None
         self._id = None
+        self._image = None
         self._linked_project_id = None
         self._linked_user_id = None
         self._message = None
@@ -127,6 +130,8 @@ class V1SLURMJob(object):
             self.created_at = created_at
         if id is not None:
             self.id = id
+        if image is not None:
+            self.image = image
         if linked_project_id is not None:
             self.linked_project_id = linked_project_id
         if linked_user_id is not None:
@@ -286,6 +291,27 @@ class V1SLURMJob(object):
 
         self._id = id
 
+    @property
+    def image(self) -> 'str':
+        """Gets the image of this V1SLURMJob.  # noqa: E501
+
+
+        :return: The image of this V1SLURMJob.  # noqa: E501
+        :rtype: str
+        """
+        return self._image
+
+    @image.setter
+    def image(self, image: 'str'):
+        """Sets the image of this V1SLURMJob.
+
+
+        :param image: The image of this V1SLURMJob.  # noqa: E501
+        :type: str
+        """
+
+        self._image = image
+
     @property
     def linked_project_id(self) -> 'str':
         """Gets the linked_project_id of this V1SLURMJob.  # noqa: E501

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