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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (110) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/ai_hub.py +10 -17
  3. lightning_sdk/api/ai_hub_api.py +20 -3
  4. lightning_sdk/api/studio_api.py +0 -8
  5. lightning_sdk/cli/serve.py +139 -22
  6. lightning_sdk/deployment/deployment.py +32 -4
  7. lightning_sdk/lightning_cloud/openapi/__init__.py +49 -1
  8. lightning_sdk/lightning_cloud/openapi/api/__init__.py +4 -0
  9. lightning_sdk/lightning_cloud/openapi/api/cloud_space_environment_template_service_api.py +537 -0
  10. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +10 -6
  11. lightning_sdk/lightning_cloud/openapi/api/lit_dataset_service_api.py +1973 -0
  12. lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py +5 -1
  13. lightning_sdk/lightning_cloud/openapi/api/models_store_api.py +20 -20
  14. lightning_sdk/lightning_cloud/openapi/api/pipeline_templates_service_api.py +339 -0
  15. lightning_sdk/lightning_cloud/openapi/api/pipelines_service_api.py +5 -1
  16. lightning_sdk/lightning_cloud/openapi/api/schedules_service_api.py +573 -0
  17. lightning_sdk/lightning_cloud/openapi/api/slurm_jobs_user_service_api.py +202 -0
  18. lightning_sdk/lightning_cloud/openapi/models/__init__.py +45 -1
  19. lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityblock_body.py +15 -15
  20. lightning_sdk/lightning_cloud/openapi/models/cluster_id_slurmusers_body.py +201 -0
  21. lightning_sdk/lightning_cloud/openapi/models/dataset_id_versions_body.py +149 -0
  22. lightning_sdk/lightning_cloud/openapi/models/dataset_id_visibility_body.py +149 -0
  23. lightning_sdk/lightning_cloud/openapi/models/environmenttemplates_id_body.py +227 -0
  24. lightning_sdk/lightning_cloud/openapi/models/externalv1_cloud_space_instance_status.py +69 -69
  25. lightning_sdk/lightning_cloud/openapi/models/litdatasets_dataset_id_body.py +149 -0
  26. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +27 -1
  27. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body.py +69 -17
  28. lightning_sdk/lightning_cloud/openapi/models/pipelinetemplates_id_body.py +331 -0
  29. lightning_sdk/lightning_cloud/openapi/models/project_id_litdatasets_body.py +227 -0
  30. lightning_sdk/lightning_cloud/openapi/models/project_id_pipelines_body.py +17 -17
  31. lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py +201 -0
  32. lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py +383 -0
  33. lightning_sdk/lightning_cloud/openapi/models/slurm_jobs_body.py +15 -15
  34. lightning_sdk/lightning_cloud/openapi/models/upload_id_complete_body1.py +149 -0
  35. lightning_sdk/lightning_cloud/openapi/models/upload_id_parts_body1.py +149 -0
  36. lightning_sdk/lightning_cloud/openapi/models/v1_agent_job.py +124 -20
  37. lightning_sdk/lightning_cloud/openapi/models/v1_assistant_model_status.py +2 -0
  38. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -0
  39. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +27 -1
  40. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template.py +253 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_complete_lit_dataset_multi_part_upload_response.py +97 -0
  43. lightning_sdk/lightning_cloud/openapi/models/v1_complete_lit_dataset_upload_response.py +97 -0
  44. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +149 -0
  45. lightning_sdk/lightning_cloud/openapi/models/v1_create_lit_dataset_multi_part_upload_response.py +123 -0
  46. lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py +27 -1
  47. lightning_sdk/lightning_cloud/openapi/models/v1_create_pipeline_template_request.py +383 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_create_project_request.py +27 -1
  49. lightning_sdk/lightning_cloud/openapi/models/{v1_pipeline_schedule.py → v1_delete_cloud_space_environment_template_response.py} +32 -32
  50. lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_dataset_response.py +97 -0
  51. lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_dataset_version_response.py +97 -0
  52. lightning_sdk/lightning_cloud/openapi/models/v1_delete_schedule_response.py +175 -0
  53. lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_size_response.py +79 -1
  54. lightning_sdk/lightning_cloud/openapi/models/v1_get_lit_dataset_file_upload_urls_response.py +123 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_get_lit_dataset_files_url_response.py +149 -0
  56. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
  57. lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.py +79 -27
  58. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +27 -1
  59. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloud_space_environment_templates_response.py +123 -0
  60. lightning_sdk/lightning_cloud/openapi/models/v1_list_lit_dataset_versions_response.py +123 -0
  61. lightning_sdk/lightning_cloud/openapi/models/v1_list_lit_datasets_response.py +123 -0
  62. lightning_sdk/lightning_cloud/openapi/models/v1_list_schedules_response.py +123 -0
  63. lightning_sdk/lightning_cloud/openapi/models/v1_list_slurm_cluster_users_response.py +123 -0
  64. lightning_sdk/lightning_cloud/openapi/models/v1_lit_dataset.py +539 -0
  65. lightning_sdk/lightning_cloud/openapi/models/v1_lit_dataset_file.py +175 -0
  66. lightning_sdk/lightning_cloud/openapi/models/v1_lit_dataset_version_archive.py +435 -0
  67. lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_project.py +27 -1
  68. lightning_sdk/lightning_cloud/openapi/models/v1_lit_repository.py +27 -1
  69. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_request.py +27 -1
  70. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +107 -3
  71. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +53 -1
  72. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py +69 -17
  73. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter.py +435 -0
  74. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter_placement.py +149 -0
  75. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter_placement_type.py +106 -0
  76. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter_type.py +106 -0
  77. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_template.py +513 -0
  78. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_template_visibility_type.py +105 -0
  79. lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py +27 -1
  80. lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py +435 -0
  81. lightning_sdk/lightning_cloud/openapi/models/v1_schedule_resource_type.py +103 -0
  82. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_cluster_user.py +227 -0
  83. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_job.py +58 -6
  84. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_node.py +31 -291
  85. lightning_sdk/lightning_cloud/openapi/models/v1_update_lit_dataset_visibility_response.py +123 -0
  86. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +27 -1
  87. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +157 -105
  88. lightning_sdk/lightning_cloud/openapi/models/v1_validate_managed_endpoint_response.py +27 -1
  89. lightning_sdk/lightning_cloud/openapi/models/v1_voltage_park_direct_v1.py +203 -0
  90. lightning_sdk/lightning_cloud/openapi/models/version_default_body.py +29 -29
  91. lightning_sdk/lightning_cloud/openapi/models/version_default_body1.py +149 -0
  92. lightning_sdk/lightning_cloud/openapi/models/version_uploads_body1.py +123 -0
  93. lightning_sdk/lightning_cloud/openapi/models/versions_version_body1.py +123 -0
  94. lightning_sdk/lightning_cloud/rest_client.py +2 -0
  95. lightning_sdk/lit_container.py +8 -1
  96. lightning_sdk/mmt/mmt.py +1 -18
  97. lightning_sdk/mmt/v1.py +1 -28
  98. lightning_sdk/models.py +15 -6
  99. lightning_sdk/pipeline/pipeline.py +2 -2
  100. lightning_sdk/pipeline/types.py +28 -2
  101. lightning_sdk/pipeline/utils.py +1 -1
  102. lightning_sdk/plugin.py +0 -6
  103. lightning_sdk/serve.py +55 -22
  104. lightning_sdk/utils/resolve.py +1 -0
  105. {lightning_sdk-0.2.3.dist-info → lightning_sdk-0.2.5.dist-info}/METADATA +1 -1
  106. {lightning_sdk-0.2.3.dist-info → lightning_sdk-0.2.5.dist-info}/RECORD +110 -62
  107. {lightning_sdk-0.2.3.dist-info → lightning_sdk-0.2.5.dist-info}/WHEEL +1 -1
  108. {lightning_sdk-0.2.3.dist-info → lightning_sdk-0.2.5.dist-info}/LICENSE +0 -0
  109. {lightning_sdk-0.2.3.dist-info → lightning_sdk-0.2.5.dist-info}/entry_points.txt +0 -0
  110. {lightning_sdk-0.2.3.dist-info → lightning_sdk-0.2.5.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,123 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ external/v1/auth_service.proto
5
+
6
+ No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen) # noqa: E501
7
+
8
+ OpenAPI spec version: version not set
9
+
10
+ Generated by: https://github.com/swagger-api/swagger-codegen.git
11
+
12
+ NOTE
13
+ ----
14
+ standard swagger-codegen-cli for this python client has been modified
15
+ by custom templates. The purpose of these templates is to include
16
+ typing information in the API and Model code. Please refer to the
17
+ main grid repository for more info
18
+ """
19
+
20
+ import pprint
21
+ import re # noqa: F401
22
+
23
+ from typing import TYPE_CHECKING
24
+
25
+ import six
26
+
27
+ if TYPE_CHECKING:
28
+ from datetime import datetime
29
+ from lightning_sdk.lightning_cloud.openapi.models import *
30
+
31
+ class V1GetLitDatasetFileUploadUrlsResponse(object):
32
+ """NOTE: This class is auto generated by the swagger code generator program.
33
+
34
+ Do not edit the class manually.
35
+ """
36
+ """
37
+ Attributes:
38
+ swagger_types (dict): The key is attribute name
39
+ and the value is attribute type.
40
+ attribute_map (dict): The key is attribute name
41
+ and the value is json key in definition.
42
+ """
43
+ swagger_types = {
44
+ 'urls': 'list[V1SignedUrl]'
45
+ }
46
+
47
+ attribute_map = {
48
+ 'urls': 'urls'
49
+ }
50
+
51
+ def __init__(self, urls: 'list[V1SignedUrl]' =None): # noqa: E501
52
+ """V1GetLitDatasetFileUploadUrlsResponse - a model defined in Swagger""" # noqa: E501
53
+ self._urls = None
54
+ self.discriminator = None
55
+ if urls is not None:
56
+ self.urls = urls
57
+
58
+ @property
59
+ def urls(self) -> 'list[V1SignedUrl]':
60
+ """Gets the urls of this V1GetLitDatasetFileUploadUrlsResponse. # noqa: E501
61
+
62
+
63
+ :return: The urls of this V1GetLitDatasetFileUploadUrlsResponse. # noqa: E501
64
+ :rtype: list[V1SignedUrl]
65
+ """
66
+ return self._urls
67
+
68
+ @urls.setter
69
+ def urls(self, urls: 'list[V1SignedUrl]'):
70
+ """Sets the urls of this V1GetLitDatasetFileUploadUrlsResponse.
71
+
72
+
73
+ :param urls: The urls of this V1GetLitDatasetFileUploadUrlsResponse. # noqa: E501
74
+ :type: list[V1SignedUrl]
75
+ """
76
+
77
+ self._urls = urls
78
+
79
+ def to_dict(self) -> dict:
80
+ """Returns the model properties as a dict"""
81
+ result = {}
82
+
83
+ for attr, _ in six.iteritems(self.swagger_types):
84
+ value = getattr(self, attr)
85
+ if isinstance(value, list):
86
+ result[attr] = list(map(
87
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
88
+ value
89
+ ))
90
+ elif hasattr(value, "to_dict"):
91
+ result[attr] = value.to_dict()
92
+ elif isinstance(value, dict):
93
+ result[attr] = dict(map(
94
+ lambda item: (item[0], item[1].to_dict())
95
+ if hasattr(item[1], "to_dict") else item,
96
+ value.items()
97
+ ))
98
+ else:
99
+ result[attr] = value
100
+ if issubclass(V1GetLitDatasetFileUploadUrlsResponse, dict):
101
+ for key, value in self.items():
102
+ result[key] = value
103
+
104
+ return result
105
+
106
+ def to_str(self) -> str:
107
+ """Returns the string representation of the model"""
108
+ return pprint.pformat(self.to_dict())
109
+
110
+ def __repr__(self) -> str:
111
+ """For `print` and `pprint`"""
112
+ return self.to_str()
113
+
114
+ def __eq__(self, other: 'V1GetLitDatasetFileUploadUrlsResponse') -> bool:
115
+ """Returns true if both objects are equal"""
116
+ if not isinstance(other, V1GetLitDatasetFileUploadUrlsResponse):
117
+ return False
118
+
119
+ return self.__dict__ == other.__dict__
120
+
121
+ def __ne__(self, other: 'V1GetLitDatasetFileUploadUrlsResponse') -> bool:
122
+ """Returns true if both objects are not equal"""
123
+ return not self == other
@@ -0,0 +1,149 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ external/v1/auth_service.proto
5
+
6
+ No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen) # noqa: E501
7
+
8
+ OpenAPI spec version: version not set
9
+
10
+ Generated by: https://github.com/swagger-api/swagger-codegen.git
11
+
12
+ NOTE
13
+ ----
14
+ standard swagger-codegen-cli for this python client has been modified
15
+ by custom templates. The purpose of these templates is to include
16
+ typing information in the API and Model code. Please refer to the
17
+ main grid repository for more info
18
+ """
19
+
20
+ import pprint
21
+ import re # noqa: F401
22
+
23
+ from typing import TYPE_CHECKING
24
+
25
+ import six
26
+
27
+ if TYPE_CHECKING:
28
+ from datetime import datetime
29
+ from lightning_sdk.lightning_cloud.openapi.models import *
30
+
31
+ class V1GetLitDatasetFilesUrlResponse(object):
32
+ """NOTE: This class is auto generated by the swagger code generator program.
33
+
34
+ Do not edit the class manually.
35
+ """
36
+ """
37
+ Attributes:
38
+ swagger_types (dict): The key is attribute name
39
+ and the value is attribute type.
40
+ attribute_map (dict): The key is attribute name
41
+ and the value is json key in definition.
42
+ """
43
+ swagger_types = {
44
+ 'files': 'list[V1LitDatasetFile]',
45
+ 'offset': 'str'
46
+ }
47
+
48
+ attribute_map = {
49
+ 'files': 'files',
50
+ 'offset': 'offset'
51
+ }
52
+
53
+ def __init__(self, files: 'list[V1LitDatasetFile]' =None, offset: 'str' =None): # noqa: E501
54
+ """V1GetLitDatasetFilesUrlResponse - a model defined in Swagger""" # noqa: E501
55
+ self._files = None
56
+ self._offset = None
57
+ self.discriminator = None
58
+ if files is not None:
59
+ self.files = files
60
+ if offset is not None:
61
+ self.offset = offset
62
+
63
+ @property
64
+ def files(self) -> 'list[V1LitDatasetFile]':
65
+ """Gets the files of this V1GetLitDatasetFilesUrlResponse. # noqa: E501
66
+
67
+
68
+ :return: The files of this V1GetLitDatasetFilesUrlResponse. # noqa: E501
69
+ :rtype: list[V1LitDatasetFile]
70
+ """
71
+ return self._files
72
+
73
+ @files.setter
74
+ def files(self, files: 'list[V1LitDatasetFile]'):
75
+ """Sets the files of this V1GetLitDatasetFilesUrlResponse.
76
+
77
+
78
+ :param files: The files of this V1GetLitDatasetFilesUrlResponse. # noqa: E501
79
+ :type: list[V1LitDatasetFile]
80
+ """
81
+
82
+ self._files = files
83
+
84
+ @property
85
+ def offset(self) -> 'str':
86
+ """Gets the offset of this V1GetLitDatasetFilesUrlResponse. # noqa: E501
87
+
88
+
89
+ :return: The offset of this V1GetLitDatasetFilesUrlResponse. # noqa: E501
90
+ :rtype: str
91
+ """
92
+ return self._offset
93
+
94
+ @offset.setter
95
+ def offset(self, offset: 'str'):
96
+ """Sets the offset of this V1GetLitDatasetFilesUrlResponse.
97
+
98
+
99
+ :param offset: The offset of this V1GetLitDatasetFilesUrlResponse. # noqa: E501
100
+ :type: str
101
+ """
102
+
103
+ self._offset = offset
104
+
105
+ def to_dict(self) -> dict:
106
+ """Returns the model properties as a dict"""
107
+ result = {}
108
+
109
+ for attr, _ in six.iteritems(self.swagger_types):
110
+ value = getattr(self, attr)
111
+ if isinstance(value, list):
112
+ result[attr] = list(map(
113
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
114
+ value
115
+ ))
116
+ elif hasattr(value, "to_dict"):
117
+ result[attr] = value.to_dict()
118
+ elif isinstance(value, dict):
119
+ result[attr] = dict(map(
120
+ lambda item: (item[0], item[1].to_dict())
121
+ if hasattr(item[1], "to_dict") else item,
122
+ value.items()
123
+ ))
124
+ else:
125
+ result[attr] = value
126
+ if issubclass(V1GetLitDatasetFilesUrlResponse, dict):
127
+ for key, value in self.items():
128
+ result[key] = value
129
+
130
+ return result
131
+
132
+ def to_str(self) -> str:
133
+ """Returns the string representation of the model"""
134
+ return pprint.pformat(self.to_dict())
135
+
136
+ def __repr__(self) -> str:
137
+ """For `print` and `pprint`"""
138
+ return self.to_str()
139
+
140
+ def __eq__(self, other: 'V1GetLitDatasetFilesUrlResponse') -> bool:
141
+ """Returns true if both objects are equal"""
142
+ if not isinstance(other, V1GetLitDatasetFilesUrlResponse):
143
+ return False
144
+
145
+ return self.__dict__ == other.__dict__
146
+
147
+ def __ne__(self, other: 'V1GetLitDatasetFilesUrlResponse') -> bool:
148
+ """Returns true if both objects are not equal"""
149
+ return not self == other
@@ -44,6 +44,7 @@ class V1GetUserResponse(object):
44
44
  'agree_to_terms_and_conditions': 'bool',
45
45
  'api_key': 'str',
46
46
  'country': 'str',
47
+ 'discounted_pro_plan': 'bool',
47
48
  'email': 'str',
48
49
  'experimentation_id': 'str',
49
50
  'features': 'V1UserFeatures',
@@ -80,6 +81,7 @@ class V1GetUserResponse(object):
80
81
  'agree_to_terms_and_conditions': 'agreeToTermsAndConditions',
81
82
  'api_key': 'apiKey',
82
83
  'country': 'country',
84
+ 'discounted_pro_plan': 'discountedProPlan',
83
85
  'email': 'email',
84
86
  'experimentation_id': 'experimentationId',
85
87
  'features': 'features',
@@ -112,11 +114,12 @@ class V1GetUserResponse(object):
112
114
  'website': 'website'
113
115
  }
114
116
 
115
- def __init__(self, agree_to_terms_and_conditions: 'bool' =None, api_key: 'str' =None, country: 'str' =None, email: 'str' =None, experimentation_id: 'str' =None, features: 'V1UserFeatures' =None, first_name: 'str' =None, general_audience_mode: 'bool' =None, id: 'str' =None, internal_docs_admin: 'bool' =None, is_internal: 'bool' =None, last_name: 'str' =None, non_developer_mode: 'bool' =None, opted_in_marketing_emails: 'bool' =None, organization: 'str' =None, organizations: 'list[V1Organization]' =None, phone_number: 'str' =None, picture_url: 'str' =None, preferred_color_scheme: 'str' =None, preferred_ide: 'str' =None, preferred_shell: 'str' =None, preferred_vscode_marketplace: 'str' =None, role: 'str' =None, saw_create_first_project_dialog: 'bool' =None, saw_forums_login_merge_dialog: 'bool' =None, saw_free_credits_notification: 'bool' =None, sb: 'bool' =None, status: 'Externalv1UserStatus' =None, storage_bytes: 'str' =None, user_metadata: 'str' =None, username: 'str' =None, waitlisted: 'bool' =None, website: 'str' =None): # noqa: E501
117
+ def __init__(self, agree_to_terms_and_conditions: 'bool' =None, api_key: 'str' =None, country: 'str' =None, discounted_pro_plan: 'bool' =None, email: 'str' =None, experimentation_id: 'str' =None, features: 'V1UserFeatures' =None, first_name: 'str' =None, general_audience_mode: 'bool' =None, id: 'str' =None, internal_docs_admin: 'bool' =None, is_internal: 'bool' =None, last_name: 'str' =None, non_developer_mode: 'bool' =None, opted_in_marketing_emails: 'bool' =None, organization: 'str' =None, organizations: 'list[V1Organization]' =None, phone_number: 'str' =None, picture_url: 'str' =None, preferred_color_scheme: 'str' =None, preferred_ide: 'str' =None, preferred_shell: 'str' =None, preferred_vscode_marketplace: 'str' =None, role: 'str' =None, saw_create_first_project_dialog: 'bool' =None, saw_forums_login_merge_dialog: 'bool' =None, saw_free_credits_notification: 'bool' =None, sb: 'bool' =None, status: 'Externalv1UserStatus' =None, storage_bytes: 'str' =None, user_metadata: 'str' =None, username: 'str' =None, waitlisted: 'bool' =None, website: 'str' =None): # noqa: E501
116
118
  """V1GetUserResponse - a model defined in Swagger""" # noqa: E501
117
119
  self._agree_to_terms_and_conditions = None
118
120
  self._api_key = None
119
121
  self._country = None
122
+ self._discounted_pro_plan = None
120
123
  self._email = None
121
124
  self._experimentation_id = None
122
125
  self._features = None
@@ -154,6 +157,8 @@ class V1GetUserResponse(object):
154
157
  self.api_key = api_key
155
158
  if country is not None:
156
159
  self.country = country
160
+ if discounted_pro_plan is not None:
161
+ self.discounted_pro_plan = discounted_pro_plan
157
162
  if email is not None:
158
163
  self.email = email
159
164
  if experimentation_id is not None:
@@ -278,6 +283,27 @@ class V1GetUserResponse(object):
278
283
 
279
284
  self._country = country
280
285
 
286
+ @property
287
+ def discounted_pro_plan(self) -> 'bool':
288
+ """Gets the discounted_pro_plan of this V1GetUserResponse. # noqa: E501
289
+
290
+
291
+ :return: The discounted_pro_plan of this V1GetUserResponse. # noqa: E501
292
+ :rtype: bool
293
+ """
294
+ return self._discounted_pro_plan
295
+
296
+ @discounted_pro_plan.setter
297
+ def discounted_pro_plan(self, discounted_pro_plan: 'bool'):
298
+ """Sets the discounted_pro_plan of this V1GetUserResponse.
299
+
300
+
301
+ :param discounted_pro_plan: The discounted_pro_plan of this V1GetUserResponse. # noqa: E501
302
+ :type: bool
303
+ """
304
+
305
+ self._discounted_pro_plan = discounted_pro_plan
306
+
281
307
  @property
282
308
  def email(self) -> 'str':
283
309
  """Gets the email of this V1GetUserResponse. # noqa: E501
@@ -41,61 +41,50 @@ class V1InstanceOverprovisioningSpec(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
- 'count': 'int',
45
44
  'instance_type': 'str',
46
45
  'is_spot': 'bool',
46
+ 'maximum_count': 'int',
47
+ 'minimum_count': 'int',
48
+ 'region': 'str',
47
49
  'resource_type': 'str',
48
50
  'volume_size': 'int'
49
51
  }
50
52
 
51
53
  attribute_map = {
52
- 'count': 'count',
53
54
  'instance_type': 'instanceType',
54
55
  'is_spot': 'isSpot',
56
+ 'maximum_count': 'maximumCount',
57
+ 'minimum_count': 'minimumCount',
58
+ 'region': 'region',
55
59
  'resource_type': 'resourceType',
56
60
  'volume_size': 'volumeSize'
57
61
  }
58
62
 
59
- def __init__(self, count: 'int' =None, instance_type: 'str' =None, is_spot: 'bool' =None, resource_type: 'str' =None, volume_size: 'int' =None): # noqa: E501
63
+ def __init__(self, instance_type: 'str' =None, is_spot: 'bool' =None, maximum_count: 'int' =None, minimum_count: 'int' =None, region: 'str' =None, resource_type: 'str' =None, volume_size: 'int' =None): # noqa: E501
60
64
  """V1InstanceOverprovisioningSpec - a model defined in Swagger""" # noqa: E501
61
- self._count = None
62
65
  self._instance_type = None
63
66
  self._is_spot = None
67
+ self._maximum_count = None
68
+ self._minimum_count = None
69
+ self._region = None
64
70
  self._resource_type = None
65
71
  self._volume_size = None
66
72
  self.discriminator = None
67
- if count is not None:
68
- self.count = count
69
73
  if instance_type is not None:
70
74
  self.instance_type = instance_type
71
75
  if is_spot is not None:
72
76
  self.is_spot = is_spot
77
+ if maximum_count is not None:
78
+ self.maximum_count = maximum_count
79
+ if minimum_count is not None:
80
+ self.minimum_count = minimum_count
81
+ if region is not None:
82
+ self.region = region
73
83
  if resource_type is not None:
74
84
  self.resource_type = resource_type
75
85
  if volume_size is not None:
76
86
  self.volume_size = volume_size
77
87
 
78
- @property
79
- def count(self) -> 'int':
80
- """Gets the count of this V1InstanceOverprovisioningSpec. # noqa: E501
81
-
82
-
83
- :return: The count of this V1InstanceOverprovisioningSpec. # noqa: E501
84
- :rtype: int
85
- """
86
- return self._count
87
-
88
- @count.setter
89
- def count(self, count: 'int'):
90
- """Sets the count of this V1InstanceOverprovisioningSpec.
91
-
92
-
93
- :param count: The count of this V1InstanceOverprovisioningSpec. # noqa: E501
94
- :type: int
95
- """
96
-
97
- self._count = count
98
-
99
88
  @property
100
89
  def instance_type(self) -> 'str':
101
90
  """Gets the instance_type of this V1InstanceOverprovisioningSpec. # noqa: E501
@@ -138,6 +127,69 @@ class V1InstanceOverprovisioningSpec(object):
138
127
 
139
128
  self._is_spot = is_spot
140
129
 
130
+ @property
131
+ def maximum_count(self) -> 'int':
132
+ """Gets the maximum_count of this V1InstanceOverprovisioningSpec. # noqa: E501
133
+
134
+
135
+ :return: The maximum_count of this V1InstanceOverprovisioningSpec. # noqa: E501
136
+ :rtype: int
137
+ """
138
+ return self._maximum_count
139
+
140
+ @maximum_count.setter
141
+ def maximum_count(self, maximum_count: 'int'):
142
+ """Sets the maximum_count of this V1InstanceOverprovisioningSpec.
143
+
144
+
145
+ :param maximum_count: The maximum_count of this V1InstanceOverprovisioningSpec. # noqa: E501
146
+ :type: int
147
+ """
148
+
149
+ self._maximum_count = maximum_count
150
+
151
+ @property
152
+ def minimum_count(self) -> 'int':
153
+ """Gets the minimum_count of this V1InstanceOverprovisioningSpec. # noqa: E501
154
+
155
+
156
+ :return: The minimum_count of this V1InstanceOverprovisioningSpec. # noqa: E501
157
+ :rtype: int
158
+ """
159
+ return self._minimum_count
160
+
161
+ @minimum_count.setter
162
+ def minimum_count(self, minimum_count: 'int'):
163
+ """Sets the minimum_count of this V1InstanceOverprovisioningSpec.
164
+
165
+
166
+ :param minimum_count: The minimum_count of this V1InstanceOverprovisioningSpec. # noqa: E501
167
+ :type: int
168
+ """
169
+
170
+ self._minimum_count = minimum_count
171
+
172
+ @property
173
+ def region(self) -> 'str':
174
+ """Gets the region of this V1InstanceOverprovisioningSpec. # noqa: E501
175
+
176
+
177
+ :return: The region of this V1InstanceOverprovisioningSpec. # noqa: E501
178
+ :rtype: str
179
+ """
180
+ return self._region
181
+
182
+ @region.setter
183
+ def region(self, region: 'str'):
184
+ """Sets the region of this V1InstanceOverprovisioningSpec.
185
+
186
+
187
+ :param region: The region of this V1InstanceOverprovisioningSpec. # noqa: E501
188
+ :type: str
189
+ """
190
+
191
+ self._region = region
192
+
141
193
  @property
142
194
  def resource_type(self) -> 'str':
143
195
  """Gets the resource_type of this V1InstanceOverprovisioningSpec. # noqa: E501
@@ -55,6 +55,7 @@ class V1JobSpec(object):
55
55
  'include_credentials': 'bool',
56
56
  'instance_name': 'str',
57
57
  'instance_type': 'str',
58
+ 'machine_image_version': 'str',
58
59
  'modified_volume': 'bool',
59
60
  'path_mappings': 'list[V1PathMapping]',
60
61
  'pipeline_id': 'str',
@@ -85,6 +86,7 @@ class V1JobSpec(object):
85
86
  'include_credentials': 'includeCredentials',
86
87
  'instance_name': 'instanceName',
87
88
  'instance_type': 'instanceType',
89
+ 'machine_image_version': 'machineImageVersion',
88
90
  'modified_volume': 'modifiedVolume',
89
91
  'path_mappings': 'pathMappings',
90
92
  'pipeline_id': 'pipelineId',
@@ -100,7 +102,7 @@ class V1JobSpec(object):
100
102
  'volumes': 'volumes'
101
103
  }
102
104
 
103
- 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, 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, volumes: 'list[V1Volume]' =None): # noqa: E501
105
+ 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, 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, volumes: 'list[V1Volume]' =None): # noqa: E501
104
106
  """V1JobSpec - a model defined in Swagger""" # noqa: E501
105
107
  self._artifacts_destination = None
106
108
  self._artifacts_source = None
@@ -116,6 +118,7 @@ class V1JobSpec(object):
116
118
  self._include_credentials = None
117
119
  self._instance_name = None
118
120
  self._instance_type = None
121
+ self._machine_image_version = None
119
122
  self._modified_volume = None
120
123
  self._path_mappings = None
121
124
  self._pipeline_id = None
@@ -158,6 +161,8 @@ class V1JobSpec(object):
158
161
  self.instance_name = instance_name
159
162
  if instance_type is not None:
160
163
  self.instance_type = instance_type
164
+ if machine_image_version is not None:
165
+ self.machine_image_version = machine_image_version
161
166
  if modified_volume is not None:
162
167
  self.modified_volume = modified_volume
163
168
  if path_mappings is not None:
@@ -479,6 +484,27 @@ class V1JobSpec(object):
479
484
 
480
485
  self._instance_type = instance_type
481
486
 
487
+ @property
488
+ def machine_image_version(self) -> 'str':
489
+ """Gets the machine_image_version of this V1JobSpec. # noqa: E501
490
+
491
+
492
+ :return: The machine_image_version of this V1JobSpec. # noqa: E501
493
+ :rtype: str
494
+ """
495
+ return self._machine_image_version
496
+
497
+ @machine_image_version.setter
498
+ def machine_image_version(self, machine_image_version: 'str'):
499
+ """Sets the machine_image_version of this V1JobSpec.
500
+
501
+
502
+ :param machine_image_version: The machine_image_version of this V1JobSpec. # noqa: E501
503
+ :type: str
504
+ """
505
+
506
+ self._machine_image_version = machine_image_version
507
+
482
508
  @property
483
509
  def modified_volume(self) -> 'bool':
484
510
  """Gets the modified_volume of this V1JobSpec. # noqa: E501
@@ -0,0 +1,123 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ external/v1/auth_service.proto
5
+
6
+ No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen) # noqa: E501
7
+
8
+ OpenAPI spec version: version not set
9
+
10
+ Generated by: https://github.com/swagger-api/swagger-codegen.git
11
+
12
+ NOTE
13
+ ----
14
+ standard swagger-codegen-cli for this python client has been modified
15
+ by custom templates. The purpose of these templates is to include
16
+ typing information in the API and Model code. Please refer to the
17
+ main grid repository for more info
18
+ """
19
+
20
+ import pprint
21
+ import re # noqa: F401
22
+
23
+ from typing import TYPE_CHECKING
24
+
25
+ import six
26
+
27
+ if TYPE_CHECKING:
28
+ from datetime import datetime
29
+ from lightning_sdk.lightning_cloud.openapi.models import *
30
+
31
+ class V1ListCloudSpaceEnvironmentTemplatesResponse(object):
32
+ """NOTE: This class is auto generated by the swagger code generator program.
33
+
34
+ Do not edit the class manually.
35
+ """
36
+ """
37
+ Attributes:
38
+ swagger_types (dict): The key is attribute name
39
+ and the value is attribute type.
40
+ attribute_map (dict): The key is attribute name
41
+ and the value is json key in definition.
42
+ """
43
+ swagger_types = {
44
+ 'templates': 'list[V1CloudSpaceEnvironmentTemplate]'
45
+ }
46
+
47
+ attribute_map = {
48
+ 'templates': 'templates'
49
+ }
50
+
51
+ def __init__(self, templates: 'list[V1CloudSpaceEnvironmentTemplate]' =None): # noqa: E501
52
+ """V1ListCloudSpaceEnvironmentTemplatesResponse - a model defined in Swagger""" # noqa: E501
53
+ self._templates = None
54
+ self.discriminator = None
55
+ if templates is not None:
56
+ self.templates = templates
57
+
58
+ @property
59
+ def templates(self) -> 'list[V1CloudSpaceEnvironmentTemplate]':
60
+ """Gets the templates of this V1ListCloudSpaceEnvironmentTemplatesResponse. # noqa: E501
61
+
62
+
63
+ :return: The templates of this V1ListCloudSpaceEnvironmentTemplatesResponse. # noqa: E501
64
+ :rtype: list[V1CloudSpaceEnvironmentTemplate]
65
+ """
66
+ return self._templates
67
+
68
+ @templates.setter
69
+ def templates(self, templates: 'list[V1CloudSpaceEnvironmentTemplate]'):
70
+ """Sets the templates of this V1ListCloudSpaceEnvironmentTemplatesResponse.
71
+
72
+
73
+ :param templates: The templates of this V1ListCloudSpaceEnvironmentTemplatesResponse. # noqa: E501
74
+ :type: list[V1CloudSpaceEnvironmentTemplate]
75
+ """
76
+
77
+ self._templates = templates
78
+
79
+ def to_dict(self) -> dict:
80
+ """Returns the model properties as a dict"""
81
+ result = {}
82
+
83
+ for attr, _ in six.iteritems(self.swagger_types):
84
+ value = getattr(self, attr)
85
+ if isinstance(value, list):
86
+ result[attr] = list(map(
87
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
88
+ value
89
+ ))
90
+ elif hasattr(value, "to_dict"):
91
+ result[attr] = value.to_dict()
92
+ elif isinstance(value, dict):
93
+ result[attr] = dict(map(
94
+ lambda item: (item[0], item[1].to_dict())
95
+ if hasattr(item[1], "to_dict") else item,
96
+ value.items()
97
+ ))
98
+ else:
99
+ result[attr] = value
100
+ if issubclass(V1ListCloudSpaceEnvironmentTemplatesResponse, dict):
101
+ for key, value in self.items():
102
+ result[key] = value
103
+
104
+ return result
105
+
106
+ def to_str(self) -> str:
107
+ """Returns the string representation of the model"""
108
+ return pprint.pformat(self.to_dict())
109
+
110
+ def __repr__(self) -> str:
111
+ """For `print` and `pprint`"""
112
+ return self.to_str()
113
+
114
+ def __eq__(self, other: 'V1ListCloudSpaceEnvironmentTemplatesResponse') -> bool:
115
+ """Returns true if both objects are equal"""
116
+ if not isinstance(other, V1ListCloudSpaceEnvironmentTemplatesResponse):
117
+ return False
118
+
119
+ return self.__dict__ == other.__dict__
120
+
121
+ def __ne__(self, other: 'V1ListCloudSpaceEnvironmentTemplatesResponse') -> bool:
122
+ """Returns true if both objects are not equal"""
123
+ return not self == other