lightning-sdk 0.1.41__py3-none-any.whl → 0.1.43__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 (77) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/ai_hub.py +8 -3
  3. lightning_sdk/api/ai_hub_api.py +3 -3
  4. lightning_sdk/api/deployment_api.py +6 -6
  5. lightning_sdk/api/job_api.py +32 -6
  6. lightning_sdk/api/mmt_api.py +59 -19
  7. lightning_sdk/api/studio_api.py +37 -19
  8. lightning_sdk/api/teamspace_api.py +34 -29
  9. lightning_sdk/api/utils.py +46 -34
  10. lightning_sdk/cli/ai_hub.py +3 -3
  11. lightning_sdk/cli/entrypoint.py +3 -1
  12. lightning_sdk/cli/run.py +122 -12
  13. lightning_sdk/cli/serve.py +218 -0
  14. lightning_sdk/deployment/deployment.py +18 -12
  15. lightning_sdk/job/base.py +118 -24
  16. lightning_sdk/job/job.py +98 -9
  17. lightning_sdk/job/v1.py +75 -18
  18. lightning_sdk/job/v2.py +51 -15
  19. lightning_sdk/job/work.py +36 -7
  20. lightning_sdk/lightning_cloud/openapi/__init__.py +12 -0
  21. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +215 -5
  22. lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py +218 -0
  23. lightning_sdk/lightning_cloud/openapi/api/models_store_api.py +226 -0
  24. lightning_sdk/lightning_cloud/openapi/api/snowflake_service_api.py +21 -1
  25. lightning_sdk/lightning_cloud/openapi/models/__init__.py +12 -0
  26. lightning_sdk/lightning_cloud/openapi/models/deploymenttemplates_id_body.py +27 -1
  27. lightning_sdk/lightning_cloud/openapi/models/id_visibility_body.py +123 -0
  28. lightning_sdk/lightning_cloud/openapi/models/model_id_versions_body.py +29 -3
  29. lightning_sdk/lightning_cloud/openapi/models/project_id_multimachinejobs_body.py +27 -1
  30. lightning_sdk/lightning_cloud/openapi/models/project_id_snowflake_body.py +15 -67
  31. lightning_sdk/lightning_cloud/openapi/models/query_query_id_body.py +17 -69
  32. lightning_sdk/lightning_cloud/openapi/models/snowflake_export_body.py +29 -81
  33. lightning_sdk/lightning_cloud/openapi/models/snowflake_query_body.py +17 -69
  34. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_file_url_response.py +27 -1
  35. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_response.py +17 -17
  36. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_url_response.py +149 -0
  37. lightning_sdk/lightning_cloud/openapi/models/v1_get_project_balance_response.py +27 -1
  38. lightning_sdk/lightning_cloud/openapi/models/v1_list_multi_machine_job_events_response.py +123 -0
  39. lightning_sdk/lightning_cloud/openapi/models/v1_metrics_stream.py +27 -1
  40. lightning_sdk/lightning_cloud/openapi/models/v1_model_file.py +175 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py +27 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_event.py +331 -0
  43. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_event_type.py +104 -0
  44. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_fault_tolerance.py +149 -0
  45. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_fault_tolerance_strategy.py +105 -0
  46. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_status.py +27 -1
  47. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +1 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_snowflake_data_connection.py +29 -81
  49. lightning_sdk/lightning_cloud/openapi/models/v1_system_metrics.py +29 -3
  50. lightning_sdk/lightning_cloud/openapi/models/v1_trainium_system_metrics.py +175 -0
  51. lightning_sdk/lightning_cloud/openapi/models/v1_update_metrics_stream_visibility_response.py +97 -0
  52. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +27 -53
  53. lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_request.py +149 -0
  54. lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_response.py +97 -0
  55. lightning_sdk/lightning_cloud/rest_client.py +2 -0
  56. lightning_sdk/mmt/__init__.py +4 -0
  57. lightning_sdk/mmt/base.py +278 -0
  58. lightning_sdk/mmt/mmt.py +267 -0
  59. lightning_sdk/mmt/v1.py +181 -0
  60. lightning_sdk/mmt/v2.py +188 -0
  61. lightning_sdk/plugin.py +43 -16
  62. lightning_sdk/services/file_endpoint.py +11 -5
  63. lightning_sdk/studio.py +16 -9
  64. lightning_sdk/teamspace.py +21 -8
  65. lightning_sdk/utils/resolve.py +18 -0
  66. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.43.dist-info}/METADATA +4 -1
  67. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.43.dist-info}/RECORD +71 -59
  68. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.43.dist-info}/entry_points.txt +0 -1
  69. lightning_sdk/_mmt/__init__.py +0 -3
  70. lightning_sdk/_mmt/base.py +0 -180
  71. lightning_sdk/_mmt/mmt.py +0 -161
  72. lightning_sdk/_mmt/v1.py +0 -69
  73. lightning_sdk/_mmt/v2.py +0 -141
  74. lightning_sdk/cli/mmt.py +0 -137
  75. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.43.dist-info}/LICENSE +0 -0
  76. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.43.dist-info}/WHEEL +0 -0
  77. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.43.dist-info}/top_level.txt +0 -0
@@ -943,6 +943,115 @@ class ModelsStoreApi(object):
943
943
  _request_timeout=params.get('_request_timeout'),
944
944
  collection_formats=collection_formats)
945
945
 
946
+ def models_store_get_model_by_name(self, project_owner_name: 'str', project_name: 'str', model_name: 'str', **kwargs) -> 'V1Model': # noqa: E501
947
+ """models_store_get_model_by_name # noqa: E501
948
+
949
+ This method makes a synchronous HTTP request by default. To make an
950
+ asynchronous HTTP request, please pass async_req=True
951
+ >>> thread = api.models_store_get_model_by_name(project_owner_name, project_name, model_name, async_req=True)
952
+ >>> result = thread.get()
953
+
954
+ :param async_req bool
955
+ :param str project_owner_name: (required)
956
+ :param str project_name: (required)
957
+ :param str model_name: (required)
958
+ :return: V1Model
959
+ If the method is called asynchronously,
960
+ returns the request thread.
961
+ """
962
+ kwargs['_return_http_data_only'] = True
963
+ if kwargs.get('async_req'):
964
+ return self.models_store_get_model_by_name_with_http_info(project_owner_name, project_name, model_name, **kwargs) # noqa: E501
965
+ else:
966
+ (data) = self.models_store_get_model_by_name_with_http_info(project_owner_name, project_name, model_name, **kwargs) # noqa: E501
967
+ return data
968
+
969
+ def models_store_get_model_by_name_with_http_info(self, project_owner_name: 'str', project_name: 'str', model_name: 'str', **kwargs) -> 'V1Model': # noqa: E501
970
+ """models_store_get_model_by_name # noqa: E501
971
+
972
+ This method makes a synchronous HTTP request by default. To make an
973
+ asynchronous HTTP request, please pass async_req=True
974
+ >>> thread = api.models_store_get_model_by_name_with_http_info(project_owner_name, project_name, model_name, async_req=True)
975
+ >>> result = thread.get()
976
+
977
+ :param async_req bool
978
+ :param str project_owner_name: (required)
979
+ :param str project_name: (required)
980
+ :param str model_name: (required)
981
+ :return: V1Model
982
+ If the method is called asynchronously,
983
+ returns the request thread.
984
+ """
985
+
986
+ all_params = ['project_owner_name', 'project_name', 'model_name'] # noqa: E501
987
+ all_params.append('async_req')
988
+ all_params.append('_return_http_data_only')
989
+ all_params.append('_preload_content')
990
+ all_params.append('_request_timeout')
991
+
992
+ params = locals()
993
+ for key, val in six.iteritems(params['kwargs']):
994
+ if key not in all_params:
995
+ raise TypeError(
996
+ "Got an unexpected keyword argument '%s'"
997
+ " to method models_store_get_model_by_name" % key
998
+ )
999
+ params[key] = val
1000
+ del params['kwargs']
1001
+ # verify the required parameter 'project_owner_name' is set
1002
+ if ('project_owner_name' not in params or
1003
+ params['project_owner_name'] is None):
1004
+ raise ValueError("Missing the required parameter `project_owner_name` when calling `models_store_get_model_by_name`") # noqa: E501
1005
+ # verify the required parameter 'project_name' is set
1006
+ if ('project_name' not in params or
1007
+ params['project_name'] is None):
1008
+ raise ValueError("Missing the required parameter `project_name` when calling `models_store_get_model_by_name`") # noqa: E501
1009
+ # verify the required parameter 'model_name' is set
1010
+ if ('model_name' not in params or
1011
+ params['model_name'] is None):
1012
+ raise ValueError("Missing the required parameter `model_name` when calling `models_store_get_model_by_name`") # noqa: E501
1013
+
1014
+ collection_formats = {}
1015
+
1016
+ path_params = {}
1017
+ if 'project_owner_name' in params:
1018
+ path_params['projectOwnerName'] = params['project_owner_name'] # noqa: E501
1019
+ if 'project_name' in params:
1020
+ path_params['projectName'] = params['project_name'] # noqa: E501
1021
+ if 'model_name' in params:
1022
+ path_params['modelName'] = params['model_name'] # noqa: E501
1023
+
1024
+ query_params = []
1025
+
1026
+ header_params = {}
1027
+
1028
+ form_params = []
1029
+ local_var_files = {}
1030
+
1031
+ body_params = None
1032
+ # HTTP header `Accept`
1033
+ header_params['Accept'] = self.api_client.select_header_accept(
1034
+ ['application/json']) # noqa: E501
1035
+
1036
+ # Authentication setting
1037
+ auth_settings = [] # noqa: E501
1038
+
1039
+ return self.api_client.call_api(
1040
+ '/v1/projects/{projectOwnerName}/{projectName}/models/{modelName}', 'GET',
1041
+ path_params,
1042
+ query_params,
1043
+ header_params,
1044
+ body=body_params,
1045
+ post_params=form_params,
1046
+ files=local_var_files,
1047
+ response_type='V1Model', # noqa: E501
1048
+ auth_settings=auth_settings,
1049
+ async_req=params.get('async_req'),
1050
+ _return_http_data_only=params.get('_return_http_data_only'),
1051
+ _preload_content=params.get('_preload_content', True),
1052
+ _request_timeout=params.get('_request_timeout'),
1053
+ collection_formats=collection_formats)
1054
+
946
1055
  def models_store_get_model_file_upload_urls(self, body: 'UploadIdPartsBody', project_id: 'str', model_id: 'str', version: 'str', upload_id: 'str', **kwargs) -> 'V1GetModelFileUploadUrlsResponse': # noqa: E501
947
1056
  """GetModelFileUploadUrls requests pre-signed URLs for a given number of file parts. # noqa: E501
948
1057
 
@@ -1294,6 +1403,123 @@ class ModelsStoreApi(object):
1294
1403
  _request_timeout=params.get('_request_timeout'),
1295
1404
  collection_formats=collection_formats)
1296
1405
 
1406
+ def models_store_get_model_files_url(self, project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1GetModelFilesUrlResponse': # noqa: E501
1407
+ """GetModelFileUrl responds with a pre-signed url for a given file # noqa: E501
1408
+
1409
+ This method makes a synchronous HTTP request by default. To make an
1410
+ asynchronous HTTP request, please pass async_req=True
1411
+ >>> thread = api.models_store_get_model_files_url(project_id, model_id, version, async_req=True)
1412
+ >>> result = thread.get()
1413
+
1414
+ :param async_req bool
1415
+ :param str project_id: (required)
1416
+ :param str model_id: (required)
1417
+ :param str version: (required)
1418
+ :param str page_size:
1419
+ :param str offset:
1420
+ :return: V1GetModelFilesUrlResponse
1421
+ If the method is called asynchronously,
1422
+ returns the request thread.
1423
+ """
1424
+ kwargs['_return_http_data_only'] = True
1425
+ if kwargs.get('async_req'):
1426
+ return self.models_store_get_model_files_url_with_http_info(project_id, model_id, version, **kwargs) # noqa: E501
1427
+ else:
1428
+ (data) = self.models_store_get_model_files_url_with_http_info(project_id, model_id, version, **kwargs) # noqa: E501
1429
+ return data
1430
+
1431
+ def models_store_get_model_files_url_with_http_info(self, project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1GetModelFilesUrlResponse': # noqa: E501
1432
+ """GetModelFileUrl responds with a pre-signed url for a given file # noqa: E501
1433
+
1434
+ This method makes a synchronous HTTP request by default. To make an
1435
+ asynchronous HTTP request, please pass async_req=True
1436
+ >>> thread = api.models_store_get_model_files_url_with_http_info(project_id, model_id, version, async_req=True)
1437
+ >>> result = thread.get()
1438
+
1439
+ :param async_req bool
1440
+ :param str project_id: (required)
1441
+ :param str model_id: (required)
1442
+ :param str version: (required)
1443
+ :param str page_size:
1444
+ :param str offset:
1445
+ :return: V1GetModelFilesUrlResponse
1446
+ If the method is called asynchronously,
1447
+ returns the request thread.
1448
+ """
1449
+
1450
+ all_params = ['project_id', 'model_id', 'version', 'page_size', 'offset'] # noqa: E501
1451
+ all_params.append('async_req')
1452
+ all_params.append('_return_http_data_only')
1453
+ all_params.append('_preload_content')
1454
+ all_params.append('_request_timeout')
1455
+
1456
+ params = locals()
1457
+ for key, val in six.iteritems(params['kwargs']):
1458
+ if key not in all_params:
1459
+ raise TypeError(
1460
+ "Got an unexpected keyword argument '%s'"
1461
+ " to method models_store_get_model_files_url" % key
1462
+ )
1463
+ params[key] = val
1464
+ del params['kwargs']
1465
+ # verify the required parameter 'project_id' is set
1466
+ if ('project_id' not in params or
1467
+ params['project_id'] is None):
1468
+ raise ValueError("Missing the required parameter `project_id` when calling `models_store_get_model_files_url`") # noqa: E501
1469
+ # verify the required parameter 'model_id' is set
1470
+ if ('model_id' not in params or
1471
+ params['model_id'] is None):
1472
+ raise ValueError("Missing the required parameter `model_id` when calling `models_store_get_model_files_url`") # noqa: E501
1473
+ # verify the required parameter 'version' is set
1474
+ if ('version' not in params or
1475
+ params['version'] is None):
1476
+ raise ValueError("Missing the required parameter `version` when calling `models_store_get_model_files_url`") # noqa: E501
1477
+
1478
+ collection_formats = {}
1479
+
1480
+ path_params = {}
1481
+ if 'project_id' in params:
1482
+ path_params['projectId'] = params['project_id'] # noqa: E501
1483
+ if 'model_id' in params:
1484
+ path_params['modelId'] = params['model_id'] # noqa: E501
1485
+ if 'version' in params:
1486
+ path_params['version'] = params['version'] # noqa: E501
1487
+
1488
+ query_params = []
1489
+ if 'page_size' in params:
1490
+ query_params.append(('pageSize', params['page_size'])) # noqa: E501
1491
+ if 'offset' in params:
1492
+ query_params.append(('offset', params['offset'])) # noqa: E501
1493
+
1494
+ header_params = {}
1495
+
1496
+ form_params = []
1497
+ local_var_files = {}
1498
+
1499
+ body_params = None
1500
+ # HTTP header `Accept`
1501
+ header_params['Accept'] = self.api_client.select_header_accept(
1502
+ ['application/json']) # noqa: E501
1503
+
1504
+ # Authentication setting
1505
+ auth_settings = [] # noqa: E501
1506
+
1507
+ return self.api_client.call_api(
1508
+ '/v1/projects/{projectId}/models/{modelId}/versions/{version}/files', 'GET',
1509
+ path_params,
1510
+ query_params,
1511
+ header_params,
1512
+ body=body_params,
1513
+ post_params=form_params,
1514
+ files=local_var_files,
1515
+ response_type='V1GetModelFilesUrlResponse', # noqa: E501
1516
+ auth_settings=auth_settings,
1517
+ async_req=params.get('async_req'),
1518
+ _return_http_data_only=params.get('_return_http_data_only'),
1519
+ _preload_content=params.get('_preload_content', True),
1520
+ _request_timeout=params.get('_request_timeout'),
1521
+ collection_formats=collection_formats)
1522
+
1297
1523
  def models_store_get_model_version(self, project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1ModelVersionArchive': # noqa: E501
1298
1524
  """GetModelVersion used to get specific model version details # noqa: E501
1299
1525
 
@@ -53,6 +53,11 @@ class SnowflakeServiceApi(object):
53
53
 
54
54
  :param async_req bool
55
55
  :param str project_id: (required)
56
+ :param str username:
57
+ :param str password:
58
+ :param str account:
59
+ :param str private_key_file:
60
+ :param str private_key_password:
56
61
  :return: V1CheckSnowflakeConnectionResponse
57
62
  If the method is called asynchronously,
58
63
  returns the request thread.
@@ -74,12 +79,17 @@ class SnowflakeServiceApi(object):
74
79
 
75
80
  :param async_req bool
76
81
  :param str project_id: (required)
82
+ :param str username:
83
+ :param str password:
84
+ :param str account:
85
+ :param str private_key_file:
86
+ :param str private_key_password:
77
87
  :return: V1CheckSnowflakeConnectionResponse
78
88
  If the method is called asynchronously,
79
89
  returns the request thread.
80
90
  """
81
91
 
82
- all_params = ['project_id'] # noqa: E501
92
+ all_params = ['project_id', 'username', 'password', 'account', 'private_key_file', 'private_key_password'] # noqa: E501
83
93
  all_params.append('async_req')
84
94
  all_params.append('_return_http_data_only')
85
95
  all_params.append('_preload_content')
@@ -106,6 +116,16 @@ class SnowflakeServiceApi(object):
106
116
  path_params['projectId'] = params['project_id'] # noqa: E501
107
117
 
108
118
  query_params = []
119
+ if 'username' in params:
120
+ query_params.append(('username', params['username'])) # noqa: E501
121
+ if 'password' in params:
122
+ query_params.append(('password', params['password'])) # noqa: E501
123
+ if 'account' in params:
124
+ query_params.append(('account', params['account'])) # noqa: E501
125
+ if 'private_key_file' in params:
126
+ query_params.append(('privateKeyFile', params['private_key_file'])) # noqa: E501
127
+ if 'private_key_password' in params:
128
+ query_params.append(('privateKeyPassword', params['private_key_password'])) # noqa: E501
109
129
 
110
130
  header_params = {}
111
131
 
@@ -89,6 +89,7 @@ from lightning_sdk.lightning_cloud.openapi.models.id_start_body import IdStartBo
89
89
  from lightning_sdk.lightning_cloud.openapi.models.id_storage_body import IdStorageBody
90
90
  from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body import IdUploadsBody
91
91
  from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body1 import IdUploadsBody1
92
+ from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body import IdVisibilityBody
92
93
  from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body import JobsIdBody
93
94
  from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body1 import JobsIdBody1
94
95
  from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body2 import JobsIdBody2
@@ -409,6 +410,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_long_running_command_in
409
410
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_file_upload_urls_response import V1GetModelFileUploadUrlsResponse
410
411
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_file_url_response import V1GetModelFileUrlResponse
411
412
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_files_response import V1GetModelFilesResponse
413
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_files_url_response import V1GetModelFilesUrlResponse
412
414
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_artifact_response import V1GetProjectArtifactResponse
413
415
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_balance_response import V1GetProjectBalanceResponse
414
416
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_compute_usage_response import V1GetProjectComputeUsageResponse
@@ -522,6 +524,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_memberships_response i
522
524
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_metrics_streams_response import V1ListMetricsStreamsResponse
523
525
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_model_versions_response import V1ListModelVersionsResponse
524
526
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_models_response import V1ListModelsResponse
527
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_multi_machine_job_events_response import V1ListMultiMachineJobEventsResponse
525
528
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_multi_machine_jobs_response import V1ListMultiMachineJobsResponse
526
529
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_new_features_for_user_response import V1ListNewFeaturesForUserResponse
527
530
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_members_response import V1ListOrgMembersResponse
@@ -576,9 +579,14 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_stream import V1Met
576
579
  from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_tags import V1MetricsTags
577
580
  from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_tracker import V1MetricsTracker
578
581
  from lightning_sdk.lightning_cloud.openapi.models.v1_model import V1Model
582
+ from lightning_sdk.lightning_cloud.openapi.models.v1_model_file import V1ModelFile
579
583
  from lightning_sdk.lightning_cloud.openapi.models.v1_model_version_archive import V1ModelVersionArchive
580
584
  from lightning_sdk.lightning_cloud.openapi.models.v1_mount_target import V1MountTarget
581
585
  from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job import V1MultiMachineJob
586
+ from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_event import V1MultiMachineJobEvent
587
+ from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_event_type import V1MultiMachineJobEventType
588
+ from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_fault_tolerance import V1MultiMachineJobFaultTolerance
589
+ from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_fault_tolerance_strategy import V1MultiMachineJobFaultToleranceStrategy
582
590
  from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_state import V1MultiMachineJobState
583
591
  from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_status import V1MultiMachineJobStatus
584
592
  from lightning_sdk.lightning_cloud.openapi.models.v1_named_get_logger_metrics import V1NamedGetLoggerMetrics
@@ -688,6 +696,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics import V1Sys
688
696
  from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_list import V1SystemMetricsList
689
697
  from lightning_sdk.lightning_cloud.openapi.models.v1_telemetry import V1Telemetry
690
698
  from lightning_sdk.lightning_cloud.openapi.models.v1_timestamp_code_telemetry import V1TimestampCodeTelemetry
699
+ from lightning_sdk.lightning_cloud.openapi.models.v1_trainium_system_metrics import V1TrainiumSystemMetrics
691
700
  from lightning_sdk.lightning_cloud.openapi.models.v1_transaction import V1Transaction
692
701
  from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_org_balance_response import V1TransferOrgBalanceResponse
693
702
  from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_project_balance_response import V1TransferProjectBalanceResponse
@@ -704,6 +713,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators
704
713
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_availability_request import V1UpdateClusterAvailabilityRequest
705
714
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_index_response import V1UpdateIndexResponse
706
715
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_page_response import V1UpdateLitPageResponse
716
+ from lightning_sdk.lightning_cloud.openapi.models.v1_update_metrics_stream_visibility_response import V1UpdateMetricsStreamVisibilityResponse
707
717
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_model_visibility_response import V1UpdateModelVisibilityResponse
708
718
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_project_cluster_accelerators_response import V1UpdateProjectClusterAcceleratorsResponse
709
719
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_shared_metrics_stream_response import V1UpdateSharedMetricsStreamResponse
@@ -733,6 +743,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_user_slurm_job_action_respo
733
743
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_assistant_status_response import V1ValidateAssistantStatusResponse
734
744
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_auto_join_domain_response import V1ValidateAutoJoinDomainResponse
735
745
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_data_connection_response import V1ValidateDataConnectionResponse
746
+ from lightning_sdk.lightning_cloud.openapi.models.v1_validate_deployment_image_request import V1ValidateDeploymentImageRequest
747
+ from lightning_sdk.lightning_cloud.openapi.models.v1_validate_deployment_image_response import V1ValidateDeploymentImageResponse
736
748
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_endpoint_request import V1ValidateManagedEndpointRequest
737
749
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_endpoint_response import V1ValidateManagedEndpointResponse
738
750
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_model_response import V1ValidateManagedModelResponse
@@ -43,6 +43,7 @@ class DeploymenttemplatesIdBody(object):
43
43
  swagger_types = {
44
44
  'about_page_content': 'str',
45
45
  'categories': 'list[str]',
46
+ 'cloud_space_id': 'str',
46
47
  'description': 'str',
47
48
  'featured': 'bool',
48
49
  'image_url': 'str',
@@ -61,6 +62,7 @@ class DeploymenttemplatesIdBody(object):
61
62
  attribute_map = {
62
63
  'about_page_content': 'aboutPageContent',
63
64
  'categories': 'categories',
65
+ 'cloud_space_id': 'cloudSpaceId',
64
66
  'description': 'description',
65
67
  'featured': 'featured',
66
68
  'image_url': 'imageUrl',
@@ -76,10 +78,11 @@ class DeploymenttemplatesIdBody(object):
76
78
  'visibility': 'visibility'
77
79
  }
78
80
 
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
81
+ def __init__(self, about_page_content: 'str' =None, categories: 'list[str]' =None, cloud_space_id: '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
80
82
  """DeploymenttemplatesIdBody - a model defined in Swagger""" # noqa: E501
81
83
  self._about_page_content = None
82
84
  self._categories = None
85
+ self._cloud_space_id = None
83
86
  self._description = None
84
87
  self._featured = None
85
88
  self._image_url = None
@@ -98,6 +101,8 @@ class DeploymenttemplatesIdBody(object):
98
101
  self.about_page_content = about_page_content
99
102
  if categories is not None:
100
103
  self.categories = categories
104
+ if cloud_space_id is not None:
105
+ self.cloud_space_id = cloud_space_id
101
106
  if description is not None:
102
107
  self.description = description
103
108
  if featured is not None:
@@ -167,6 +172,27 @@ class DeploymenttemplatesIdBody(object):
167
172
 
168
173
  self._categories = categories
169
174
 
175
+ @property
176
+ def cloud_space_id(self) -> 'str':
177
+ """Gets the cloud_space_id of this DeploymenttemplatesIdBody. # noqa: E501
178
+
179
+
180
+ :return: The cloud_space_id of this DeploymenttemplatesIdBody. # noqa: E501
181
+ :rtype: str
182
+ """
183
+ return self._cloud_space_id
184
+
185
+ @cloud_space_id.setter
186
+ def cloud_space_id(self, cloud_space_id: 'str'):
187
+ """Sets the cloud_space_id of this DeploymenttemplatesIdBody.
188
+
189
+
190
+ :param cloud_space_id: The cloud_space_id of this DeploymenttemplatesIdBody. # noqa: E501
191
+ :type: str
192
+ """
193
+
194
+ self._cloud_space_id = cloud_space_id
195
+
170
196
  @property
171
197
  def description(self) -> 'str':
172
198
  """Gets the description of this DeploymenttemplatesIdBody. # noqa: E501
@@ -0,0 +1,123 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ external/v1/auth_service.proto
5
+
6
+ No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen) # noqa: E501
7
+
8
+ OpenAPI spec version: version not set
9
+
10
+ Generated by: https://github.com/swagger-api/swagger-codegen.git
11
+
12
+ NOTE
13
+ ----
14
+ standard swagger-codegen-cli for this python client has been modified
15
+ by custom templates. The purpose of these templates is to include
16
+ typing information in the API and Model code. Please refer to the
17
+ main grid repository for more info
18
+ """
19
+
20
+ import pprint
21
+ import re # noqa: F401
22
+
23
+ from typing import TYPE_CHECKING
24
+
25
+ import six
26
+
27
+ if TYPE_CHECKING:
28
+ from datetime import datetime
29
+ from lightning_sdk.lightning_cloud.openapi.models import *
30
+
31
+ class IdVisibilityBody(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
+ 'visibility': 'V1ResourceVisibility'
45
+ }
46
+
47
+ attribute_map = {
48
+ 'visibility': 'visibility'
49
+ }
50
+
51
+ def __init__(self, visibility: 'V1ResourceVisibility' =None): # noqa: E501
52
+ """IdVisibilityBody - a model defined in Swagger""" # noqa: E501
53
+ self._visibility = None
54
+ self.discriminator = None
55
+ if visibility is not None:
56
+ self.visibility = visibility
57
+
58
+ @property
59
+ def visibility(self) -> 'V1ResourceVisibility':
60
+ """Gets the visibility of this IdVisibilityBody. # noqa: E501
61
+
62
+
63
+ :return: The visibility of this IdVisibilityBody. # noqa: E501
64
+ :rtype: V1ResourceVisibility
65
+ """
66
+ return self._visibility
67
+
68
+ @visibility.setter
69
+ def visibility(self, visibility: 'V1ResourceVisibility'):
70
+ """Sets the visibility of this IdVisibilityBody.
71
+
72
+
73
+ :param visibility: The visibility of this IdVisibilityBody. # noqa: E501
74
+ :type: V1ResourceVisibility
75
+ """
76
+
77
+ self._visibility = visibility
78
+
79
+ def to_dict(self) -> dict:
80
+ """Returns the model properties as a dict"""
81
+ result = {}
82
+
83
+ for attr, _ in six.iteritems(self.swagger_types):
84
+ value = getattr(self, attr)
85
+ if isinstance(value, list):
86
+ result[attr] = list(map(
87
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
88
+ value
89
+ ))
90
+ elif hasattr(value, "to_dict"):
91
+ result[attr] = value.to_dict()
92
+ elif isinstance(value, dict):
93
+ result[attr] = dict(map(
94
+ lambda item: (item[0], item[1].to_dict())
95
+ if hasattr(item[1], "to_dict") else item,
96
+ value.items()
97
+ ))
98
+ else:
99
+ result[attr] = value
100
+ if issubclass(IdVisibilityBody, dict):
101
+ for key, value in self.items():
102
+ result[key] = value
103
+
104
+ return result
105
+
106
+ def to_str(self) -> str:
107
+ """Returns the string representation of the model"""
108
+ return pprint.pformat(self.to_dict())
109
+
110
+ def __repr__(self) -> str:
111
+ """For `print` and `pprint`"""
112
+ return self.to_str()
113
+
114
+ def __eq__(self, other: 'IdVisibilityBody') -> bool:
115
+ """Returns true if both objects are equal"""
116
+ if not isinstance(other, IdVisibilityBody):
117
+ return False
118
+
119
+ return self.__dict__ == other.__dict__
120
+
121
+ def __ne__(self, other: 'IdVisibilityBody') -> bool:
122
+ """Returns true if both objects are not equal"""
123
+ return not self == other
@@ -41,19 +41,24 @@ class ModelIdVersionsBody(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
- 'cluster_id': 'str'
44
+ 'cluster_id': 'str',
45
+ 'version': 'str'
45
46
  }
46
47
 
47
48
  attribute_map = {
48
- 'cluster_id': 'clusterId'
49
+ 'cluster_id': 'clusterId',
50
+ 'version': 'version'
49
51
  }
50
52
 
51
- def __init__(self, cluster_id: 'str' =None): # noqa: E501
53
+ def __init__(self, cluster_id: 'str' =None, version: 'str' =None): # noqa: E501
52
54
  """ModelIdVersionsBody - a model defined in Swagger""" # noqa: E501
53
55
  self._cluster_id = None
56
+ self._version = None
54
57
  self.discriminator = None
55
58
  if cluster_id is not None:
56
59
  self.cluster_id = cluster_id
60
+ if version is not None:
61
+ self.version = version
57
62
 
58
63
  @property
59
64
  def cluster_id(self) -> 'str':
@@ -76,6 +81,27 @@ class ModelIdVersionsBody(object):
76
81
 
77
82
  self._cluster_id = cluster_id
78
83
 
84
+ @property
85
+ def version(self) -> 'str':
86
+ """Gets the version of this ModelIdVersionsBody. # noqa: E501
87
+
88
+
89
+ :return: The version of this ModelIdVersionsBody. # noqa: E501
90
+ :rtype: str
91
+ """
92
+ return self._version
93
+
94
+ @version.setter
95
+ def version(self, version: 'str'):
96
+ """Sets the version of this ModelIdVersionsBody.
97
+
98
+
99
+ :param version: The version of this ModelIdVersionsBody. # noqa: E501
100
+ :type: str
101
+ """
102
+
103
+ self._version = version
104
+
79
105
  def to_dict(self) -> dict:
80
106
  """Returns the model properties as a dict"""
81
107
  result = {}
@@ -42,6 +42,7 @@ class ProjectIdMultimachinejobsBody(object):
42
42
  """
43
43
  swagger_types = {
44
44
  'cluster_id': 'str',
45
+ 'fault_tolerance': 'V1MultiMachineJobFaultTolerance',
45
46
  'machines': 'int',
46
47
  'name': 'str',
47
48
  'spec': 'V1JobSpec'
@@ -49,20 +50,24 @@ class ProjectIdMultimachinejobsBody(object):
49
50
 
50
51
  attribute_map = {
51
52
  'cluster_id': 'clusterId',
53
+ 'fault_tolerance': 'faultTolerance',
52
54
  'machines': 'machines',
53
55
  'name': 'name',
54
56
  'spec': 'spec'
55
57
  }
56
58
 
57
- def __init__(self, cluster_id: 'str' =None, machines: 'int' =None, name: 'str' =None, spec: 'V1JobSpec' =None): # noqa: E501
59
+ def __init__(self, cluster_id: 'str' =None, fault_tolerance: 'V1MultiMachineJobFaultTolerance' =None, machines: 'int' =None, name: 'str' =None, spec: 'V1JobSpec' =None): # noqa: E501
58
60
  """ProjectIdMultimachinejobsBody - a model defined in Swagger""" # noqa: E501
59
61
  self._cluster_id = None
62
+ self._fault_tolerance = None
60
63
  self._machines = None
61
64
  self._name = None
62
65
  self._spec = None
63
66
  self.discriminator = None
64
67
  if cluster_id is not None:
65
68
  self.cluster_id = cluster_id
69
+ if fault_tolerance is not None:
70
+ self.fault_tolerance = fault_tolerance
66
71
  if machines is not None:
67
72
  self.machines = machines
68
73
  if name is not None:
@@ -91,6 +96,27 @@ class ProjectIdMultimachinejobsBody(object):
91
96
 
92
97
  self._cluster_id = cluster_id
93
98
 
99
+ @property
100
+ def fault_tolerance(self) -> 'V1MultiMachineJobFaultTolerance':
101
+ """Gets the fault_tolerance of this ProjectIdMultimachinejobsBody. # noqa: E501
102
+
103
+
104
+ :return: The fault_tolerance of this ProjectIdMultimachinejobsBody. # noqa: E501
105
+ :rtype: V1MultiMachineJobFaultTolerance
106
+ """
107
+ return self._fault_tolerance
108
+
109
+ @fault_tolerance.setter
110
+ def fault_tolerance(self, fault_tolerance: 'V1MultiMachineJobFaultTolerance'):
111
+ """Sets the fault_tolerance of this ProjectIdMultimachinejobsBody.
112
+
113
+
114
+ :param fault_tolerance: The fault_tolerance of this ProjectIdMultimachinejobsBody. # noqa: E501
115
+ :type: V1MultiMachineJobFaultTolerance
116
+ """
117
+
118
+ self._fault_tolerance = fault_tolerance
119
+
94
120
  @property
95
121
  def machines(self) -> 'int':
96
122
  """Gets the machines of this ProjectIdMultimachinejobsBody. # noqa: E501