lightning-sdk 0.2.4__py3-none-any.whl → 0.2.6__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.
- lightning_sdk/__init__.py +1 -1
- lightning_sdk/ai_hub.py +3 -1
- lightning_sdk/api/ai_hub_api.py +15 -0
- lightning_sdk/api/studio_api.py +0 -8
- lightning_sdk/cli/serve.py +29 -9
- lightning_sdk/deployment/deployment.py +32 -4
- lightning_sdk/lightning_cloud/openapi/__init__.py +57 -1
- lightning_sdk/lightning_cloud/openapi/api/__init__.py +4 -0
- lightning_sdk/lightning_cloud/openapi/api/cloud_space_environment_template_service_api.py +537 -0
- lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +115 -6
- lightning_sdk/lightning_cloud/openapi/api/lit_dataset_service_api.py +1973 -0
- lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py +20 -4
- lightning_sdk/lightning_cloud/openapi/api/models_store_api.py +20 -20
- lightning_sdk/lightning_cloud/openapi/api/pipeline_templates_service_api.py +339 -0
- lightning_sdk/lightning_cloud/openapi/api/pipelines_service_api.py +5 -1
- lightning_sdk/lightning_cloud/openapi/api/schedules_service_api.py +573 -0
- lightning_sdk/lightning_cloud/openapi/api/slurm_jobs_user_service_api.py +202 -0
- lightning_sdk/lightning_cloud/openapi/models/__init__.py +53 -1
- lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityblock_body.py +15 -15
- lightning_sdk/lightning_cloud/openapi/models/cluster_id_slurmusers_body.py +201 -0
- lightning_sdk/lightning_cloud/openapi/models/dataset_id_versions_body.py +149 -0
- lightning_sdk/lightning_cloud/openapi/models/dataset_id_visibility_body.py +149 -0
- lightning_sdk/lightning_cloud/openapi/models/environmenttemplates_id_body.py +253 -0
- lightning_sdk/lightning_cloud/openapi/models/externalv1_cloud_space_instance_status.py +95 -69
- lightning_sdk/lightning_cloud/openapi/models/externalv1_cluster.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/litdatasets_dataset_id_body.py +149 -0
- lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body.py +69 -17
- lightning_sdk/lightning_cloud/openapi/models/pipelinetemplates_id_body.py +331 -0
- lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/project_id_litdatasets_body.py +227 -0
- lightning_sdk/lightning_cloud/openapi/models/project_id_pipelines_body.py +17 -17
- lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py +201 -0
- lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py +383 -0
- lightning_sdk/lightning_cloud/openapi/models/server_id_alerts_body.py +175 -0
- lightning_sdk/lightning_cloud/openapi/models/slurm_jobs_body.py +15 -15
- lightning_sdk/lightning_cloud/openapi/models/upload_id_complete_body1.py +149 -0
- lightning_sdk/lightning_cloud/openapi/models/upload_id_parts_body1.py +149 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_agent_job.py +124 -20
- lightning_sdk/lightning_cloud/openapi/models/v1_assistant_model_status.py +2 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template.py +279 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +123 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +157 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_complete_lit_dataset_multi_part_upload_response.py +97 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_complete_lit_dataset_upload_response.py +97 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +175 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_create_lit_dataset_multi_part_upload_response.py +123 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_create_pipeline_template_request.py +383 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_create_project_request.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_create_server_alert_response.py +97 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/{v1_pipeline_schedule.py → v1_delete_cloud_space_environment_template_response.py} +32 -32
- lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_dataset_response.py +97 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_dataset_version_response.py +97 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_delete_schedule_response.py +175 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_size_response.py +79 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_get_lit_dataset_file_upload_urls_response.py +123 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_get_lit_dataset_files_url_response.py +149 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py +55 -3
- lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.py +79 -27
- lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_lambda_labs_direct_v1.py +1 -29
- lightning_sdk/lightning_cloud/openapi/models/v1_list_cloud_space_environment_templates_response.py +123 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_list_lit_dataset_versions_response.py +123 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_list_lit_datasets_response.py +123 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_list_lit_registry_repository_image_artifact_versions_response.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_list_schedules_response.py +123 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_list_slurm_cluster_users_response.py +123 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_lit_dataset.py +539 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_lit_dataset_file.py +175 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_lit_dataset_version_archive.py +435 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_project.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_lit_repository.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_request.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +107 -3
- lightning_sdk/lightning_cloud/openapi/models/v1_metadata.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +53 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py +69 -17
- lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter.py +435 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter_placement.py +149 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter_placement_type.py +106 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter_type.py +106 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_template.py +513 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_template_visibility_type.py +105 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_reservation_details.py +201 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +1 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py +435 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_schedule_resource_type.py +103 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_server_alert.py +175 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_server_alert_severity.py +103 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_server_alert_type.py +105 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_slurm_cluster_user.py +227 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_slurm_job.py +58 -6
- lightning_sdk/lightning_cloud/openapi/models/v1_slurm_node.py +31 -291
- lightning_sdk/lightning_cloud/openapi/models/v1_subnet_spec.py +149 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_update_lit_dataset_visibility_response.py +123 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +223 -171
- lightning_sdk/lightning_cloud/openapi/models/v1_validate_managed_endpoint_response.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_voltage_park_direct_v1.py +203 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_vultr_direct_v1.py +1 -27
- lightning_sdk/lightning_cloud/openapi/models/version_default_body.py +29 -29
- lightning_sdk/lightning_cloud/openapi/models/version_default_body1.py +149 -0
- lightning_sdk/lightning_cloud/openapi/models/version_uploads_body1.py +123 -0
- lightning_sdk/lightning_cloud/openapi/models/versions_version_body1.py +123 -0
- lightning_sdk/lightning_cloud/rest_client.py +44 -40
- lightning_sdk/pipeline/types.py +28 -2
- lightning_sdk/pipeline/utils.py +1 -1
- lightning_sdk/plugin.py +0 -6
- lightning_sdk/serve.py +5 -3
- lightning_sdk/utils/resolve.py +1 -0
- {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.6.dist-info}/METADATA +1 -1
- {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.6.dist-info}/RECORD +123 -67
- {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.6.dist-info}/WHEEL +1 -1
- {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.6.dist-info}/LICENSE +0 -0
- {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.6.dist-info}/entry_points.txt +0 -0
- {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.6.dist-info}/top_level.txt +0 -0
|
@@ -665,6 +665,111 @@ class ClusterServiceApi(object):
|
|
|
665
665
|
_request_timeout=params.get('_request_timeout'),
|
|
666
666
|
collection_formats=collection_formats)
|
|
667
667
|
|
|
668
|
+
def cluster_service_create_server_alert(self, body: 'ServerIdAlertsBody', server_id: 'str', **kwargs) -> 'V1CreateServerAlertResponse': # noqa: E501
|
|
669
|
+
"""cluster_service_create_server_alert # noqa: E501
|
|
670
|
+
|
|
671
|
+
This method makes a synchronous HTTP request by default. To make an
|
|
672
|
+
asynchronous HTTP request, please pass async_req=True
|
|
673
|
+
>>> thread = api.cluster_service_create_server_alert(body, server_id, async_req=True)
|
|
674
|
+
>>> result = thread.get()
|
|
675
|
+
|
|
676
|
+
:param async_req bool
|
|
677
|
+
:param ServerIdAlertsBody body: (required)
|
|
678
|
+
:param str server_id: (required)
|
|
679
|
+
:return: V1CreateServerAlertResponse
|
|
680
|
+
If the method is called asynchronously,
|
|
681
|
+
returns the request thread.
|
|
682
|
+
"""
|
|
683
|
+
kwargs['_return_http_data_only'] = True
|
|
684
|
+
if kwargs.get('async_req'):
|
|
685
|
+
return self.cluster_service_create_server_alert_with_http_info(body, server_id, **kwargs) # noqa: E501
|
|
686
|
+
else:
|
|
687
|
+
(data) = self.cluster_service_create_server_alert_with_http_info(body, server_id, **kwargs) # noqa: E501
|
|
688
|
+
return data
|
|
689
|
+
|
|
690
|
+
def cluster_service_create_server_alert_with_http_info(self, body: 'ServerIdAlertsBody', server_id: 'str', **kwargs) -> 'V1CreateServerAlertResponse': # noqa: E501
|
|
691
|
+
"""cluster_service_create_server_alert # noqa: E501
|
|
692
|
+
|
|
693
|
+
This method makes a synchronous HTTP request by default. To make an
|
|
694
|
+
asynchronous HTTP request, please pass async_req=True
|
|
695
|
+
>>> thread = api.cluster_service_create_server_alert_with_http_info(body, server_id, async_req=True)
|
|
696
|
+
>>> result = thread.get()
|
|
697
|
+
|
|
698
|
+
:param async_req bool
|
|
699
|
+
:param ServerIdAlertsBody body: (required)
|
|
700
|
+
:param str server_id: (required)
|
|
701
|
+
:return: V1CreateServerAlertResponse
|
|
702
|
+
If the method is called asynchronously,
|
|
703
|
+
returns the request thread.
|
|
704
|
+
"""
|
|
705
|
+
|
|
706
|
+
all_params = ['body', 'server_id'] # noqa: E501
|
|
707
|
+
all_params.append('async_req')
|
|
708
|
+
all_params.append('_return_http_data_only')
|
|
709
|
+
all_params.append('_preload_content')
|
|
710
|
+
all_params.append('_request_timeout')
|
|
711
|
+
|
|
712
|
+
params = locals()
|
|
713
|
+
for key, val in six.iteritems(params['kwargs']):
|
|
714
|
+
if key not in all_params:
|
|
715
|
+
raise TypeError(
|
|
716
|
+
"Got an unexpected keyword argument '%s'"
|
|
717
|
+
" to method cluster_service_create_server_alert" % key
|
|
718
|
+
)
|
|
719
|
+
params[key] = val
|
|
720
|
+
del params['kwargs']
|
|
721
|
+
# verify the required parameter 'body' is set
|
|
722
|
+
if ('body' not in params or
|
|
723
|
+
params['body'] is None):
|
|
724
|
+
raise ValueError("Missing the required parameter `body` when calling `cluster_service_create_server_alert`") # noqa: E501
|
|
725
|
+
# verify the required parameter 'server_id' is set
|
|
726
|
+
if ('server_id' not in params or
|
|
727
|
+
params['server_id'] is None):
|
|
728
|
+
raise ValueError("Missing the required parameter `server_id` when calling `cluster_service_create_server_alert`") # noqa: E501
|
|
729
|
+
|
|
730
|
+
collection_formats = {}
|
|
731
|
+
|
|
732
|
+
path_params = {}
|
|
733
|
+
if 'server_id' in params:
|
|
734
|
+
path_params['serverId'] = params['server_id'] # noqa: E501
|
|
735
|
+
|
|
736
|
+
query_params = []
|
|
737
|
+
|
|
738
|
+
header_params = {}
|
|
739
|
+
|
|
740
|
+
form_params = []
|
|
741
|
+
local_var_files = {}
|
|
742
|
+
|
|
743
|
+
body_params = None
|
|
744
|
+
if 'body' in params:
|
|
745
|
+
body_params = params['body']
|
|
746
|
+
# HTTP header `Accept`
|
|
747
|
+
header_params['Accept'] = self.api_client.select_header_accept(
|
|
748
|
+
['application/json']) # noqa: E501
|
|
749
|
+
|
|
750
|
+
# HTTP header `Content-Type`
|
|
751
|
+
header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
|
|
752
|
+
['application/json']) # noqa: E501
|
|
753
|
+
|
|
754
|
+
# Authentication setting
|
|
755
|
+
auth_settings = [] # noqa: E501
|
|
756
|
+
|
|
757
|
+
return self.api_client.call_api(
|
|
758
|
+
'/v1/core/servers/{serverId}/alerts', 'POST',
|
|
759
|
+
path_params,
|
|
760
|
+
query_params,
|
|
761
|
+
header_params,
|
|
762
|
+
body=body_params,
|
|
763
|
+
post_params=form_params,
|
|
764
|
+
files=local_var_files,
|
|
765
|
+
response_type='V1CreateServerAlertResponse', # noqa: E501
|
|
766
|
+
auth_settings=auth_settings,
|
|
767
|
+
async_req=params.get('async_req'),
|
|
768
|
+
_return_http_data_only=params.get('_return_http_data_only'),
|
|
769
|
+
_preload_content=params.get('_preload_content', True),
|
|
770
|
+
_request_timeout=params.get('_request_timeout'),
|
|
771
|
+
collection_formats=collection_formats)
|
|
772
|
+
|
|
668
773
|
def cluster_service_delete_cluster(self, id: 'str', **kwargs) -> 'V1DeleteClusterResponse': # noqa: E501
|
|
669
774
|
"""cluster_service_delete_cluster # noqa: E501
|
|
670
775
|
|
|
@@ -1319,7 +1424,7 @@ class ClusterServiceApi(object):
|
|
|
1319
1424
|
:param datetime start_date:
|
|
1320
1425
|
:param str timezone:
|
|
1321
1426
|
:param str org_id:
|
|
1322
|
-
:param int
|
|
1427
|
+
:param int capacity_block_duration_hours:
|
|
1323
1428
|
:return: V1FindCapacityBlockOfferingResponse
|
|
1324
1429
|
If the method is called asynchronously,
|
|
1325
1430
|
returns the request thread.
|
|
@@ -1347,13 +1452,13 @@ class ClusterServiceApi(object):
|
|
|
1347
1452
|
:param datetime start_date:
|
|
1348
1453
|
:param str timezone:
|
|
1349
1454
|
:param str org_id:
|
|
1350
|
-
:param int
|
|
1455
|
+
:param int capacity_block_duration_hours:
|
|
1351
1456
|
:return: V1FindCapacityBlockOfferingResponse
|
|
1352
1457
|
If the method is called asynchronously,
|
|
1353
1458
|
returns the request thread.
|
|
1354
1459
|
"""
|
|
1355
1460
|
|
|
1356
|
-
all_params = ['project_id', 'cluster_id', 'instance_type', 'instance_count', 'start_date', 'timezone', 'org_id', '
|
|
1461
|
+
all_params = ['project_id', 'cluster_id', 'instance_type', 'instance_count', 'start_date', 'timezone', 'org_id', 'capacity_block_duration_hours'] # noqa: E501
|
|
1357
1462
|
all_params.append('async_req')
|
|
1358
1463
|
all_params.append('_return_http_data_only')
|
|
1359
1464
|
all_params.append('_preload_content')
|
|
@@ -1396,8 +1501,8 @@ class ClusterServiceApi(object):
|
|
|
1396
1501
|
query_params.append(('timezone', params['timezone'])) # noqa: E501
|
|
1397
1502
|
if 'org_id' in params:
|
|
1398
1503
|
query_params.append(('orgId', params['org_id'])) # noqa: E501
|
|
1399
|
-
if '
|
|
1400
|
-
query_params.append(('
|
|
1504
|
+
if 'capacity_block_duration_hours' in params:
|
|
1505
|
+
query_params.append(('capacityBlockDurationHours', params['capacity_block_duration_hours'])) # noqa: E501
|
|
1401
1506
|
|
|
1402
1507
|
header_params = {}
|
|
1403
1508
|
|
|
@@ -2611,6 +2716,7 @@ class ClusterServiceApi(object):
|
|
|
2611
2716
|
:param async_req bool
|
|
2612
2717
|
:param bool include_pricing:
|
|
2613
2718
|
:param str cloud_provider:
|
|
2719
|
+
:param str project_id:
|
|
2614
2720
|
:return: V1ListDefaultClusterAcceleratorsResponse
|
|
2615
2721
|
If the method is called asynchronously,
|
|
2616
2722
|
returns the request thread.
|
|
@@ -2633,12 +2739,13 @@ class ClusterServiceApi(object):
|
|
|
2633
2739
|
:param async_req bool
|
|
2634
2740
|
:param bool include_pricing:
|
|
2635
2741
|
:param str cloud_provider:
|
|
2742
|
+
:param str project_id:
|
|
2636
2743
|
:return: V1ListDefaultClusterAcceleratorsResponse
|
|
2637
2744
|
If the method is called asynchronously,
|
|
2638
2745
|
returns the request thread.
|
|
2639
2746
|
"""
|
|
2640
2747
|
|
|
2641
|
-
all_params = ['include_pricing', 'cloud_provider'] # noqa: E501
|
|
2748
|
+
all_params = ['include_pricing', 'cloud_provider', 'project_id'] # noqa: E501
|
|
2642
2749
|
all_params.append('async_req')
|
|
2643
2750
|
all_params.append('_return_http_data_only')
|
|
2644
2751
|
all_params.append('_preload_content')
|
|
@@ -2663,6 +2770,8 @@ class ClusterServiceApi(object):
|
|
|
2663
2770
|
query_params.append(('includePricing', params['include_pricing'])) # noqa: E501
|
|
2664
2771
|
if 'cloud_provider' in params:
|
|
2665
2772
|
query_params.append(('cloudProvider', params['cloud_provider'])) # noqa: E501
|
|
2773
|
+
if 'project_id' in params:
|
|
2774
|
+
query_params.append(('projectId', params['project_id'])) # noqa: E501
|
|
2666
2775
|
|
|
2667
2776
|
header_params = {}
|
|
2668
2777
|
|