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,435 @@
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 V1LitDatasetVersionArchive(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
+ 'cluster_id': 'str',
45
+ 'created_at': 'datetime',
46
+ 'dataset_id': 'str',
47
+ 'downloads': 'str',
48
+ 'index': 'int',
49
+ 'metadata': 'dict(str, str)',
50
+ 'number_of_files': 'str',
51
+ 'project_id': 'str',
52
+ 'size_bytes': 'str',
53
+ 'updated_at': 'datetime',
54
+ 'upload_complete': 'bool',
55
+ 'user_id': 'str',
56
+ 'version': 'str'
57
+ }
58
+
59
+ attribute_map = {
60
+ 'cluster_id': 'clusterId',
61
+ 'created_at': 'createdAt',
62
+ 'dataset_id': 'datasetId',
63
+ 'downloads': 'downloads',
64
+ 'index': 'index',
65
+ 'metadata': 'metadata',
66
+ 'number_of_files': 'numberOfFiles',
67
+ 'project_id': 'projectId',
68
+ 'size_bytes': 'sizeBytes',
69
+ 'updated_at': 'updatedAt',
70
+ 'upload_complete': 'uploadComplete',
71
+ 'user_id': 'userId',
72
+ 'version': 'version'
73
+ }
74
+
75
+ def __init__(self, cluster_id: 'str' =None, created_at: 'datetime' =None, dataset_id: 'str' =None, downloads: 'str' =None, index: 'int' =None, metadata: 'dict(str, str)' =None, number_of_files: 'str' =None, project_id: 'str' =None, size_bytes: 'str' =None, updated_at: 'datetime' =None, upload_complete: 'bool' =None, user_id: 'str' =None, version: 'str' =None): # noqa: E501
76
+ """V1LitDatasetVersionArchive - a model defined in Swagger""" # noqa: E501
77
+ self._cluster_id = None
78
+ self._created_at = None
79
+ self._dataset_id = None
80
+ self._downloads = None
81
+ self._index = None
82
+ self._metadata = None
83
+ self._number_of_files = None
84
+ self._project_id = None
85
+ self._size_bytes = None
86
+ self._updated_at = None
87
+ self._upload_complete = None
88
+ self._user_id = None
89
+ self._version = None
90
+ self.discriminator = None
91
+ if cluster_id is not None:
92
+ self.cluster_id = cluster_id
93
+ if created_at is not None:
94
+ self.created_at = created_at
95
+ if dataset_id is not None:
96
+ self.dataset_id = dataset_id
97
+ if downloads is not None:
98
+ self.downloads = downloads
99
+ if index is not None:
100
+ self.index = index
101
+ if metadata is not None:
102
+ self.metadata = metadata
103
+ if number_of_files is not None:
104
+ self.number_of_files = number_of_files
105
+ if project_id is not None:
106
+ self.project_id = project_id
107
+ if size_bytes is not None:
108
+ self.size_bytes = size_bytes
109
+ if updated_at is not None:
110
+ self.updated_at = updated_at
111
+ if upload_complete is not None:
112
+ self.upload_complete = upload_complete
113
+ if user_id is not None:
114
+ self.user_id = user_id
115
+ if version is not None:
116
+ self.version = version
117
+
118
+ @property
119
+ def cluster_id(self) -> 'str':
120
+ """Gets the cluster_id of this V1LitDatasetVersionArchive. # noqa: E501
121
+
122
+
123
+ :return: The cluster_id of this V1LitDatasetVersionArchive. # noqa: E501
124
+ :rtype: str
125
+ """
126
+ return self._cluster_id
127
+
128
+ @cluster_id.setter
129
+ def cluster_id(self, cluster_id: 'str'):
130
+ """Sets the cluster_id of this V1LitDatasetVersionArchive.
131
+
132
+
133
+ :param cluster_id: The cluster_id of this V1LitDatasetVersionArchive. # noqa: E501
134
+ :type: str
135
+ """
136
+
137
+ self._cluster_id = cluster_id
138
+
139
+ @property
140
+ def created_at(self) -> 'datetime':
141
+ """Gets the created_at of this V1LitDatasetVersionArchive. # noqa: E501
142
+
143
+
144
+ :return: The created_at of this V1LitDatasetVersionArchive. # noqa: E501
145
+ :rtype: datetime
146
+ """
147
+ return self._created_at
148
+
149
+ @created_at.setter
150
+ def created_at(self, created_at: 'datetime'):
151
+ """Sets the created_at of this V1LitDatasetVersionArchive.
152
+
153
+
154
+ :param created_at: The created_at of this V1LitDatasetVersionArchive. # noqa: E501
155
+ :type: datetime
156
+ """
157
+
158
+ self._created_at = created_at
159
+
160
+ @property
161
+ def dataset_id(self) -> 'str':
162
+ """Gets the dataset_id of this V1LitDatasetVersionArchive. # noqa: E501
163
+
164
+
165
+ :return: The dataset_id of this V1LitDatasetVersionArchive. # noqa: E501
166
+ :rtype: str
167
+ """
168
+ return self._dataset_id
169
+
170
+ @dataset_id.setter
171
+ def dataset_id(self, dataset_id: 'str'):
172
+ """Sets the dataset_id of this V1LitDatasetVersionArchive.
173
+
174
+
175
+ :param dataset_id: The dataset_id of this V1LitDatasetVersionArchive. # noqa: E501
176
+ :type: str
177
+ """
178
+
179
+ self._dataset_id = dataset_id
180
+
181
+ @property
182
+ def downloads(self) -> 'str':
183
+ """Gets the downloads of this V1LitDatasetVersionArchive. # noqa: E501
184
+
185
+
186
+ :return: The downloads of this V1LitDatasetVersionArchive. # noqa: E501
187
+ :rtype: str
188
+ """
189
+ return self._downloads
190
+
191
+ @downloads.setter
192
+ def downloads(self, downloads: 'str'):
193
+ """Sets the downloads of this V1LitDatasetVersionArchive.
194
+
195
+
196
+ :param downloads: The downloads of this V1LitDatasetVersionArchive. # noqa: E501
197
+ :type: str
198
+ """
199
+
200
+ self._downloads = downloads
201
+
202
+ @property
203
+ def index(self) -> 'int':
204
+ """Gets the index of this V1LitDatasetVersionArchive. # noqa: E501
205
+
206
+
207
+ :return: The index of this V1LitDatasetVersionArchive. # noqa: E501
208
+ :rtype: int
209
+ """
210
+ return self._index
211
+
212
+ @index.setter
213
+ def index(self, index: 'int'):
214
+ """Sets the index of this V1LitDatasetVersionArchive.
215
+
216
+
217
+ :param index: The index of this V1LitDatasetVersionArchive. # noqa: E501
218
+ :type: int
219
+ """
220
+
221
+ self._index = index
222
+
223
+ @property
224
+ def metadata(self) -> 'dict(str, str)':
225
+ """Gets the metadata of this V1LitDatasetVersionArchive. # noqa: E501
226
+
227
+
228
+ :return: The metadata of this V1LitDatasetVersionArchive. # noqa: E501
229
+ :rtype: dict(str, str)
230
+ """
231
+ return self._metadata
232
+
233
+ @metadata.setter
234
+ def metadata(self, metadata: 'dict(str, str)'):
235
+ """Sets the metadata of this V1LitDatasetVersionArchive.
236
+
237
+
238
+ :param metadata: The metadata of this V1LitDatasetVersionArchive. # noqa: E501
239
+ :type: dict(str, str)
240
+ """
241
+
242
+ self._metadata = metadata
243
+
244
+ @property
245
+ def number_of_files(self) -> 'str':
246
+ """Gets the number_of_files of this V1LitDatasetVersionArchive. # noqa: E501
247
+
248
+
249
+ :return: The number_of_files of this V1LitDatasetVersionArchive. # noqa: E501
250
+ :rtype: str
251
+ """
252
+ return self._number_of_files
253
+
254
+ @number_of_files.setter
255
+ def number_of_files(self, number_of_files: 'str'):
256
+ """Sets the number_of_files of this V1LitDatasetVersionArchive.
257
+
258
+
259
+ :param number_of_files: The number_of_files of this V1LitDatasetVersionArchive. # noqa: E501
260
+ :type: str
261
+ """
262
+
263
+ self._number_of_files = number_of_files
264
+
265
+ @property
266
+ def project_id(self) -> 'str':
267
+ """Gets the project_id of this V1LitDatasetVersionArchive. # noqa: E501
268
+
269
+
270
+ :return: The project_id of this V1LitDatasetVersionArchive. # noqa: E501
271
+ :rtype: str
272
+ """
273
+ return self._project_id
274
+
275
+ @project_id.setter
276
+ def project_id(self, project_id: 'str'):
277
+ """Sets the project_id of this V1LitDatasetVersionArchive.
278
+
279
+
280
+ :param project_id: The project_id of this V1LitDatasetVersionArchive. # noqa: E501
281
+ :type: str
282
+ """
283
+
284
+ self._project_id = project_id
285
+
286
+ @property
287
+ def size_bytes(self) -> 'str':
288
+ """Gets the size_bytes of this V1LitDatasetVersionArchive. # noqa: E501
289
+
290
+
291
+ :return: The size_bytes of this V1LitDatasetVersionArchive. # noqa: E501
292
+ :rtype: str
293
+ """
294
+ return self._size_bytes
295
+
296
+ @size_bytes.setter
297
+ def size_bytes(self, size_bytes: 'str'):
298
+ """Sets the size_bytes of this V1LitDatasetVersionArchive.
299
+
300
+
301
+ :param size_bytes: The size_bytes of this V1LitDatasetVersionArchive. # noqa: E501
302
+ :type: str
303
+ """
304
+
305
+ self._size_bytes = size_bytes
306
+
307
+ @property
308
+ def updated_at(self) -> 'datetime':
309
+ """Gets the updated_at of this V1LitDatasetVersionArchive. # noqa: E501
310
+
311
+
312
+ :return: The updated_at of this V1LitDatasetVersionArchive. # noqa: E501
313
+ :rtype: datetime
314
+ """
315
+ return self._updated_at
316
+
317
+ @updated_at.setter
318
+ def updated_at(self, updated_at: 'datetime'):
319
+ """Sets the updated_at of this V1LitDatasetVersionArchive.
320
+
321
+
322
+ :param updated_at: The updated_at of this V1LitDatasetVersionArchive. # noqa: E501
323
+ :type: datetime
324
+ """
325
+
326
+ self._updated_at = updated_at
327
+
328
+ @property
329
+ def upload_complete(self) -> 'bool':
330
+ """Gets the upload_complete of this V1LitDatasetVersionArchive. # noqa: E501
331
+
332
+
333
+ :return: The upload_complete of this V1LitDatasetVersionArchive. # noqa: E501
334
+ :rtype: bool
335
+ """
336
+ return self._upload_complete
337
+
338
+ @upload_complete.setter
339
+ def upload_complete(self, upload_complete: 'bool'):
340
+ """Sets the upload_complete of this V1LitDatasetVersionArchive.
341
+
342
+
343
+ :param upload_complete: The upload_complete of this V1LitDatasetVersionArchive. # noqa: E501
344
+ :type: bool
345
+ """
346
+
347
+ self._upload_complete = upload_complete
348
+
349
+ @property
350
+ def user_id(self) -> 'str':
351
+ """Gets the user_id of this V1LitDatasetVersionArchive. # noqa: E501
352
+
353
+
354
+ :return: The user_id of this V1LitDatasetVersionArchive. # noqa: E501
355
+ :rtype: str
356
+ """
357
+ return self._user_id
358
+
359
+ @user_id.setter
360
+ def user_id(self, user_id: 'str'):
361
+ """Sets the user_id of this V1LitDatasetVersionArchive.
362
+
363
+
364
+ :param user_id: The user_id of this V1LitDatasetVersionArchive. # noqa: E501
365
+ :type: str
366
+ """
367
+
368
+ self._user_id = user_id
369
+
370
+ @property
371
+ def version(self) -> 'str':
372
+ """Gets the version of this V1LitDatasetVersionArchive. # noqa: E501
373
+
374
+
375
+ :return: The version of this V1LitDatasetVersionArchive. # noqa: E501
376
+ :rtype: str
377
+ """
378
+ return self._version
379
+
380
+ @version.setter
381
+ def version(self, version: 'str'):
382
+ """Sets the version of this V1LitDatasetVersionArchive.
383
+
384
+
385
+ :param version: The version of this V1LitDatasetVersionArchive. # noqa: E501
386
+ :type: str
387
+ """
388
+
389
+ self._version = version
390
+
391
+ def to_dict(self) -> dict:
392
+ """Returns the model properties as a dict"""
393
+ result = {}
394
+
395
+ for attr, _ in six.iteritems(self.swagger_types):
396
+ value = getattr(self, attr)
397
+ if isinstance(value, list):
398
+ result[attr] = list(map(
399
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
400
+ value
401
+ ))
402
+ elif hasattr(value, "to_dict"):
403
+ result[attr] = value.to_dict()
404
+ elif isinstance(value, dict):
405
+ result[attr] = dict(map(
406
+ lambda item: (item[0], item[1].to_dict())
407
+ if hasattr(item[1], "to_dict") else item,
408
+ value.items()
409
+ ))
410
+ else:
411
+ result[attr] = value
412
+ if issubclass(V1LitDatasetVersionArchive, dict):
413
+ for key, value in self.items():
414
+ result[key] = value
415
+
416
+ return result
417
+
418
+ def to_str(self) -> str:
419
+ """Returns the string representation of the model"""
420
+ return pprint.pformat(self.to_dict())
421
+
422
+ def __repr__(self) -> str:
423
+ """For `print` and `pprint`"""
424
+ return self.to_str()
425
+
426
+ def __eq__(self, other: 'V1LitDatasetVersionArchive') -> bool:
427
+ """Returns true if both objects are equal"""
428
+ if not isinstance(other, V1LitDatasetVersionArchive):
429
+ return False
430
+
431
+ return self.__dict__ == other.__dict__
432
+
433
+ def __ne__(self, other: 'V1LitDatasetVersionArchive') -> bool:
434
+ """Returns true if both objects are not equal"""
435
+ return not self == other
@@ -41,6 +41,7 @@ class V1LitRegistryProject(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
+ 'cluster_id': 'str',
44
45
  'id': 'int',
45
46
  'lit_reg_project_name': 'str',
46
47
  'project_id': 'str',
@@ -49,6 +50,7 @@ class V1LitRegistryProject(object):
49
50
  }
50
51
 
51
52
  attribute_map = {
53
+ 'cluster_id': 'clusterId',
52
54
  'id': 'id',
53
55
  'lit_reg_project_name': 'litRegProjectName',
54
56
  'project_id': 'projectId',
@@ -56,14 +58,17 @@ class V1LitRegistryProject(object):
56
58
  'repositories': 'repositories'
57
59
  }
58
60
 
59
- def __init__(self, id: 'int' =None, lit_reg_project_name: 'str' =None, project_id: 'str' =None, repo_count: 'str' =None, repositories: 'list[V1LitRepository]' =None): # noqa: E501
61
+ def __init__(self, cluster_id: 'str' =None, id: 'int' =None, lit_reg_project_name: 'str' =None, project_id: 'str' =None, repo_count: 'str' =None, repositories: 'list[V1LitRepository]' =None): # noqa: E501
60
62
  """V1LitRegistryProject - a model defined in Swagger""" # noqa: E501
63
+ self._cluster_id = None
61
64
  self._id = None
62
65
  self._lit_reg_project_name = None
63
66
  self._project_id = None
64
67
  self._repo_count = None
65
68
  self._repositories = None
66
69
  self.discriminator = None
70
+ if cluster_id is not None:
71
+ self.cluster_id = cluster_id
67
72
  if id is not None:
68
73
  self.id = id
69
74
  if lit_reg_project_name is not None:
@@ -75,6 +80,27 @@ class V1LitRegistryProject(object):
75
80
  if repositories is not None:
76
81
  self.repositories = repositories
77
82
 
83
+ @property
84
+ def cluster_id(self) -> 'str':
85
+ """Gets the cluster_id of this V1LitRegistryProject. # noqa: E501
86
+
87
+
88
+ :return: The cluster_id of this V1LitRegistryProject. # noqa: E501
89
+ :rtype: str
90
+ """
91
+ return self._cluster_id
92
+
93
+ @cluster_id.setter
94
+ def cluster_id(self, cluster_id: 'str'):
95
+ """Sets the cluster_id of this V1LitRegistryProject.
96
+
97
+
98
+ :param cluster_id: The cluster_id of this V1LitRegistryProject. # noqa: E501
99
+ :type: str
100
+ """
101
+
102
+ self._cluster_id = cluster_id
103
+
78
104
  @property
79
105
  def id(self) -> 'int':
80
106
  """Gets the id of this V1LitRegistryProject. # noqa: E501
@@ -43,6 +43,7 @@ class V1LitRepository(object):
43
43
  swagger_types = {
44
44
  'artifact_count': 'str',
45
45
  'artifacts': 'list[V1LitRegistryArtifact]',
46
+ 'cluster_id': 'str',
46
47
  'creation_time': 'datetime',
47
48
  'description': 'str',
48
49
  'id': 'str',
@@ -55,6 +56,7 @@ class V1LitRepository(object):
55
56
  attribute_map = {
56
57
  'artifact_count': 'artifactCount',
57
58
  'artifacts': 'artifacts',
59
+ 'cluster_id': 'clusterId',
58
60
  'creation_time': 'creationTime',
59
61
  'description': 'description',
60
62
  'id': 'id',
@@ -64,10 +66,11 @@ class V1LitRepository(object):
64
66
  'update_time': 'updateTime'
65
67
  }
66
68
 
67
- def __init__(self, artifact_count: 'str' =None, artifacts: 'list[V1LitRegistryArtifact]' =None, creation_time: 'datetime' =None, description: 'str' =None, id: 'str' =None, latest_artifact: 'V1LitRegistryArtifact' =None, name: 'str' =None, pull_count: 'str' =None, update_time: 'datetime' =None): # noqa: E501
69
+ def __init__(self, artifact_count: 'str' =None, artifacts: 'list[V1LitRegistryArtifact]' =None, cluster_id: 'str' =None, creation_time: 'datetime' =None, description: 'str' =None, id: 'str' =None, latest_artifact: 'V1LitRegistryArtifact' =None, name: 'str' =None, pull_count: 'str' =None, update_time: 'datetime' =None): # noqa: E501
68
70
  """V1LitRepository - a model defined in Swagger""" # noqa: E501
69
71
  self._artifact_count = None
70
72
  self._artifacts = None
73
+ self._cluster_id = None
71
74
  self._creation_time = None
72
75
  self._description = None
73
76
  self._id = None
@@ -80,6 +83,8 @@ class V1LitRepository(object):
80
83
  self.artifact_count = artifact_count
81
84
  if artifacts is not None:
82
85
  self.artifacts = artifacts
86
+ if cluster_id is not None:
87
+ self.cluster_id = cluster_id
83
88
  if creation_time is not None:
84
89
  self.creation_time = creation_time
85
90
  if description is not None:
@@ -139,6 +144,27 @@ class V1LitRepository(object):
139
144
 
140
145
  self._artifacts = artifacts
141
146
 
147
+ @property
148
+ def cluster_id(self) -> 'str':
149
+ """Gets the cluster_id of this V1LitRepository. # noqa: E501
150
+
151
+
152
+ :return: The cluster_id of this V1LitRepository. # noqa: E501
153
+ :rtype: str
154
+ """
155
+ return self._cluster_id
156
+
157
+ @cluster_id.setter
158
+ def cluster_id(self, cluster_id: 'str'):
159
+ """Sets the cluster_id of this V1LitRepository.
160
+
161
+
162
+ :param cluster_id: The cluster_id of this V1LitRepository. # noqa: E501
163
+ :type: str
164
+ """
165
+
166
+ self._cluster_id = cluster_id
167
+
142
168
  @property
143
169
  def creation_time(self) -> 'datetime':
144
170
  """Gets the creation_time of this V1LitRepository. # noqa: E501
@@ -42,6 +42,7 @@ class V1MagicLinkLoginRequest(object):
42
42
  """
43
43
  swagger_types = {
44
44
  'collab_session_id': 'str',
45
+ 'discounted_pro_plan': 'bool',
45
46
  'email': 'str',
46
47
  'experimentation_id': 'str',
47
48
  'invite_code': 'str',
@@ -51,6 +52,7 @@ class V1MagicLinkLoginRequest(object):
51
52
 
52
53
  attribute_map = {
53
54
  'collab_session_id': 'collabSessionId',
55
+ 'discounted_pro_plan': 'discountedProPlan',
54
56
  'email': 'email',
55
57
  'experimentation_id': 'experimentationId',
56
58
  'invite_code': 'inviteCode',
@@ -58,9 +60,10 @@ class V1MagicLinkLoginRequest(object):
58
60
  'referrer_params': 'referrerParams'
59
61
  }
60
62
 
61
- def __init__(self, collab_session_id: 'str' =None, email: 'str' =None, experimentation_id: 'str' =None, invite_code: 'str' =None, redirect_url: 'str' =None, referrer_params: 'str' =None): # noqa: E501
63
+ def __init__(self, collab_session_id: 'str' =None, discounted_pro_plan: 'bool' =None, email: 'str' =None, experimentation_id: 'str' =None, invite_code: 'str' =None, redirect_url: 'str' =None, referrer_params: 'str' =None): # noqa: E501
62
64
  """V1MagicLinkLoginRequest - a model defined in Swagger""" # noqa: E501
63
65
  self._collab_session_id = None
66
+ self._discounted_pro_plan = None
64
67
  self._email = None
65
68
  self._experimentation_id = None
66
69
  self._invite_code = None
@@ -69,6 +72,8 @@ class V1MagicLinkLoginRequest(object):
69
72
  self.discriminator = None
70
73
  if collab_session_id is not None:
71
74
  self.collab_session_id = collab_session_id
75
+ if discounted_pro_plan is not None:
76
+ self.discounted_pro_plan = discounted_pro_plan
72
77
  if email is not None:
73
78
  self.email = email
74
79
  if experimentation_id is not None:
@@ -101,6 +106,27 @@ class V1MagicLinkLoginRequest(object):
101
106
 
102
107
  self._collab_session_id = collab_session_id
103
108
 
109
+ @property
110
+ def discounted_pro_plan(self) -> 'bool':
111
+ """Gets the discounted_pro_plan of this V1MagicLinkLoginRequest. # noqa: E501
112
+
113
+
114
+ :return: The discounted_pro_plan of this V1MagicLinkLoginRequest. # noqa: E501
115
+ :rtype: bool
116
+ """
117
+ return self._discounted_pro_plan
118
+
119
+ @discounted_pro_plan.setter
120
+ def discounted_pro_plan(self, discounted_pro_plan: 'bool'):
121
+ """Sets the discounted_pro_plan of this V1MagicLinkLoginRequest.
122
+
123
+
124
+ :param discounted_pro_plan: The discounted_pro_plan of this V1MagicLinkLoginRequest. # noqa: E501
125
+ :type: bool
126
+ """
127
+
128
+ self._discounted_pro_plan = discounted_pro_plan
129
+
104
130
  @property
105
131
  def email(self) -> 'str':
106
132
  """Gets the email of this V1MagicLinkLoginRequest. # noqa: E501