lightning-sdk 0.2.14__py3-none-any.whl → 0.2.15__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_organization.py

@@ -44,10 +44,12 @@ class V1Organization(object):
         'alerts_config': 'V1AlertsConfig',
         'allow_aws_saas': 'bool',
         'allow_budgeting': 'bool',
+        'allow_dgx_saas': 'bool',
         'allow_external_project_duplication': 'bool',
         'allow_gcp_saas': 'bool',
         'allow_guest': 'bool',
         'allow_lambda_saas': 'bool',
+        'allow_lightning_saas': 'bool',
         'allow_marketplace': 'bool',
         'allow_member_invitations': 'bool',
         'allow_member_teamspace_creation': 'bool',
@@ -55,6 +57,7 @@ class V1Organization(object):
         'auto_invite_by_domain': 'bool',
         'auto_join_domain_validations': 'dict(str, V1AutoJoinDomainValidation)',
         'auto_join_domains': 'list[str]',
+        'auto_switch_machine': 'bool',
         'created_at': 'datetime',
         'default_machine_image_version': 'str',
         'default_machine_type': 'str',
@@ -83,10 +86,12 @@ class V1Organization(object):
         'alerts_config': 'alertsConfig',
         'allow_aws_saas': 'allowAwsSaas',
         'allow_budgeting': 'allowBudgeting',
+        'allow_dgx_saas': 'allowDgxSaas',
         'allow_external_project_duplication': 'allowExternalProjectDuplication',
         'allow_gcp_saas': 'allowGcpSaas',
         'allow_guest': 'allowGuest',
         'allow_lambda_saas': 'allowLambdaSaas',
+        'allow_lightning_saas': 'allowLightningSaas',
         'allow_marketplace': 'allowMarketplace',
         'allow_member_invitations': 'allowMemberInvitations',
         'allow_member_teamspace_creation': 'allowMemberTeamspaceCreation',
@@ -94,6 +99,7 @@ class V1Organization(object):
         'auto_invite_by_domain': 'autoInviteByDomain',
         'auto_join_domain_validations': 'autoJoinDomainValidations',
         'auto_join_domains': 'autoJoinDomains',
+        'auto_switch_machine': 'autoSwitchMachine',
         'created_at': 'createdAt',
         'default_machine_image_version': 'defaultMachineImageVersion',
         'default_machine_type': 'defaultMachineType',
@@ -118,15 +124,17 @@ class V1Organization(object):
         'workload_max_run_duration': 'workloadMaxRunDuration'
     }
 
-    def __init__(self, alerts_config: 'V1AlertsConfig' =None, allow_aws_saas: 'bool' =None, allow_budgeting: 'bool' =None, allow_external_project_duplication: 'bool' =None, allow_gcp_saas: 'bool' =None, allow_guest: 'bool' =None, allow_lambda_saas: 'bool' =None, allow_marketplace: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, allow_vultr_saas: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domain_validations: 'dict(str, V1AutoJoinDomainValidation)' =None, auto_join_domains: 'list[str]' =None, created_at: 'datetime' =None, default_machine_image_version: 'str' =None, default_machine_type: 'str' =None, description: 'str' =None, display_name: 'str' =None, domain: 'str' =None, email: 'str' =None, featured_gallery: 'bool' =None, full_story_end_date: 'datetime' =None, full_story_start_date: 'datetime' =None, general_teamspace: 'bool' =None, id: 'str' =None, location: 'str' =None, name: 'str' =None, preferred_cluster: 'str' =None, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, start_studios_on_spot_instance: 'bool' =None, teamspace_default_credits: 'float' =None, twitter_username: 'str' =None, updated_at: 'datetime' =None, workload_max_run_duration: 'str' =None):  # noqa: E501
+    def __init__(self, alerts_config: 'V1AlertsConfig' =None, allow_aws_saas: 'bool' =None, allow_budgeting: 'bool' =None, allow_dgx_saas: 'bool' =None, allow_external_project_duplication: 'bool' =None, allow_gcp_saas: 'bool' =None, allow_guest: 'bool' =None, allow_lambda_saas: 'bool' =None, allow_lightning_saas: 'bool' =None, allow_marketplace: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, allow_vultr_saas: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domain_validations: 'dict(str, V1AutoJoinDomainValidation)' =None, auto_join_domains: 'list[str]' =None, auto_switch_machine: 'bool' =None, created_at: 'datetime' =None, default_machine_image_version: 'str' =None, default_machine_type: 'str' =None, description: 'str' =None, display_name: 'str' =None, domain: 'str' =None, email: 'str' =None, featured_gallery: 'bool' =None, full_story_end_date: 'datetime' =None, full_story_start_date: 'datetime' =None, general_teamspace: 'bool' =None, id: 'str' =None, location: 'str' =None, name: 'str' =None, preferred_cluster: 'str' =None, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, start_studios_on_spot_instance: 'bool' =None, teamspace_default_credits: 'float' =None, twitter_username: 'str' =None, updated_at: 'datetime' =None, workload_max_run_duration: 'str' =None):  # noqa: E501
         """V1Organization - a model defined in Swagger"""  # noqa: E501
         self._alerts_config = None
         self._allow_aws_saas = None
         self._allow_budgeting = None
+        self._allow_dgx_saas = None
         self._allow_external_project_duplication = None
         self._allow_gcp_saas = None
         self._allow_guest = None
         self._allow_lambda_saas = None
+        self._allow_lightning_saas = None
         self._allow_marketplace = None
         self._allow_member_invitations = None
         self._allow_member_teamspace_creation = None
@@ -134,6 +142,7 @@ class V1Organization(object):
         self._auto_invite_by_domain = None
         self._auto_join_domain_validations = None
         self._auto_join_domains = None
+        self._auto_switch_machine = None
         self._created_at = None
         self._default_machine_image_version = None
         self._default_machine_type = None
@@ -163,6 +172,8 @@ class V1Organization(object):
             self.allow_aws_saas = allow_aws_saas
         if allow_budgeting is not None:
             self.allow_budgeting = allow_budgeting
+        if allow_dgx_saas is not None:
+            self.allow_dgx_saas = allow_dgx_saas
         if allow_external_project_duplication is not None:
             self.allow_external_project_duplication = allow_external_project_duplication
         if allow_gcp_saas is not None:
@@ -171,6 +182,8 @@ class V1Organization(object):
             self.allow_guest = allow_guest
         if allow_lambda_saas is not None:
             self.allow_lambda_saas = allow_lambda_saas
+        if allow_lightning_saas is not None:
+            self.allow_lightning_saas = allow_lightning_saas
         if allow_marketplace is not None:
             self.allow_marketplace = allow_marketplace
         if allow_member_invitations is not None:
@@ -185,6 +198,8 @@ class V1Organization(object):
             self.auto_join_domain_validations = auto_join_domain_validations
         if auto_join_domains is not None:
             self.auto_join_domains = auto_join_domains
+        if auto_switch_machine is not None:
+            self.auto_switch_machine = auto_switch_machine
         if created_at is not None:
             self.created_at = created_at
         if default_machine_image_version is not None:
@@ -293,6 +308,27 @@ class V1Organization(object):
 
         self._allow_budgeting = allow_budgeting
 
+    @property
+    def allow_dgx_saas(self) -> 'bool':
+        """Gets the allow_dgx_saas of this V1Organization.  # noqa: E501
+
+
+        :return: The allow_dgx_saas of this V1Organization.  # noqa: E501
+        :rtype: bool
+        """
+        return self._allow_dgx_saas
+
+    @allow_dgx_saas.setter
+    def allow_dgx_saas(self, allow_dgx_saas: 'bool'):
+        """Sets the allow_dgx_saas of this V1Organization.
+
+
+        :param allow_dgx_saas: The allow_dgx_saas of this V1Organization.  # noqa: E501
+        :type: bool
+        """
+
+        self._allow_dgx_saas = allow_dgx_saas
+
     @property
     def allow_external_project_duplication(self) -> 'bool':
         """Gets the allow_external_project_duplication of this V1Organization.  # noqa: E501
@@ -379,6 +415,27 @@ class V1Organization(object):
 
         self._allow_lambda_saas = allow_lambda_saas
 
+    @property
+    def allow_lightning_saas(self) -> 'bool':
+        """Gets the allow_lightning_saas of this V1Organization.  # noqa: E501
+
+
+        :return: The allow_lightning_saas of this V1Organization.  # noqa: E501
+        :rtype: bool
+        """
+        return self._allow_lightning_saas
+
+    @allow_lightning_saas.setter
+    def allow_lightning_saas(self, allow_lightning_saas: 'bool'):
+        """Sets the allow_lightning_saas of this V1Organization.
+
+
+        :param allow_lightning_saas: The allow_lightning_saas of this V1Organization.  # noqa: E501
+        :type: bool
+        """
+
+        self._allow_lightning_saas = allow_lightning_saas
+
     @property
     def allow_marketplace(self) -> 'bool':
         """Gets the allow_marketplace of this V1Organization.  # noqa: E501
@@ -526,6 +583,27 @@ class V1Organization(object):
 
         self._auto_join_domains = auto_join_domains
 
+    @property
+    def auto_switch_machine(self) -> 'bool':
+        """Gets the auto_switch_machine of this V1Organization.  # noqa: E501
+
+
+        :return: The auto_switch_machine of this V1Organization.  # noqa: E501
+        :rtype: bool
+        """
+        return self._auto_switch_machine
+
+    @auto_switch_machine.setter
+    def auto_switch_machine(self, auto_switch_machine: 'bool'):
+        """Sets the auto_switch_machine of this V1Organization.
+
+
+        :param auto_switch_machine: The auto_switch_machine of this V1Organization.  # noqa: E501
+        :type: bool
+        """
+
+        self._auto_switch_machine = auto_switch_machine
+
     @property
     def created_at(self) -> 'datetime':
         """Gets the created_at of this V1Organization.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py

@@ -54,7 +54,7 @@ class V1Pipeline(object):
         'project_id': 'str',
         'schedule_id': 'str',
         'shared_filesystem': 'V1SharedFilesystem',
-        'state': 'str',
+        'state': 'V1PipelineState',
         'statuses': 'list[V1PipelineStepStatus]',
         'steps': 'list[V1PipelineStep]',
         'updated_at': 'datetime',
@@ -82,7 +82,7 @@ class V1Pipeline(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, cluster_id: 'str' =None, created_at: 'datetime' =None, display_name: 'str' =None, error: 'str' =None, id: 'str' =None, is_published: 'bool' =None, message: 'str' =None, name: 'str' =None, parameters: 'list[V1PipelineParameter]' =None, parent_pipeline_id: 'str' =None, project_id: 'str' =None, schedule_id: 'str' =None, shared_filesystem: 'V1SharedFilesystem' =None, state: 'str' =None, statuses: 'list[V1PipelineStepStatus]' =None, steps: 'list[V1PipelineStep]' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, cluster_id: 'str' =None, created_at: 'datetime' =None, display_name: 'str' =None, error: 'str' =None, id: 'str' =None, is_published: 'bool' =None, message: 'str' =None, name: 'str' =None, parameters: 'list[V1PipelineParameter]' =None, parent_pipeline_id: 'str' =None, project_id: 'str' =None, schedule_id: 'str' =None, shared_filesystem: 'V1SharedFilesystem' =None, state: 'V1PipelineState' =None, statuses: 'list[V1PipelineStepStatus]' =None, steps: 'list[V1PipelineStep]' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """V1Pipeline - a model defined in Swagger"""  # noqa: E501
         self._cluster_id = None
         self._created_at = None
@@ -414,22 +414,22 @@ class V1Pipeline(object):
         self._shared_filesystem = shared_filesystem
 
     @property
-    def state(self) -> 'str':
+    def state(self) -> 'V1PipelineState':
         """Gets the state of this V1Pipeline.  # noqa: E501
 
 
         :return: The state of this V1Pipeline.  # noqa: E501
-        :rtype: str
+        :rtype: V1PipelineState
         """
         return self._state
 
     @state.setter
-    def state(self, state: 'str'):
+    def state(self, state: 'V1PipelineState'):
         """Sets the state of this V1Pipeline.
 
 
         :param state: The state of this V1Pipeline.  # noqa: E501
-        :type: str
+        :type: V1PipelineState
         """
 
         self._state = state

lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_state.py

@@ -0,0 +1,111 @@
+# 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 V1PipelineState(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+
+    """
+    allowed enum values
+    """
+    UNSPECIFIED = "PIPELINE_STATE_UNSPECIFIED"
+    PENDING = "PIPELINE_STATE_PENDING"
+    RUNNING = "PIPELINE_STATE_RUNNING"
+    FAILED = "PIPELINE_STATE_FAILED"
+    DELETED = "PIPELINE_STATE_DELETED"
+    STOPPED = "PIPELINE_STATE_STOPPED"
+    SUSPENDED = "PIPELINE_STATE_SUSPENDED"
+    COMPLETED = "PIPELINE_STATE_COMPLETED"
+    STOP = "PIPELINE_STATE_STOP"
+    DELETE = "PIPELINE_STATE_DELETE"
+    """
+    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
+        """V1PipelineState - 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(V1PipelineState, 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: 'V1PipelineState') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1PipelineState):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1PipelineState') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_presigned_url.py

@@ -41,25 +41,56 @@ class V1PresignedUrl(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'cluster_id': 'str',
         'part_number': 'str',
+        'upload_id': 'str',
         'url': 'str'
     }
 
     attribute_map = {
+        'cluster_id': 'clusterId',
         'part_number': 'partNumber',
+        'upload_id': 'uploadId',
         'url': 'url'
     }
 
-    def __init__(self, part_number: 'str' =None, url: 'str' =None):  # noqa: E501
+    def __init__(self, cluster_id: 'str' =None, part_number: 'str' =None, upload_id: 'str' =None, url: 'str' =None):  # noqa: E501
         """V1PresignedUrl - a model defined in Swagger"""  # noqa: E501
+        self._cluster_id = None
         self._part_number = None
+        self._upload_id = None
         self._url = None
         self.discriminator = None
+        if cluster_id is not None:
+            self.cluster_id = cluster_id
         if part_number is not None:
             self.part_number = part_number
+        if upload_id is not None:
+            self.upload_id = upload_id
         if url is not None:
             self.url = url
 
+    @property
+    def cluster_id(self) -> 'str':
+        """Gets the cluster_id of this V1PresignedUrl.  # noqa: E501
+
+
+        :return: The cluster_id of this V1PresignedUrl.  # noqa: E501
+        :rtype: str
+        """
+        return self._cluster_id
+
+    @cluster_id.setter
+    def cluster_id(self, cluster_id: 'str'):
+        """Sets the cluster_id of this V1PresignedUrl.
+
+
+        :param cluster_id: The cluster_id of this V1PresignedUrl.  # noqa: E501
+        :type: str
+        """
+
+        self._cluster_id = cluster_id
+
     @property
     def part_number(self) -> 'str':
         """Gets the part_number of this V1PresignedUrl.  # noqa: E501
@@ -81,6 +112,27 @@ class V1PresignedUrl(object):
 
         self._part_number = part_number
 
+    @property
+    def upload_id(self) -> 'str':
+        """Gets the upload_id of this V1PresignedUrl.  # noqa: E501
+
+
+        :return: The upload_id of this V1PresignedUrl.  # noqa: E501
+        :rtype: str
+        """
+        return self._upload_id
+
+    @upload_id.setter
+    def upload_id(self, upload_id: 'str'):
+        """Sets the upload_id of this V1PresignedUrl.
+
+
+        :param upload_id: The upload_id of this V1PresignedUrl.  # noqa: E501
+        :type: str
+        """
+
+        self._upload_id = upload_id
+
     @property
     def url(self) -> 'str':
         """Gets the url of this V1PresignedUrl.  # noqa: E501