lightning-sdk 0.1.37__py3-none-any.whl → 0.1.39__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 (72) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/ai_hub.py +21 -23
  3. lightning_sdk/api/ai_hub_api.py +29 -4
  4. lightning_sdk/api/deployment_api.py +0 -2
  5. lightning_sdk/api/job_api.py +10 -2
  6. lightning_sdk/api/teamspace_api.py +22 -16
  7. lightning_sdk/api/utils.py +25 -3
  8. lightning_sdk/cli/ai_hub.py +1 -1
  9. lightning_sdk/cli/download.py +3 -5
  10. lightning_sdk/cli/run.py +24 -0
  11. lightning_sdk/cli/upload.py +3 -10
  12. lightning_sdk/job/base.py +35 -0
  13. lightning_sdk/job/job.py +18 -1
  14. lightning_sdk/job/v1.py +10 -1
  15. lightning_sdk/job/v2.py +16 -0
  16. lightning_sdk/lightning_cloud/openapi/__init__.py +13 -2
  17. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +5 -1
  18. lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py +6 -1
  19. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +680 -62
  20. lightning_sdk/lightning_cloud/openapi/api/models_store_api.py +118 -1
  21. lightning_sdk/lightning_cloud/openapi/models/__init__.py +13 -2
  22. lightning_sdk/lightning_cloud/openapi/models/create.py +6 -32
  23. lightning_sdk/lightning_cloud/openapi/models/deploymenttemplates_id_body.py +32 -6
  24. lightning_sdk/lightning_cloud/openapi/models/externalv1_cloud_space_instance_status.py +27 -1
  25. lightning_sdk/lightning_cloud/openapi/models/id_start_body.py +29 -3
  26. lightning_sdk/lightning_cloud/openapi/models/multimachinejobs_id_body.py +123 -0
  27. lightning_sdk/lightning_cloud/openapi/models/project_id_agents_body.py +53 -1
  28. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +53 -1
  29. lightning_sdk/lightning_cloud/openapi/models/project_id_multimachinejobs_body.py +201 -0
  30. lightning_sdk/lightning_cloud/openapi/models/update.py +6 -32
  31. lightning_sdk/lightning_cloud/openapi/models/v1_api_pricing_spec.py +149 -0
  32. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  33. lightning_sdk/lightning_cloud/openapi/models/v1_create_deployment_template_request.py +32 -6
  34. lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py +6 -32
  35. lightning_sdk/lightning_cloud/openapi/models/v1_data_path.py +29 -3
  36. lightning_sdk/lightning_cloud/openapi/models/v1_delete_multi_machine_job_response.py +97 -0
  37. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_metrics.py +43 -17
  38. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_performance.py +305 -0
  39. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template.py +32 -6
  40. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_parameter.py +27 -1
  41. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_parameter_type.py +1 -0
  42. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_summary.py +27 -1
  43. lightning_sdk/lightning_cloud/openapi/models/{v1_efs_data_connection.py → v1_efs_config.py} +22 -22
  44. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_response.py +27 -1
  45. lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py +53 -1
  46. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +53 -53
  47. lightning_sdk/lightning_cloud/openapi/models/v1_lambda_labs_direct_v1.py +125 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_list_multi_machine_jobs_response.py +123 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_machines_selector.py +149 -0
  50. lightning_sdk/lightning_cloud/openapi/models/v1_message.py +6 -6
  51. lightning_sdk/lightning_cloud/openapi/models/v1_message_content.py +6 -6
  52. lightning_sdk/lightning_cloud/openapi/models/v1_message_content_type.py +103 -0
  53. lightning_sdk/lightning_cloud/openapi/models/v1_metrics_stream.py +53 -1
  54. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py +409 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_state.py +106 -0
  56. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_status.py +279 -0
  57. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +2 -0
  58. lightning_sdk/lightning_cloud/openapi/models/v1_system_info.py +27 -1
  59. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +53 -1
  60. lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_compute_config.py +27 -1
  61. lightning_sdk/lightning_cloud/openapi/models/v1_validate_data_connection_response.py +6 -32
  62. lightning_sdk/lightning_cloud/openapi/models/validate.py +6 -32
  63. lightning_sdk/models.py +132 -0
  64. lightning_sdk/teamspace.py +8 -2
  65. {lightning_sdk-0.1.37.dist-info → lightning_sdk-0.1.39.dist-info}/METADATA +1 -1
  66. {lightning_sdk-0.1.37.dist-info → lightning_sdk-0.1.39.dist-info}/RECORD +70 -59
  67. lightning_sdk/cli/models.py +0 -68
  68. lightning_sdk/lightning_cloud/openapi/models/v1_efs_folder_data_connection.py +0 -201
  69. {lightning_sdk-0.1.37.dist-info → lightning_sdk-0.1.39.dist-info}/LICENSE +0 -0
  70. {lightning_sdk-0.1.37.dist-info → lightning_sdk-0.1.39.dist-info}/WHEEL +0 -0
  71. {lightning_sdk-0.1.37.dist-info → lightning_sdk-0.1.39.dist-info}/entry_points.txt +0 -0
  72. {lightning_sdk-0.1.37.dist-info → lightning_sdk-0.1.39.dist-info}/top_level.txt +0 -0
@@ -1197,6 +1197,8 @@ class ModelsStoreApi(object):
1197
1197
  :param str name:
1198
1198
  :param str version:
1199
1199
  :param str project_id:
1200
+ :param str project_name:
1201
+ :param str project_owner_name:
1200
1202
  :return: V1GetModelFilesResponse
1201
1203
  If the method is called asynchronously,
1202
1204
  returns the request thread.
@@ -1220,12 +1222,14 @@ class ModelsStoreApi(object):
1220
1222
  :param str name:
1221
1223
  :param str version:
1222
1224
  :param str project_id:
1225
+ :param str project_name:
1226
+ :param str project_owner_name:
1223
1227
  :return: V1GetModelFilesResponse
1224
1228
  If the method is called asynchronously,
1225
1229
  returns the request thread.
1226
1230
  """
1227
1231
 
1228
- all_params = ['name', 'version', 'project_id'] # noqa: E501
1232
+ all_params = ['name', 'version', 'project_id', 'project_name', 'project_owner_name'] # noqa: E501
1229
1233
  all_params.append('async_req')
1230
1234
  all_params.append('_return_http_data_only')
1231
1235
  all_params.append('_preload_content')
@@ -1252,6 +1256,10 @@ class ModelsStoreApi(object):
1252
1256
  query_params.append(('version', params['version'])) # noqa: E501
1253
1257
  if 'project_id' in params:
1254
1258
  query_params.append(('projectId', params['project_id'])) # noqa: E501
1259
+ if 'project_name' in params:
1260
+ query_params.append(('projectName', params['project_name'])) # noqa: E501
1261
+ if 'project_owner_name' in params:
1262
+ query_params.append(('projectOwnerName', params['project_owner_name'])) # noqa: E501
1255
1263
 
1256
1264
  header_params = {}
1257
1265
 
@@ -1282,6 +1290,115 @@ class ModelsStoreApi(object):
1282
1290
  _request_timeout=params.get('_request_timeout'),
1283
1291
  collection_formats=collection_formats)
1284
1292
 
1293
+ def models_store_get_model_version(self, project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1ModelVersionArchive': # noqa: E501
1294
+ """GetModelVersion used to get specific model version details # noqa: E501
1295
+
1296
+ This method makes a synchronous HTTP request by default. To make an
1297
+ asynchronous HTTP request, please pass async_req=True
1298
+ >>> thread = api.models_store_get_model_version(project_id, model_id, version, async_req=True)
1299
+ >>> result = thread.get()
1300
+
1301
+ :param async_req bool
1302
+ :param str project_id: (required)
1303
+ :param str model_id: (required)
1304
+ :param str version: (required)
1305
+ :return: V1ModelVersionArchive
1306
+ If the method is called asynchronously,
1307
+ returns the request thread.
1308
+ """
1309
+ kwargs['_return_http_data_only'] = True
1310
+ if kwargs.get('async_req'):
1311
+ return self.models_store_get_model_version_with_http_info(project_id, model_id, version, **kwargs) # noqa: E501
1312
+ else:
1313
+ (data) = self.models_store_get_model_version_with_http_info(project_id, model_id, version, **kwargs) # noqa: E501
1314
+ return data
1315
+
1316
+ def models_store_get_model_version_with_http_info(self, project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1ModelVersionArchive': # noqa: E501
1317
+ """GetModelVersion used to get specific model version details # noqa: E501
1318
+
1319
+ This method makes a synchronous HTTP request by default. To make an
1320
+ asynchronous HTTP request, please pass async_req=True
1321
+ >>> thread = api.models_store_get_model_version_with_http_info(project_id, model_id, version, async_req=True)
1322
+ >>> result = thread.get()
1323
+
1324
+ :param async_req bool
1325
+ :param str project_id: (required)
1326
+ :param str model_id: (required)
1327
+ :param str version: (required)
1328
+ :return: V1ModelVersionArchive
1329
+ If the method is called asynchronously,
1330
+ returns the request thread.
1331
+ """
1332
+
1333
+ all_params = ['project_id', 'model_id', 'version'] # noqa: E501
1334
+ all_params.append('async_req')
1335
+ all_params.append('_return_http_data_only')
1336
+ all_params.append('_preload_content')
1337
+ all_params.append('_request_timeout')
1338
+
1339
+ params = locals()
1340
+ for key, val in six.iteritems(params['kwargs']):
1341
+ if key not in all_params:
1342
+ raise TypeError(
1343
+ "Got an unexpected keyword argument '%s'"
1344
+ " to method models_store_get_model_version" % key
1345
+ )
1346
+ params[key] = val
1347
+ del params['kwargs']
1348
+ # verify the required parameter 'project_id' is set
1349
+ if ('project_id' not in params or
1350
+ params['project_id'] is None):
1351
+ raise ValueError("Missing the required parameter `project_id` when calling `models_store_get_model_version`") # noqa: E501
1352
+ # verify the required parameter 'model_id' is set
1353
+ if ('model_id' not in params or
1354
+ params['model_id'] is None):
1355
+ raise ValueError("Missing the required parameter `model_id` when calling `models_store_get_model_version`") # noqa: E501
1356
+ # verify the required parameter 'version' is set
1357
+ if ('version' not in params or
1358
+ params['version'] is None):
1359
+ raise ValueError("Missing the required parameter `version` when calling `models_store_get_model_version`") # noqa: E501
1360
+
1361
+ collection_formats = {}
1362
+
1363
+ path_params = {}
1364
+ if 'project_id' in params:
1365
+ path_params['projectId'] = params['project_id'] # noqa: E501
1366
+ if 'model_id' in params:
1367
+ path_params['modelId'] = params['model_id'] # noqa: E501
1368
+ if 'version' in params:
1369
+ path_params['version'] = params['version'] # noqa: E501
1370
+
1371
+ query_params = []
1372
+
1373
+ header_params = {}
1374
+
1375
+ form_params = []
1376
+ local_var_files = {}
1377
+
1378
+ body_params = None
1379
+ # HTTP header `Accept`
1380
+ header_params['Accept'] = self.api_client.select_header_accept(
1381
+ ['application/json']) # noqa: E501
1382
+
1383
+ # Authentication setting
1384
+ auth_settings = [] # noqa: E501
1385
+
1386
+ return self.api_client.call_api(
1387
+ '/v1/projects/{projectId}/models/{modelId}/versions/{version}', 'GET',
1388
+ path_params,
1389
+ query_params,
1390
+ header_params,
1391
+ body=body_params,
1392
+ post_params=form_params,
1393
+ files=local_var_files,
1394
+ response_type='V1ModelVersionArchive', # noqa: E501
1395
+ auth_settings=auth_settings,
1396
+ async_req=params.get('async_req'),
1397
+ _return_http_data_only=params.get('_return_http_data_only'),
1398
+ _preload_content=params.get('_preload_content', True),
1399
+ _request_timeout=params.get('_request_timeout'),
1400
+ collection_formats=collection_formats)
1401
+
1285
1402
  def models_store_list_model_versions(self, project_id: 'str', model_id: 'str', **kwargs) -> 'V1ListModelVersionsResponse': # noqa: E501
1286
1403
  """models_store_list_model_versions # noqa: E501
1287
1404
 
@@ -102,6 +102,7 @@ from lightning_sdk.lightning_cloud.openapi.models.metricsstream_delete_body impo
102
102
  from lightning_sdk.lightning_cloud.openapi.models.metricsstream_id_body import MetricsstreamIdBody
103
103
  from lightning_sdk.lightning_cloud.openapi.models.model_id_versions_body import ModelIdVersionsBody
104
104
  from lightning_sdk.lightning_cloud.openapi.models.models_model_id_body import ModelsModelIdBody
105
+ from lightning_sdk.lightning_cloud.openapi.models.multimachinejobs_id_body import MultimachinejobsIdBody
105
106
  from lightning_sdk.lightning_cloud.openapi.models.multipartuploads_upload_id_body import MultipartuploadsUploadIdBody
106
107
  from lightning_sdk.lightning_cloud.openapi.models.org_id_memberships_body import OrgIdMembershipsBody
107
108
  from lightning_sdk.lightning_cloud.openapi.models.org_id_roles_body import OrgIdRolesBody
@@ -121,6 +122,7 @@ from lightning_sdk.lightning_cloud.openapi.models.project_id_invite_body import
121
122
  from lightning_sdk.lightning_cloud.openapi.models.project_id_jobs_body import ProjectIdJobsBody
122
123
  from lightning_sdk.lightning_cloud.openapi.models.project_id_memberships_body import ProjectIdMembershipsBody
123
124
  from lightning_sdk.lightning_cloud.openapi.models.project_id_models_body import ProjectIdModelsBody
125
+ from lightning_sdk.lightning_cloud.openapi.models.project_id_multimachinejobs_body import ProjectIdMultimachinejobsBody
124
126
  from lightning_sdk.lightning_cloud.openapi.models.project_id_projectclustersbindings_body import ProjectIdProjectclustersbindingsBody
125
127
  from lightning_sdk.lightning_cloud.openapi.models.project_id_secrets_body import ProjectIdSecretsBody
126
128
  from lightning_sdk.lightning_cloud.openapi.models.project_id_serviceexecution_body import ProjectIdServiceexecutionBody
@@ -164,6 +166,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_agent_job import V1AgentJob
164
166
  from lightning_sdk.lightning_cloud.openapi.models.v1_agent_job_artifact import V1AgentJobArtifact
165
167
  from lightning_sdk.lightning_cloud.openapi.models.v1_agent_upload_multipart_url import V1AgentUploadMultipartUrl
166
168
  from lightning_sdk.lightning_cloud.openapi.models.v1_agent_upload_part_response import V1AgentUploadPartResponse
169
+ from lightning_sdk.lightning_cloud.openapi.models.v1_api_pricing_spec import V1ApiPricingSpec
167
170
  from lightning_sdk.lightning_cloud.openapi.models.v1_app_type import V1AppType
168
171
  from lightning_sdk.lightning_cloud.openapi.models.v1_append_logger_metrics_response import V1AppendLoggerMetricsResponse
169
172
  from lightning_sdk.lightning_cloud.openapi.models.v1_approve_auto_join_domain_response import V1ApproveAutoJoinDomainResponse
@@ -304,6 +307,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_managed_model_respon
304
307
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_metrics_stream_response import V1DeleteMetricsStreamResponse
305
308
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_model_response import V1DeleteModelResponse
306
309
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_model_version_response import V1DeleteModelVersionResponse
310
+ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_multi_machine_job_response import V1DeleteMultiMachineJobResponse
307
311
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_membership_response import V1DeleteOrgMembershipResponse
308
312
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_membership_role_binding_response import V1DeleteOrgMembershipRoleBindingResponse
309
313
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_role_response import V1DeleteOrgRoleResponse
@@ -328,6 +332,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment import V1Deploym
328
332
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_event import V1DeploymentEvent
329
333
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_event_type import V1DeploymentEventType
330
334
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_metrics import V1DeploymentMetrics
335
+ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_performance import V1DeploymentPerformance
331
336
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_release import V1DeploymentRelease
332
337
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_spec import V1DeploymentSpec
333
338
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_state import V1DeploymentState
@@ -352,9 +357,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_drive_status import V1Drive
352
357
  from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type import V1DriveType
353
358
  from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type_spec import V1DriveTypeSpec
354
359
  from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type_status import V1DriveTypeStatus
355
- from lightning_sdk.lightning_cloud.openapi.models.v1_efs_folder_data_connection import V1EFSFolderDataConnection
356
360
  from lightning_sdk.lightning_cloud.openapi.models.v1_ebs import V1Ebs
357
- from lightning_sdk.lightning_cloud.openapi.models.v1_efs_data_connection import V1EfsDataConnection
361
+ from lightning_sdk.lightning_cloud.openapi.models.v1_efs_config import V1EfsConfig
358
362
  from lightning_sdk.lightning_cloud.openapi.models.v1_endpoint import V1Endpoint
359
363
  from lightning_sdk.lightning_cloud.openapi.models.v1_endpoint_auth import V1EndpointAuth
360
364
  from lightning_sdk.lightning_cloud.openapi.models.v1_endpoint_prewarm import V1EndpointPrewarm
@@ -442,6 +446,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_job_timing import V1JobTimi
442
446
  from lightning_sdk.lightning_cloud.openapi.models.v1_joinable_organization import V1JoinableOrganization
443
447
  from lightning_sdk.lightning_cloud.openapi.models.v1_keep_alive_cloud_space_instance_response import V1KeepAliveCloudSpaceInstanceResponse
444
448
  from lightning_sdk.lightning_cloud.openapi.models.v1_knowledge_configuration import V1KnowledgeConfiguration
449
+ from lightning_sdk.lightning_cloud.openapi.models.v1_lambda_labs_direct_v1 import V1LambdaLabsDirectV1
445
450
  from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_app_user import V1LightningAppUser
446
451
  from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_auth import V1LightningAuth
447
452
  from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_basic_auth import V1LightningBasicAuth
@@ -516,6 +521,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_memberships_response i
516
521
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_metrics_streams_response import V1ListMetricsStreamsResponse
517
522
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_model_versions_response import V1ListModelVersionsResponse
518
523
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_models_response import V1ListModelsResponse
524
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_multi_machine_jobs_response import V1ListMultiMachineJobsResponse
519
525
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_new_features_for_user_response import V1ListNewFeaturesForUserResponse
520
526
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_members_response import V1ListOrgMembersResponse
521
527
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_membership_role_binding_response import V1ListOrgMembershipRoleBindingResponse
@@ -551,6 +557,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_login_response import V1Log
551
557
  from lightning_sdk.lightning_cloud.openapi.models.v1_logout_request import V1LogoutRequest
552
558
  from lightning_sdk.lightning_cloud.openapi.models.v1_logout_response import V1LogoutResponse
553
559
  from lightning_sdk.lightning_cloud.openapi.models.v1_logs_response import V1LogsResponse
560
+ from lightning_sdk.lightning_cloud.openapi.models.v1_machines_selector import V1MachinesSelector
554
561
  from lightning_sdk.lightning_cloud.openapi.models.v1_magic_link_login_request import V1MagicLinkLoginRequest
555
562
  from lightning_sdk.lightning_cloud.openapi.models.v1_magic_link_login_response import V1MagicLinkLoginResponse
556
563
  from lightning_sdk.lightning_cloud.openapi.models.v1_managed_endpoint import V1ManagedEndpoint
@@ -559,6 +566,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_membership import V1Members
559
566
  from lightning_sdk.lightning_cloud.openapi.models.v1_message import V1Message
560
567
  from lightning_sdk.lightning_cloud.openapi.models.v1_message_author import V1MessageAuthor
561
568
  from lightning_sdk.lightning_cloud.openapi.models.v1_message_content import V1MessageContent
569
+ from lightning_sdk.lightning_cloud.openapi.models.v1_message_content_type import V1MessageContentType
562
570
  from lightning_sdk.lightning_cloud.openapi.models.v1_metadata import V1Metadata
563
571
  from lightning_sdk.lightning_cloud.openapi.models.v1_metric_value import V1MetricValue
564
572
  from lightning_sdk.lightning_cloud.openapi.models.v1_metrics import V1Metrics
@@ -568,6 +576,9 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_tracker import V1Me
568
576
  from lightning_sdk.lightning_cloud.openapi.models.v1_model import V1Model
569
577
  from lightning_sdk.lightning_cloud.openapi.models.v1_model_version_archive import V1ModelVersionArchive
570
578
  from lightning_sdk.lightning_cloud.openapi.models.v1_mount_target import V1MountTarget
579
+ from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job import V1MultiMachineJob
580
+ from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_state import V1MultiMachineJobState
581
+ from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_status import V1MultiMachineJobStatus
571
582
  from lightning_sdk.lightning_cloud.openapi.models.v1_named_get_logger_metrics import V1NamedGetLoggerMetrics
572
583
  from lightning_sdk.lightning_cloud.openapi.models.v1_network_config import V1NetworkConfig
573
584
  from lightning_sdk.lightning_cloud.openapi.models.v1_new_feature import V1NewFeature
@@ -46,8 +46,7 @@ class Create(object):
46
46
  'cluster_id': 'str',
47
47
  'create_index': 'bool',
48
48
  'create_resources': 'bool',
49
- 'efs': 'V1EfsDataConnection',
50
- 'efs_folder': 'V1EFSFolderDataConnection',
49
+ 'efs': 'V1EfsConfig',
51
50
  'force': 'bool',
52
51
  'gcp': 'V1GcpDataConnection',
53
52
  'gcs_folder': 'V1GCSFolderDataConnection',
@@ -66,7 +65,6 @@ class Create(object):
66
65
  'create_index': 'createIndex',
67
66
  'create_resources': 'createResources',
68
67
  'efs': 'efs',
69
- 'efs_folder': 'efsFolder',
70
68
  'force': 'force',
71
69
  'gcp': 'gcp',
72
70
  'gcs_folder': 'gcsFolder',
@@ -78,7 +76,7 @@ class Create(object):
78
76
  'writable': 'writable'
79
77
  }
80
78
 
81
- def __init__(self, access_cluster_ids: 'list[str]' =None, aws: 'V1AwsDataConnection' =None, cluster_id: 'str' =None, create_index: 'bool' =None, create_resources: 'bool' =None, efs: 'V1EfsDataConnection' =None, efs_folder: 'V1EFSFolderDataConnection' =None, force: 'bool' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, id: 'str' =None, name: 'str' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, writable: 'bool' =None): # noqa: E501
79
+ def __init__(self, access_cluster_ids: 'list[str]' =None, aws: 'V1AwsDataConnection' =None, cluster_id: 'str' =None, create_index: 'bool' =None, create_resources: 'bool' =None, efs: 'V1EfsConfig' =None, force: 'bool' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, id: 'str' =None, name: 'str' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, writable: 'bool' =None): # noqa: E501
82
80
  """Create - a model defined in Swagger""" # noqa: E501
83
81
  self._access_cluster_ids = None
84
82
  self._aws = None
@@ -86,7 +84,6 @@ class Create(object):
86
84
  self._create_index = None
87
85
  self._create_resources = None
88
86
  self._efs = None
89
- self._efs_folder = None
90
87
  self._force = None
91
88
  self._gcp = None
92
89
  self._gcs_folder = None
@@ -109,8 +106,6 @@ class Create(object):
109
106
  self.create_resources = create_resources
110
107
  if efs is not None:
111
108
  self.efs = efs
112
- if efs_folder is not None:
113
- self.efs_folder = efs_folder
114
109
  if force is not None:
115
110
  self.force = force
116
111
  if gcp is not None:
@@ -236,47 +231,26 @@ class Create(object):
236
231
  self._create_resources = create_resources
237
232
 
238
233
  @property
239
- def efs(self) -> 'V1EfsDataConnection':
234
+ def efs(self) -> 'V1EfsConfig':
240
235
  """Gets the efs of this Create. # noqa: E501
241
236
 
242
237
 
243
238
  :return: The efs of this Create. # noqa: E501
244
- :rtype: V1EfsDataConnection
239
+ :rtype: V1EfsConfig
245
240
  """
246
241
  return self._efs
247
242
 
248
243
  @efs.setter
249
- def efs(self, efs: 'V1EfsDataConnection'):
244
+ def efs(self, efs: 'V1EfsConfig'):
250
245
  """Sets the efs of this Create.
251
246
 
252
247
 
253
248
  :param efs: The efs of this Create. # noqa: E501
254
- :type: V1EfsDataConnection
249
+ :type: V1EfsConfig
255
250
  """
256
251
 
257
252
  self._efs = efs
258
253
 
259
- @property
260
- def efs_folder(self) -> 'V1EFSFolderDataConnection':
261
- """Gets the efs_folder of this Create. # noqa: E501
262
-
263
-
264
- :return: The efs_folder of this Create. # noqa: E501
265
- :rtype: V1EFSFolderDataConnection
266
- """
267
- return self._efs_folder
268
-
269
- @efs_folder.setter
270
- def efs_folder(self, efs_folder: 'V1EFSFolderDataConnection'):
271
- """Sets the efs_folder of this Create.
272
-
273
-
274
- :param efs_folder: The efs_folder of this Create. # noqa: E501
275
- :type: V1EFSFolderDataConnection
276
- """
277
-
278
- self._efs_folder = efs_folder
279
-
280
254
  @property
281
255
  def force(self) -> 'bool':
282
256
  """Gets the force of this Create. # noqa: E501
@@ -46,10 +46,11 @@ class DeploymenttemplatesIdBody(object):
46
46
  'description': 'str',
47
47
  'featured': 'bool',
48
48
  'image_url': 'str',
49
- 'metrics': 'list[V1DeploymentMetrics]',
49
+ 'metrics': 'V1DeploymentMetrics',
50
50
  'name': 'str',
51
51
  'org_id': 'str',
52
52
  'parameter_spec': 'V1ParameterizationSpec',
53
+ 'pricing': 'V1ApiPricingSpec',
53
54
  'spec': 'str',
54
55
  'tags': 'list[V1ResourceTag]',
55
56
  'thumbnail': 'str',
@@ -67,6 +68,7 @@ class DeploymenttemplatesIdBody(object):
67
68
  'name': 'name',
68
69
  'org_id': 'orgId',
69
70
  'parameter_spec': 'parameterSpec',
71
+ 'pricing': 'pricing',
70
72
  'spec': 'spec',
71
73
  'tags': 'tags',
72
74
  'thumbnail': 'thumbnail',
@@ -74,7 +76,7 @@ class DeploymenttemplatesIdBody(object):
74
76
  'visibility': 'visibility'
75
77
  }
76
78
 
77
- def __init__(self, about_page_content: 'str' =None, categories: 'list[str]' =None, description: 'str' =None, featured: 'bool' =None, image_url: 'str' =None, metrics: 'list[V1DeploymentMetrics]' =None, name: 'str' =None, org_id: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, spec: 'str' =None, tags: 'list[V1ResourceTag]' =None, thumbnail: 'str' =None, thumbnail_file_type: 'str' =None, visibility: 'V1DeploymentTemplateType' =None): # noqa: E501
79
+ def __init__(self, about_page_content: 'str' =None, categories: 'list[str]' =None, description: 'str' =None, featured: 'bool' =None, image_url: 'str' =None, metrics: 'V1DeploymentMetrics' =None, name: 'str' =None, org_id: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, pricing: 'V1ApiPricingSpec' =None, spec: 'str' =None, tags: 'list[V1ResourceTag]' =None, thumbnail: 'str' =None, thumbnail_file_type: 'str' =None, visibility: 'V1DeploymentTemplateType' =None): # noqa: E501
78
80
  """DeploymenttemplatesIdBody - a model defined in Swagger""" # noqa: E501
79
81
  self._about_page_content = None
80
82
  self._categories = None
@@ -85,6 +87,7 @@ class DeploymenttemplatesIdBody(object):
85
87
  self._name = None
86
88
  self._org_id = None
87
89
  self._parameter_spec = None
90
+ self._pricing = None
88
91
  self._spec = None
89
92
  self._tags = None
90
93
  self._thumbnail = None
@@ -109,6 +112,8 @@ class DeploymenttemplatesIdBody(object):
109
112
  self.org_id = org_id
110
113
  if parameter_spec is not None:
111
114
  self.parameter_spec = parameter_spec
115
+ if pricing is not None:
116
+ self.pricing = pricing
112
117
  if spec is not None:
113
118
  self.spec = spec
114
119
  if tags is not None:
@@ -226,22 +231,22 @@ class DeploymenttemplatesIdBody(object):
226
231
  self._image_url = image_url
227
232
 
228
233
  @property
229
- def metrics(self) -> 'list[V1DeploymentMetrics]':
234
+ def metrics(self) -> 'V1DeploymentMetrics':
230
235
  """Gets the metrics of this DeploymenttemplatesIdBody. # noqa: E501
231
236
 
232
237
 
233
238
  :return: The metrics of this DeploymenttemplatesIdBody. # noqa: E501
234
- :rtype: list[V1DeploymentMetrics]
239
+ :rtype: V1DeploymentMetrics
235
240
  """
236
241
  return self._metrics
237
242
 
238
243
  @metrics.setter
239
- def metrics(self, metrics: 'list[V1DeploymentMetrics]'):
244
+ def metrics(self, metrics: 'V1DeploymentMetrics'):
240
245
  """Sets the metrics of this DeploymenttemplatesIdBody.
241
246
 
242
247
 
243
248
  :param metrics: The metrics of this DeploymenttemplatesIdBody. # noqa: E501
244
- :type: list[V1DeploymentMetrics]
249
+ :type: V1DeploymentMetrics
245
250
  """
246
251
 
247
252
  self._metrics = metrics
@@ -309,6 +314,27 @@ class DeploymenttemplatesIdBody(object):
309
314
 
310
315
  self._parameter_spec = parameter_spec
311
316
 
317
+ @property
318
+ def pricing(self) -> 'V1ApiPricingSpec':
319
+ """Gets the pricing of this DeploymenttemplatesIdBody. # noqa: E501
320
+
321
+
322
+ :return: The pricing of this DeploymenttemplatesIdBody. # noqa: E501
323
+ :rtype: V1ApiPricingSpec
324
+ """
325
+ return self._pricing
326
+
327
+ @pricing.setter
328
+ def pricing(self, pricing: 'V1ApiPricingSpec'):
329
+ """Sets the pricing of this DeploymenttemplatesIdBody.
330
+
331
+
332
+ :param pricing: The pricing of this DeploymenttemplatesIdBody. # noqa: E501
333
+ :type: V1ApiPricingSpec
334
+ """
335
+
336
+ self._pricing = pricing
337
+
312
338
  @property
313
339
  def spec(self) -> 'str':
314
340
  """Gets the spec of this DeploymenttemplatesIdBody. # noqa: E501
@@ -68,6 +68,7 @@ class Externalv1CloudSpaceInstanceStatus(object):
68
68
  'sync_eta_seconds': 'str',
69
69
  'sync_in_progress': 'bool',
70
70
  'sync_percentage': 'str',
71
+ 'termination_time': 'datetime',
71
72
  'vscode_url': 'str'
72
73
  }
73
74
 
@@ -99,10 +100,11 @@ class Externalv1CloudSpaceInstanceStatus(object):
99
100
  'sync_eta_seconds': 'syncEtaSeconds',
100
101
  'sync_in_progress': 'syncInProgress',
101
102
  'sync_percentage': 'syncPercentage',
103
+ 'termination_time': 'terminationTime',
102
104
  'vscode_url': 'vscodeUrl'
103
105
  }
104
106
 
105
- def __init__(self, app_url: 'str' =None, cloud_space_id: 'str' =None, cloud_space_instance_id: 'str' =None, compute_config: 'V1UserRequestedComputeConfig' =None, creation_timestamp: 'datetime' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, free: 'bool' =None, ide: 'str' =None, instance_id: 'str' =None, instance_region: 'str' =None, instance_url: 'str' =None, jupyterlab_url: 'str' =None, phase: 'V1CloudSpaceInstanceState' =None, price: 'float' =None, ssh_host: 'str' =None, ssh_port: 'int' =None, ssh_username: 'str' =None, start_timestamp: 'datetime' =None, startup_eta_seconds: 'str' =None, startup_percentage: 'str' =None, startup_phase: 'str' =None, startup_status: 'V1CloudSpaceInstanceStartupStatus' =None, status_message: 'str' =None, switched_at: 'datetime' =None, sync_eta_seconds: 'str' =None, sync_in_progress: 'bool' =None, sync_percentage: 'str' =None, vscode_url: 'str' =None): # noqa: E501
107
+ def __init__(self, app_url: 'str' =None, cloud_space_id: 'str' =None, cloud_space_instance_id: 'str' =None, compute_config: 'V1UserRequestedComputeConfig' =None, creation_timestamp: 'datetime' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, free: 'bool' =None, ide: 'str' =None, instance_id: 'str' =None, instance_region: 'str' =None, instance_url: 'str' =None, jupyterlab_url: 'str' =None, phase: 'V1CloudSpaceInstanceState' =None, price: 'float' =None, ssh_host: 'str' =None, ssh_port: 'int' =None, ssh_username: 'str' =None, start_timestamp: 'datetime' =None, startup_eta_seconds: 'str' =None, startup_percentage: 'str' =None, startup_phase: 'str' =None, startup_status: 'V1CloudSpaceInstanceStartupStatus' =None, status_message: 'str' =None, switched_at: 'datetime' =None, sync_eta_seconds: 'str' =None, sync_in_progress: 'bool' =None, sync_percentage: 'str' =None, termination_time: 'datetime' =None, vscode_url: 'str' =None): # noqa: E501
106
108
  """Externalv1CloudSpaceInstanceStatus - a model defined in Swagger""" # noqa: E501
107
109
  self._app_url = None
108
110
  self._cloud_space_id = None
@@ -131,6 +133,7 @@ class Externalv1CloudSpaceInstanceStatus(object):
131
133
  self._sync_eta_seconds = None
132
134
  self._sync_in_progress = None
133
135
  self._sync_percentage = None
136
+ self._termination_time = None
134
137
  self._vscode_url = None
135
138
  self.discriminator = None
136
139
  if app_url is not None:
@@ -187,6 +190,8 @@ class Externalv1CloudSpaceInstanceStatus(object):
187
190
  self.sync_in_progress = sync_in_progress
188
191
  if sync_percentage is not None:
189
192
  self.sync_percentage = sync_percentage
193
+ if termination_time is not None:
194
+ self.termination_time = termination_time
190
195
  if vscode_url is not None:
191
196
  self.vscode_url = vscode_url
192
197
 
@@ -759,6 +764,27 @@ class Externalv1CloudSpaceInstanceStatus(object):
759
764
 
760
765
  self._sync_percentage = sync_percentage
761
766
 
767
+ @property
768
+ def termination_time(self) -> 'datetime':
769
+ """Gets the termination_time of this Externalv1CloudSpaceInstanceStatus. # noqa: E501
770
+
771
+
772
+ :return: The termination_time of this Externalv1CloudSpaceInstanceStatus. # noqa: E501
773
+ :rtype: datetime
774
+ """
775
+ return self._termination_time
776
+
777
+ @termination_time.setter
778
+ def termination_time(self, termination_time: 'datetime'):
779
+ """Sets the termination_time of this Externalv1CloudSpaceInstanceStatus.
780
+
781
+
782
+ :param termination_time: The termination_time of this Externalv1CloudSpaceInstanceStatus. # noqa: E501
783
+ :type: datetime
784
+ """
785
+
786
+ self._termination_time = termination_time
787
+
762
788
  @property
763
789
  def vscode_url(self) -> 'str':
764
790
  """Gets the vscode_url of this Externalv1CloudSpaceInstanceStatus. # noqa: E501
@@ -41,19 +41,24 @@ class IdStartBody(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
- 'compute_config': 'V1UserRequestedComputeConfig'
44
+ 'compute_config': 'V1UserRequestedComputeConfig',
45
+ 'is_forked': 'bool'
45
46
  }
46
47
 
47
48
  attribute_map = {
48
- 'compute_config': 'computeConfig'
49
+ 'compute_config': 'computeConfig',
50
+ 'is_forked': 'isForked'
49
51
  }
50
52
 
51
- def __init__(self, compute_config: 'V1UserRequestedComputeConfig' =None): # noqa: E501
53
+ def __init__(self, compute_config: 'V1UserRequestedComputeConfig' =None, is_forked: 'bool' =None): # noqa: E501
52
54
  """IdStartBody - a model defined in Swagger""" # noqa: E501
53
55
  self._compute_config = None
56
+ self._is_forked = None
54
57
  self.discriminator = None
55
58
  if compute_config is not None:
56
59
  self.compute_config = compute_config
60
+ if is_forked is not None:
61
+ self.is_forked = is_forked
57
62
 
58
63
  @property
59
64
  def compute_config(self) -> 'V1UserRequestedComputeConfig':
@@ -76,6 +81,27 @@ class IdStartBody(object):
76
81
 
77
82
  self._compute_config = compute_config
78
83
 
84
+ @property
85
+ def is_forked(self) -> 'bool':
86
+ """Gets the is_forked of this IdStartBody. # noqa: E501
87
+
88
+
89
+ :return: The is_forked of this IdStartBody. # noqa: E501
90
+ :rtype: bool
91
+ """
92
+ return self._is_forked
93
+
94
+ @is_forked.setter
95
+ def is_forked(self, is_forked: 'bool'):
96
+ """Sets the is_forked of this IdStartBody.
97
+
98
+
99
+ :param is_forked: The is_forked of this IdStartBody. # noqa: E501
100
+ :type: bool
101
+ """
102
+
103
+ self._is_forked = is_forked
104
+
79
105
  def to_dict(self) -> dict:
80
106
  """Returns the model properties as a dict"""
81
107
  result = {}