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,383 @@
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 SchedulesIdBody(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
+ 'created_at': 'datetime',
45
+ 'cron_expression': 'str',
46
+ 'display_name': 'str',
47
+ 'name': 'str',
48
+ 'next': 'datetime',
49
+ 'resource_id': 'str',
50
+ 'resource_type': 'V1ScheduleResourceType',
51
+ 'state': 'str',
52
+ 'total': 'int',
53
+ 'updated_at': 'datetime',
54
+ 'user_id': 'str'
55
+ }
56
+
57
+ attribute_map = {
58
+ 'created_at': 'createdAt',
59
+ 'cron_expression': 'cronExpression',
60
+ 'display_name': 'displayName',
61
+ 'name': 'name',
62
+ 'next': 'next',
63
+ 'resource_id': 'resourceId',
64
+ 'resource_type': 'resourceType',
65
+ 'state': 'state',
66
+ 'total': 'total',
67
+ 'updated_at': 'updatedAt',
68
+ 'user_id': 'userId'
69
+ }
70
+
71
+ def __init__(self, created_at: 'datetime' =None, cron_expression: 'str' =None, display_name: 'str' =None, name: 'str' =None, next: 'datetime' =None, resource_id: 'str' =None, resource_type: 'V1ScheduleResourceType' =None, state: 'str' =None, total: 'int' =None, updated_at: 'datetime' =None, user_id: 'str' =None): # noqa: E501
72
+ """SchedulesIdBody - a model defined in Swagger""" # noqa: E501
73
+ self._created_at = None
74
+ self._cron_expression = None
75
+ self._display_name = None
76
+ self._name = None
77
+ self._next = None
78
+ self._resource_id = None
79
+ self._resource_type = None
80
+ self._state = None
81
+ self._total = None
82
+ self._updated_at = None
83
+ self._user_id = None
84
+ self.discriminator = None
85
+ if created_at is not None:
86
+ self.created_at = created_at
87
+ if cron_expression is not None:
88
+ self.cron_expression = cron_expression
89
+ if display_name is not None:
90
+ self.display_name = display_name
91
+ if name is not None:
92
+ self.name = name
93
+ if next is not None:
94
+ self.next = next
95
+ if resource_id is not None:
96
+ self.resource_id = resource_id
97
+ if resource_type is not None:
98
+ self.resource_type = resource_type
99
+ if state is not None:
100
+ self.state = state
101
+ if total is not None:
102
+ self.total = total
103
+ if updated_at is not None:
104
+ self.updated_at = updated_at
105
+ if user_id is not None:
106
+ self.user_id = user_id
107
+
108
+ @property
109
+ def created_at(self) -> 'datetime':
110
+ """Gets the created_at of this SchedulesIdBody. # noqa: E501
111
+
112
+
113
+ :return: The created_at of this SchedulesIdBody. # noqa: E501
114
+ :rtype: datetime
115
+ """
116
+ return self._created_at
117
+
118
+ @created_at.setter
119
+ def created_at(self, created_at: 'datetime'):
120
+ """Sets the created_at of this SchedulesIdBody.
121
+
122
+
123
+ :param created_at: The created_at of this SchedulesIdBody. # noqa: E501
124
+ :type: datetime
125
+ """
126
+
127
+ self._created_at = created_at
128
+
129
+ @property
130
+ def cron_expression(self) -> 'str':
131
+ """Gets the cron_expression of this SchedulesIdBody. # noqa: E501
132
+
133
+
134
+ :return: The cron_expression of this SchedulesIdBody. # noqa: E501
135
+ :rtype: str
136
+ """
137
+ return self._cron_expression
138
+
139
+ @cron_expression.setter
140
+ def cron_expression(self, cron_expression: 'str'):
141
+ """Sets the cron_expression of this SchedulesIdBody.
142
+
143
+
144
+ :param cron_expression: The cron_expression of this SchedulesIdBody. # noqa: E501
145
+ :type: str
146
+ """
147
+
148
+ self._cron_expression = cron_expression
149
+
150
+ @property
151
+ def display_name(self) -> 'str':
152
+ """Gets the display_name of this SchedulesIdBody. # noqa: E501
153
+
154
+
155
+ :return: The display_name of this SchedulesIdBody. # noqa: E501
156
+ :rtype: str
157
+ """
158
+ return self._display_name
159
+
160
+ @display_name.setter
161
+ def display_name(self, display_name: 'str'):
162
+ """Sets the display_name of this SchedulesIdBody.
163
+
164
+
165
+ :param display_name: The display_name of this SchedulesIdBody. # noqa: E501
166
+ :type: str
167
+ """
168
+
169
+ self._display_name = display_name
170
+
171
+ @property
172
+ def name(self) -> 'str':
173
+ """Gets the name of this SchedulesIdBody. # noqa: E501
174
+
175
+
176
+ :return: The name of this SchedulesIdBody. # noqa: E501
177
+ :rtype: str
178
+ """
179
+ return self._name
180
+
181
+ @name.setter
182
+ def name(self, name: 'str'):
183
+ """Sets the name of this SchedulesIdBody.
184
+
185
+
186
+ :param name: The name of this SchedulesIdBody. # noqa: E501
187
+ :type: str
188
+ """
189
+
190
+ self._name = name
191
+
192
+ @property
193
+ def next(self) -> 'datetime':
194
+ """Gets the next of this SchedulesIdBody. # noqa: E501
195
+
196
+
197
+ :return: The next of this SchedulesIdBody. # noqa: E501
198
+ :rtype: datetime
199
+ """
200
+ return self._next
201
+
202
+ @next.setter
203
+ def next(self, next: 'datetime'):
204
+ """Sets the next of this SchedulesIdBody.
205
+
206
+
207
+ :param next: The next of this SchedulesIdBody. # noqa: E501
208
+ :type: datetime
209
+ """
210
+
211
+ self._next = next
212
+
213
+ @property
214
+ def resource_id(self) -> 'str':
215
+ """Gets the resource_id of this SchedulesIdBody. # noqa: E501
216
+
217
+
218
+ :return: The resource_id of this SchedulesIdBody. # noqa: E501
219
+ :rtype: str
220
+ """
221
+ return self._resource_id
222
+
223
+ @resource_id.setter
224
+ def resource_id(self, resource_id: 'str'):
225
+ """Sets the resource_id of this SchedulesIdBody.
226
+
227
+
228
+ :param resource_id: The resource_id of this SchedulesIdBody. # noqa: E501
229
+ :type: str
230
+ """
231
+
232
+ self._resource_id = resource_id
233
+
234
+ @property
235
+ def resource_type(self) -> 'V1ScheduleResourceType':
236
+ """Gets the resource_type of this SchedulesIdBody. # noqa: E501
237
+
238
+
239
+ :return: The resource_type of this SchedulesIdBody. # noqa: E501
240
+ :rtype: V1ScheduleResourceType
241
+ """
242
+ return self._resource_type
243
+
244
+ @resource_type.setter
245
+ def resource_type(self, resource_type: 'V1ScheduleResourceType'):
246
+ """Sets the resource_type of this SchedulesIdBody.
247
+
248
+
249
+ :param resource_type: The resource_type of this SchedulesIdBody. # noqa: E501
250
+ :type: V1ScheduleResourceType
251
+ """
252
+
253
+ self._resource_type = resource_type
254
+
255
+ @property
256
+ def state(self) -> 'str':
257
+ """Gets the state of this SchedulesIdBody. # noqa: E501
258
+
259
+
260
+ :return: The state of this SchedulesIdBody. # noqa: E501
261
+ :rtype: str
262
+ """
263
+ return self._state
264
+
265
+ @state.setter
266
+ def state(self, state: 'str'):
267
+ """Sets the state of this SchedulesIdBody.
268
+
269
+
270
+ :param state: The state of this SchedulesIdBody. # noqa: E501
271
+ :type: str
272
+ """
273
+
274
+ self._state = state
275
+
276
+ @property
277
+ def total(self) -> 'int':
278
+ """Gets the total of this SchedulesIdBody. # noqa: E501
279
+
280
+
281
+ :return: The total of this SchedulesIdBody. # noqa: E501
282
+ :rtype: int
283
+ """
284
+ return self._total
285
+
286
+ @total.setter
287
+ def total(self, total: 'int'):
288
+ """Sets the total of this SchedulesIdBody.
289
+
290
+
291
+ :param total: The total of this SchedulesIdBody. # noqa: E501
292
+ :type: int
293
+ """
294
+
295
+ self._total = total
296
+
297
+ @property
298
+ def updated_at(self) -> 'datetime':
299
+ """Gets the updated_at of this SchedulesIdBody. # noqa: E501
300
+
301
+
302
+ :return: The updated_at of this SchedulesIdBody. # noqa: E501
303
+ :rtype: datetime
304
+ """
305
+ return self._updated_at
306
+
307
+ @updated_at.setter
308
+ def updated_at(self, updated_at: 'datetime'):
309
+ """Sets the updated_at of this SchedulesIdBody.
310
+
311
+
312
+ :param updated_at: The updated_at of this SchedulesIdBody. # noqa: E501
313
+ :type: datetime
314
+ """
315
+
316
+ self._updated_at = updated_at
317
+
318
+ @property
319
+ def user_id(self) -> 'str':
320
+ """Gets the user_id of this SchedulesIdBody. # noqa: E501
321
+
322
+
323
+ :return: The user_id of this SchedulesIdBody. # noqa: E501
324
+ :rtype: str
325
+ """
326
+ return self._user_id
327
+
328
+ @user_id.setter
329
+ def user_id(self, user_id: 'str'):
330
+ """Sets the user_id of this SchedulesIdBody.
331
+
332
+
333
+ :param user_id: The user_id of this SchedulesIdBody. # noqa: E501
334
+ :type: str
335
+ """
336
+
337
+ self._user_id = user_id
338
+
339
+ def to_dict(self) -> dict:
340
+ """Returns the model properties as a dict"""
341
+ result = {}
342
+
343
+ for attr, _ in six.iteritems(self.swagger_types):
344
+ value = getattr(self, attr)
345
+ if isinstance(value, list):
346
+ result[attr] = list(map(
347
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
348
+ value
349
+ ))
350
+ elif hasattr(value, "to_dict"):
351
+ result[attr] = value.to_dict()
352
+ elif isinstance(value, dict):
353
+ result[attr] = dict(map(
354
+ lambda item: (item[0], item[1].to_dict())
355
+ if hasattr(item[1], "to_dict") else item,
356
+ value.items()
357
+ ))
358
+ else:
359
+ result[attr] = value
360
+ if issubclass(SchedulesIdBody, dict):
361
+ for key, value in self.items():
362
+ result[key] = value
363
+
364
+ return result
365
+
366
+ def to_str(self) -> str:
367
+ """Returns the string representation of the model"""
368
+ return pprint.pformat(self.to_dict())
369
+
370
+ def __repr__(self) -> str:
371
+ """For `print` and `pprint`"""
372
+ return self.to_str()
373
+
374
+ def __eq__(self, other: 'SchedulesIdBody') -> bool:
375
+ """Returns true if both objects are equal"""
376
+ if not isinstance(other, SchedulesIdBody):
377
+ return False
378
+
379
+ return self.__dict__ == other.__dict__
380
+
381
+ def __ne__(self, other: 'SchedulesIdBody') -> bool:
382
+ """Returns true if both objects are not equal"""
383
+ return not self == other
@@ -49,7 +49,7 @@ class SlurmJobsBody(object):
49
49
  'image': 'str',
50
50
  'image_secret_ref': 'str',
51
51
  'name': 'str',
52
- 'num_gpus': 'str',
52
+ 'partition': 'str',
53
53
  'service_id': 'str',
54
54
  'sync_env': 'bool',
55
55
  'work_dir': 'str'
@@ -64,13 +64,13 @@ class SlurmJobsBody(object):
64
64
  'image': 'image',
65
65
  'image_secret_ref': 'imageSecretRef',
66
66
  'name': 'name',
67
- 'num_gpus': 'numGpus',
67
+ 'partition': 'partition',
68
68
  'service_id': 'serviceId',
69
69
  'sync_env': 'syncEnv',
70
70
  'work_dir': 'workDir'
71
71
  }
72
72
 
73
- def __init__(self, cache_id: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, entrypoint: 'str' =None, image: 'str' =None, image_secret_ref: 'str' =None, name: 'str' =None, num_gpus: 'str' =None, service_id: 'str' =None, sync_env: 'bool' =None, work_dir: 'str' =None): # noqa: E501
73
+ def __init__(self, cache_id: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, entrypoint: 'str' =None, image: 'str' =None, image_secret_ref: 'str' =None, name: 'str' =None, partition: 'str' =None, service_id: 'str' =None, sync_env: 'bool' =None, work_dir: 'str' =None): # noqa: E501
74
74
  """SlurmJobsBody - a model defined in Swagger""" # noqa: E501
75
75
  self._cache_id = None
76
76
  self._cloudspace_id = None
@@ -80,7 +80,7 @@ class SlurmJobsBody(object):
80
80
  self._image = None
81
81
  self._image_secret_ref = None
82
82
  self._name = None
83
- self._num_gpus = None
83
+ self._partition = None
84
84
  self._service_id = None
85
85
  self._sync_env = None
86
86
  self._work_dir = None
@@ -101,8 +101,8 @@ class SlurmJobsBody(object):
101
101
  self.image_secret_ref = image_secret_ref
102
102
  if name is not None:
103
103
  self.name = name
104
- if num_gpus is not None:
105
- self.num_gpus = num_gpus
104
+ if partition is not None:
105
+ self.partition = partition
106
106
  if service_id is not None:
107
107
  self.service_id = service_id
108
108
  if sync_env is not None:
@@ -281,25 +281,25 @@ class SlurmJobsBody(object):
281
281
  self._name = name
282
282
 
283
283
  @property
284
- def num_gpus(self) -> 'str':
285
- """Gets the num_gpus of this SlurmJobsBody. # noqa: E501
284
+ def partition(self) -> 'str':
285
+ """Gets the partition of this SlurmJobsBody. # noqa: E501
286
286
 
287
287
 
288
- :return: The num_gpus of this SlurmJobsBody. # noqa: E501
288
+ :return: The partition of this SlurmJobsBody. # noqa: E501
289
289
  :rtype: str
290
290
  """
291
- return self._num_gpus
291
+ return self._partition
292
292
 
293
- @num_gpus.setter
294
- def num_gpus(self, num_gpus: 'str'):
295
- """Sets the num_gpus of this SlurmJobsBody.
293
+ @partition.setter
294
+ def partition(self, partition: 'str'):
295
+ """Sets the partition of this SlurmJobsBody.
296
296
 
297
297
 
298
- :param num_gpus: The num_gpus of this SlurmJobsBody. # noqa: E501
298
+ :param partition: The partition of this SlurmJobsBody. # noqa: E501
299
299
  :type: str
300
300
  """
301
301
 
302
- self._num_gpus = num_gpus
302
+ self._partition = partition
303
303
 
304
304
  @property
305
305
  def service_id(self) -> 'str':
@@ -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 UploadIdCompleteBody1(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
+ 'filepath': 'str',
45
+ 'parts': 'list[V1CompletedPart]'
46
+ }
47
+
48
+ attribute_map = {
49
+ 'filepath': 'filepath',
50
+ 'parts': 'parts'
51
+ }
52
+
53
+ def __init__(self, filepath: 'str' =None, parts: 'list[V1CompletedPart]' =None): # noqa: E501
54
+ """UploadIdCompleteBody1 - a model defined in Swagger""" # noqa: E501
55
+ self._filepath = None
56
+ self._parts = None
57
+ self.discriminator = None
58
+ if filepath is not None:
59
+ self.filepath = filepath
60
+ if parts is not None:
61
+ self.parts = parts
62
+
63
+ @property
64
+ def filepath(self) -> 'str':
65
+ """Gets the filepath of this UploadIdCompleteBody1. # noqa: E501
66
+
67
+
68
+ :return: The filepath of this UploadIdCompleteBody1. # noqa: E501
69
+ :rtype: str
70
+ """
71
+ return self._filepath
72
+
73
+ @filepath.setter
74
+ def filepath(self, filepath: 'str'):
75
+ """Sets the filepath of this UploadIdCompleteBody1.
76
+
77
+
78
+ :param filepath: The filepath of this UploadIdCompleteBody1. # noqa: E501
79
+ :type: str
80
+ """
81
+
82
+ self._filepath = filepath
83
+
84
+ @property
85
+ def parts(self) -> 'list[V1CompletedPart]':
86
+ """Gets the parts of this UploadIdCompleteBody1. # noqa: E501
87
+
88
+
89
+ :return: The parts of this UploadIdCompleteBody1. # noqa: E501
90
+ :rtype: list[V1CompletedPart]
91
+ """
92
+ return self._parts
93
+
94
+ @parts.setter
95
+ def parts(self, parts: 'list[V1CompletedPart]'):
96
+ """Sets the parts of this UploadIdCompleteBody1.
97
+
98
+
99
+ :param parts: The parts of this UploadIdCompleteBody1. # noqa: E501
100
+ :type: list[V1CompletedPart]
101
+ """
102
+
103
+ self._parts = parts
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(UploadIdCompleteBody1, 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: 'UploadIdCompleteBody1') -> bool:
141
+ """Returns true if both objects are equal"""
142
+ if not isinstance(other, UploadIdCompleteBody1):
143
+ return False
144
+
145
+ return self.__dict__ == other.__dict__
146
+
147
+ def __ne__(self, other: 'UploadIdCompleteBody1') -> bool:
148
+ """Returns true if both objects are not equal"""
149
+ return not self == other