lightning-sdk 0.2.3__py3-none-any.whl → 0.2.5__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_schedule_resource_type.py

@@ -0,0 +1,103 @@
+# 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 V1ScheduleResourceType(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+
+    """
+    allowed enum values
+    """
+    UNSPECIFIED = "UNSPECIFIED"
+    PIPELINE = "PIPELINE"
+    """
+    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
+        """V1ScheduleResourceType - 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(V1ScheduleResourceType, 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: 'V1ScheduleResourceType') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1ScheduleResourceType):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1ScheduleResourceType') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_slurm_cluster_user.py

@@ -0,0 +1,227 @@
+# 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 V1SlurmClusterUser(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 = {
+        'cluster_id': 'str',
+        'home_directory': 'str',
+        'id': 'str',
+        'user_id': 'str',
+        'username': 'str'
+    }
+
+    attribute_map = {
+        'cluster_id': 'clusterId',
+        'home_directory': 'homeDirectory',
+        'id': 'id',
+        'user_id': 'userId',
+        'username': 'username'
+    }
+
+    def __init__(self, cluster_id: 'str' =None, home_directory: 'str' =None, id: 'str' =None, user_id: 'str' =None, username: 'str' =None):  # noqa: E501
+        """V1SlurmClusterUser - a model defined in Swagger"""  # noqa: E501
+        self._cluster_id = None
+        self._home_directory = None
+        self._id = None
+        self._user_id = None
+        self._username = None
+        self.discriminator = None
+        if cluster_id is not None:
+            self.cluster_id = cluster_id
+        if home_directory is not None:
+            self.home_directory = home_directory
+        if id is not None:
+            self.id = id
+        if user_id is not None:
+            self.user_id = user_id
+        if username is not None:
+            self.username = username
+
+    @property
+    def cluster_id(self) -> 'str':
+        """Gets the cluster_id of this V1SlurmClusterUser.  # noqa: E501
+
+
+        :return: The cluster_id of this V1SlurmClusterUser.  # noqa: E501
+        :rtype: str
+        """
+        return self._cluster_id
+
+    @cluster_id.setter
+    def cluster_id(self, cluster_id: 'str'):
+        """Sets the cluster_id of this V1SlurmClusterUser.
+
+
+        :param cluster_id: The cluster_id of this V1SlurmClusterUser.  # noqa: E501
+        :type: str
+        """
+
+        self._cluster_id = cluster_id
+
+    @property
+    def home_directory(self) -> 'str':
+        """Gets the home_directory of this V1SlurmClusterUser.  # noqa: E501
+
+
+        :return: The home_directory of this V1SlurmClusterUser.  # noqa: E501
+        :rtype: str
+        """
+        return self._home_directory
+
+    @home_directory.setter
+    def home_directory(self, home_directory: 'str'):
+        """Sets the home_directory of this V1SlurmClusterUser.
+
+
+        :param home_directory: The home_directory of this V1SlurmClusterUser.  # noqa: E501
+        :type: str
+        """
+
+        self._home_directory = home_directory
+
+    @property
+    def id(self) -> 'str':
+        """Gets the id of this V1SlurmClusterUser.  # noqa: E501
+
+
+        :return: The id of this V1SlurmClusterUser.  # noqa: E501
+        :rtype: str
+        """
+        return self._id
+
+    @id.setter
+    def id(self, id: 'str'):
+        """Sets the id of this V1SlurmClusterUser.
+
+
+        :param id: The id of this V1SlurmClusterUser.  # noqa: E501
+        :type: str
+        """
+
+        self._id = id
+
+    @property
+    def user_id(self) -> 'str':
+        """Gets the user_id of this V1SlurmClusterUser.  # noqa: E501
+
+
+        :return: The user_id of this V1SlurmClusterUser.  # noqa: E501
+        :rtype: str
+        """
+        return self._user_id
+
+    @user_id.setter
+    def user_id(self, user_id: 'str'):
+        """Sets the user_id of this V1SlurmClusterUser.
+
+
+        :param user_id: The user_id of this V1SlurmClusterUser.  # noqa: E501
+        :type: str
+        """
+
+        self._user_id = user_id
+
+    @property
+    def username(self) -> 'str':
+        """Gets the username of this V1SlurmClusterUser.  # noqa: E501
+
+
+        :return: The username of this V1SlurmClusterUser.  # noqa: E501
+        :rtype: str
+        """
+        return self._username
+
+    @username.setter
+    def username(self, username: 'str'):
+        """Sets the username of this V1SlurmClusterUser.
+
+
+        :param username: The username of this V1SlurmClusterUser.  # noqa: E501
+        :type: str
+        """
+
+        self._username = username
+
+    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(V1SlurmClusterUser, 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: 'V1SlurmClusterUser') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1SlurmClusterUser):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1SlurmClusterUser') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_slurm_job.py

@@ -46,13 +46,15 @@ class V1SLURMJob(object):
         'cluster_id': 'str',
         'command': 'str',
         'created_at': 'datetime',
+        'entrypoint': 'str',
         'id': 'str',
         'image': 'str',
         'linked_project_id': 'str',
         'linked_user_id': 'str',
         'message': 'str',
         'name': 'str',
-        'num_gpus': 'str',
+        'num_gpus': 'int',
+        'partition': 'str',
         'project_id': 'str',
         'run_id': 'str',
         'service_id': 'str',
@@ -72,6 +74,7 @@ class V1SLURMJob(object):
         'cluster_id': 'clusterId',
         'command': 'command',
         'created_at': 'createdAt',
+        'entrypoint': 'entrypoint',
         'id': 'id',
         'image': 'image',
         'linked_project_id': 'linkedProjectId',
@@ -79,6 +82,7 @@ class V1SLURMJob(object):
         'message': 'message',
         'name': 'name',
         'num_gpus': 'numGpus',
+        'partition': 'partition',
         'project_id': 'projectId',
         'run_id': 'runId',
         'service_id': 'serviceId',
@@ -92,13 +96,14 @@ 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, 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
+    def __init__(self, cache_id: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, created_at: 'datetime' =None, entrypoint: 'str' =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: 'int' =None, partition: '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
         self._cluster_id = None
         self._command = None
         self._created_at = None
+        self._entrypoint = None
         self._id = None
         self._image = None
         self._linked_project_id = None
@@ -106,6 +111,7 @@ class V1SLURMJob(object):
         self._message = None
         self._name = None
         self._num_gpus = None
+        self._partition = None
         self._project_id = None
         self._run_id = None
         self._service_id = None
@@ -128,6 +134,8 @@ class V1SLURMJob(object):
             self.command = command
         if created_at is not None:
             self.created_at = created_at
+        if entrypoint is not None:
+            self.entrypoint = entrypoint
         if id is not None:
             self.id = id
         if image is not None:
@@ -142,6 +150,8 @@ class V1SLURMJob(object):
             self.name = name
         if num_gpus is not None:
             self.num_gpus = num_gpus
+        if partition is not None:
+            self.partition = partition
         if project_id is not None:
             self.project_id = project_id
         if run_id is not None:
@@ -270,6 +280,27 @@ class V1SLURMJob(object):
 
         self._created_at = created_at
 
+    @property
+    def entrypoint(self) -> 'str':
+        """Gets the entrypoint of this V1SLURMJob.  # noqa: E501
+
+
+        :return: The entrypoint of this V1SLURMJob.  # noqa: E501
+        :rtype: str
+        """
+        return self._entrypoint
+
+    @entrypoint.setter
+    def entrypoint(self, entrypoint: 'str'):
+        """Sets the entrypoint of this V1SLURMJob.
+
+
+        :param entrypoint: The entrypoint of this V1SLURMJob.  # noqa: E501
+        :type: str
+        """
+
+        self._entrypoint = entrypoint
+
     @property
     def id(self) -> 'str':
         """Gets the id of this V1SLURMJob.  # noqa: E501
@@ -397,26 +428,47 @@ class V1SLURMJob(object):
         self._name = name
 
     @property
-    def num_gpus(self) -> 'str':
+    def num_gpus(self) -> 'int':
         """Gets the num_gpus of this V1SLURMJob.  # noqa: E501
 
 
         :return: The num_gpus of this V1SLURMJob.  # noqa: E501
-        :rtype: str
+        :rtype: int
         """
         return self._num_gpus
 
     @num_gpus.setter
-    def num_gpus(self, num_gpus: 'str'):
+    def num_gpus(self, num_gpus: 'int'):
         """Sets the num_gpus of this V1SLURMJob.
 
 
         :param num_gpus: The num_gpus of this V1SLURMJob.  # noqa: E501
-        :type: str
+        :type: int
         """
 
         self._num_gpus = num_gpus
 
+    @property
+    def partition(self) -> 'str':
+        """Gets the partition of this V1SLURMJob.  # noqa: E501
+
+
+        :return: The partition of this V1SLURMJob.  # noqa: E501
+        :rtype: str
+        """
+        return self._partition
+
+    @partition.setter
+    def partition(self, partition: 'str'):
+        """Sets the partition of this V1SLURMJob.
+
+
+        :param partition: The partition of this V1SLURMJob.  # noqa: E501
+        :type: str
+        """
+
+        self._partition = partition
+
     @property
     def project_id(self) -> 'str':
         """Gets the project_id of this V1SLURMJob.  # noqa: E501