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.
Files changed (123) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/ai_hub.py +3 -1
  3. lightning_sdk/api/ai_hub_api.py +15 -0
  4. lightning_sdk/api/studio_api.py +0 -8
  5. lightning_sdk/cli/serve.py +29 -9
  6. lightning_sdk/deployment/deployment.py +32 -4
  7. lightning_sdk/lightning_cloud/openapi/__init__.py +57 -1
  8. lightning_sdk/lightning_cloud/openapi/api/__init__.py +4 -0
  9. lightning_sdk/lightning_cloud/openapi/api/cloud_space_environment_template_service_api.py +537 -0
  10. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +115 -6
  11. lightning_sdk/lightning_cloud/openapi/api/lit_dataset_service_api.py +1973 -0
  12. lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py +20 -4
  13. lightning_sdk/lightning_cloud/openapi/api/models_store_api.py +20 -20
  14. lightning_sdk/lightning_cloud/openapi/api/pipeline_templates_service_api.py +339 -0
  15. lightning_sdk/lightning_cloud/openapi/api/pipelines_service_api.py +5 -1
  16. lightning_sdk/lightning_cloud/openapi/api/schedules_service_api.py +573 -0
  17. lightning_sdk/lightning_cloud/openapi/api/slurm_jobs_user_service_api.py +202 -0
  18. lightning_sdk/lightning_cloud/openapi/models/__init__.py +53 -1
  19. lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityblock_body.py +15 -15
  20. lightning_sdk/lightning_cloud/openapi/models/cluster_id_slurmusers_body.py +201 -0
  21. lightning_sdk/lightning_cloud/openapi/models/dataset_id_versions_body.py +149 -0
  22. lightning_sdk/lightning_cloud/openapi/models/dataset_id_visibility_body.py +149 -0
  23. lightning_sdk/lightning_cloud/openapi/models/environmenttemplates_id_body.py +253 -0
  24. lightning_sdk/lightning_cloud/openapi/models/externalv1_cloud_space_instance_status.py +95 -69
  25. lightning_sdk/lightning_cloud/openapi/models/externalv1_cluster.py +27 -1
  26. lightning_sdk/lightning_cloud/openapi/models/litdatasets_dataset_id_body.py +149 -0
  27. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body.py +69 -17
  29. lightning_sdk/lightning_cloud/openapi/models/pipelinetemplates_id_body.py +331 -0
  30. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
  31. lightning_sdk/lightning_cloud/openapi/models/project_id_litdatasets_body.py +227 -0
  32. lightning_sdk/lightning_cloud/openapi/models/project_id_pipelines_body.py +17 -17
  33. lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py +201 -0
  34. lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py +383 -0
  35. lightning_sdk/lightning_cloud/openapi/models/server_id_alerts_body.py +175 -0
  36. lightning_sdk/lightning_cloud/openapi/models/slurm_jobs_body.py +15 -15
  37. lightning_sdk/lightning_cloud/openapi/models/upload_id_complete_body1.py +149 -0
  38. lightning_sdk/lightning_cloud/openapi/models/upload_id_parts_body1.py +149 -0
  39. lightning_sdk/lightning_cloud/openapi/models/v1_agent_job.py +124 -20
  40. lightning_sdk/lightning_cloud/openapi/models/v1_assistant_model_status.py +2 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -0
  42. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +27 -1
  43. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template.py +279 -0
  44. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +123 -0
  45. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py +27 -1
  46. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +157 -1
  47. lightning_sdk/lightning_cloud/openapi/models/v1_complete_lit_dataset_multi_part_upload_response.py +97 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_complete_lit_dataset_upload_response.py +97 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +175 -0
  50. lightning_sdk/lightning_cloud/openapi/models/v1_create_lit_dataset_multi_part_upload_response.py +123 -0
  51. lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py +27 -1
  52. lightning_sdk/lightning_cloud/openapi/models/v1_create_pipeline_template_request.py +383 -0
  53. lightning_sdk/lightning_cloud/openapi/models/v1_create_project_request.py +27 -1
  54. lightning_sdk/lightning_cloud/openapi/models/v1_create_server_alert_response.py +97 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py +27 -1
  56. lightning_sdk/lightning_cloud/openapi/models/{v1_pipeline_schedule.py → v1_delete_cloud_space_environment_template_response.py} +32 -32
  57. lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_dataset_response.py +97 -0
  58. lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_dataset_version_response.py +97 -0
  59. lightning_sdk/lightning_cloud/openapi/models/v1_delete_schedule_response.py +175 -0
  60. lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_size_response.py +79 -1
  61. lightning_sdk/lightning_cloud/openapi/models/v1_get_lit_dataset_file_upload_urls_response.py +123 -0
  62. lightning_sdk/lightning_cloud/openapi/models/v1_get_lit_dataset_files_url_response.py +149 -0
  63. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
  64. lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py +55 -3
  65. lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.py +79 -27
  66. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +27 -1
  67. lightning_sdk/lightning_cloud/openapi/models/v1_lambda_labs_direct_v1.py +1 -29
  68. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloud_space_environment_templates_response.py +123 -0
  69. lightning_sdk/lightning_cloud/openapi/models/v1_list_lit_dataset_versions_response.py +123 -0
  70. lightning_sdk/lightning_cloud/openapi/models/v1_list_lit_datasets_response.py +123 -0
  71. lightning_sdk/lightning_cloud/openapi/models/v1_list_lit_registry_repository_image_artifact_versions_response.py +27 -1
  72. lightning_sdk/lightning_cloud/openapi/models/v1_list_schedules_response.py +123 -0
  73. lightning_sdk/lightning_cloud/openapi/models/v1_list_slurm_cluster_users_response.py +123 -0
  74. lightning_sdk/lightning_cloud/openapi/models/v1_lit_dataset.py +539 -0
  75. lightning_sdk/lightning_cloud/openapi/models/v1_lit_dataset_file.py +175 -0
  76. lightning_sdk/lightning_cloud/openapi/models/v1_lit_dataset_version_archive.py +435 -0
  77. lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_project.py +27 -1
  78. lightning_sdk/lightning_cloud/openapi/models/v1_lit_repository.py +27 -1
  79. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_request.py +27 -1
  80. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +107 -3
  81. lightning_sdk/lightning_cloud/openapi/models/v1_metadata.py +27 -1
  82. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +53 -1
  83. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py +69 -17
  84. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter.py +435 -0
  85. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter_placement.py +149 -0
  86. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter_placement_type.py +106 -0
  87. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter_type.py +106 -0
  88. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_template.py +513 -0
  89. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_template_visibility_type.py +105 -0
  90. lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py +27 -1
  91. lightning_sdk/lightning_cloud/openapi/models/v1_reservation_details.py +201 -0
  92. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +1 -0
  93. lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py +435 -0
  94. lightning_sdk/lightning_cloud/openapi/models/v1_schedule_resource_type.py +103 -0
  95. lightning_sdk/lightning_cloud/openapi/models/v1_server_alert.py +175 -0
  96. lightning_sdk/lightning_cloud/openapi/models/v1_server_alert_severity.py +103 -0
  97. lightning_sdk/lightning_cloud/openapi/models/v1_server_alert_type.py +105 -0
  98. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_cluster_user.py +227 -0
  99. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_job.py +58 -6
  100. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_node.py +31 -291
  101. lightning_sdk/lightning_cloud/openapi/models/v1_subnet_spec.py +149 -0
  102. lightning_sdk/lightning_cloud/openapi/models/v1_update_lit_dataset_visibility_response.py +123 -0
  103. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +27 -1
  104. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +223 -171
  105. lightning_sdk/lightning_cloud/openapi/models/v1_validate_managed_endpoint_response.py +27 -1
  106. lightning_sdk/lightning_cloud/openapi/models/v1_voltage_park_direct_v1.py +203 -0
  107. lightning_sdk/lightning_cloud/openapi/models/v1_vultr_direct_v1.py +1 -27
  108. lightning_sdk/lightning_cloud/openapi/models/version_default_body.py +29 -29
  109. lightning_sdk/lightning_cloud/openapi/models/version_default_body1.py +149 -0
  110. lightning_sdk/lightning_cloud/openapi/models/version_uploads_body1.py +123 -0
  111. lightning_sdk/lightning_cloud/openapi/models/versions_version_body1.py +123 -0
  112. lightning_sdk/lightning_cloud/rest_client.py +44 -40
  113. lightning_sdk/pipeline/types.py +28 -2
  114. lightning_sdk/pipeline/utils.py +1 -1
  115. lightning_sdk/plugin.py +0 -6
  116. lightning_sdk/serve.py +5 -3
  117. lightning_sdk/utils/resolve.py +1 -0
  118. {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.6.dist-info}/METADATA +1 -1
  119. {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.6.dist-info}/RECORD +123 -67
  120. {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.6.dist-info}/WHEEL +1 -1
  121. {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.6.dist-info}/LICENSE +0 -0
  122. {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.6.dist-info}/entry_points.txt +0 -0
  123. {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.6.dist-info}/top_level.txt +0 -0
@@ -43,6 +43,111 @@ class SlurmJobsUserServiceApi(object):
43
43
  api_client = ApiClient()
44
44
  self.api_client = api_client
45
45
 
46
+ def slurm_jobs_user_service_create_slurm_cluster_user(self, body: 'ClusterIdSlurmusersBody', cluster_id: 'str', **kwargs) -> 'V1SlurmClusterUser': # noqa: E501
47
+ """CreateSLURMClusterUser is used to create a new user on a SLURM cluster. # noqa: E501
48
+
49
+ This method makes a synchronous HTTP request by default. To make an
50
+ asynchronous HTTP request, please pass async_req=True
51
+ >>> thread = api.slurm_jobs_user_service_create_slurm_cluster_user(body, cluster_id, async_req=True)
52
+ >>> result = thread.get()
53
+
54
+ :param async_req bool
55
+ :param ClusterIdSlurmusersBody body: (required)
56
+ :param str cluster_id: (required)
57
+ :return: V1SlurmClusterUser
58
+ If the method is called asynchronously,
59
+ returns the request thread.
60
+ """
61
+ kwargs['_return_http_data_only'] = True
62
+ if kwargs.get('async_req'):
63
+ return self.slurm_jobs_user_service_create_slurm_cluster_user_with_http_info(body, cluster_id, **kwargs) # noqa: E501
64
+ else:
65
+ (data) = self.slurm_jobs_user_service_create_slurm_cluster_user_with_http_info(body, cluster_id, **kwargs) # noqa: E501
66
+ return data
67
+
68
+ def slurm_jobs_user_service_create_slurm_cluster_user_with_http_info(self, body: 'ClusterIdSlurmusersBody', cluster_id: 'str', **kwargs) -> 'V1SlurmClusterUser': # noqa: E501
69
+ """CreateSLURMClusterUser is used to create a new user on a SLURM cluster. # noqa: E501
70
+
71
+ This method makes a synchronous HTTP request by default. To make an
72
+ asynchronous HTTP request, please pass async_req=True
73
+ >>> thread = api.slurm_jobs_user_service_create_slurm_cluster_user_with_http_info(body, cluster_id, async_req=True)
74
+ >>> result = thread.get()
75
+
76
+ :param async_req bool
77
+ :param ClusterIdSlurmusersBody body: (required)
78
+ :param str cluster_id: (required)
79
+ :return: V1SlurmClusterUser
80
+ If the method is called asynchronously,
81
+ returns the request thread.
82
+ """
83
+
84
+ all_params = ['body', 'cluster_id'] # noqa: E501
85
+ all_params.append('async_req')
86
+ all_params.append('_return_http_data_only')
87
+ all_params.append('_preload_content')
88
+ all_params.append('_request_timeout')
89
+
90
+ params = locals()
91
+ for key, val in six.iteritems(params['kwargs']):
92
+ if key not in all_params:
93
+ raise TypeError(
94
+ "Got an unexpected keyword argument '%s'"
95
+ " to method slurm_jobs_user_service_create_slurm_cluster_user" % key
96
+ )
97
+ params[key] = val
98
+ del params['kwargs']
99
+ # verify the required parameter 'body' is set
100
+ if ('body' not in params or
101
+ params['body'] is None):
102
+ raise ValueError("Missing the required parameter `body` when calling `slurm_jobs_user_service_create_slurm_cluster_user`") # noqa: E501
103
+ # verify the required parameter 'cluster_id' is set
104
+ if ('cluster_id' not in params or
105
+ params['cluster_id'] is None):
106
+ raise ValueError("Missing the required parameter `cluster_id` when calling `slurm_jobs_user_service_create_slurm_cluster_user`") # noqa: E501
107
+
108
+ collection_formats = {}
109
+
110
+ path_params = {}
111
+ if 'cluster_id' in params:
112
+ path_params['clusterId'] = params['cluster_id'] # noqa: E501
113
+
114
+ query_params = []
115
+
116
+ header_params = {}
117
+
118
+ form_params = []
119
+ local_var_files = {}
120
+
121
+ body_params = None
122
+ if 'body' in params:
123
+ body_params = params['body']
124
+ # HTTP header `Accept`
125
+ header_params['Accept'] = self.api_client.select_header_accept(
126
+ ['application/json']) # noqa: E501
127
+
128
+ # HTTP header `Content-Type`
129
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
130
+ ['application/json']) # noqa: E501
131
+
132
+ # Authentication setting
133
+ auth_settings = [] # noqa: E501
134
+
135
+ return self.api_client.call_api(
136
+ '/v1/core/clusters/{clusterId}/slurm-users', 'POST',
137
+ path_params,
138
+ query_params,
139
+ header_params,
140
+ body=body_params,
141
+ post_params=form_params,
142
+ files=local_var_files,
143
+ response_type='V1SlurmClusterUser', # noqa: E501
144
+ auth_settings=auth_settings,
145
+ async_req=params.get('async_req'),
146
+ _return_http_data_only=params.get('_return_http_data_only'),
147
+ _preload_content=params.get('_preload_content', True),
148
+ _request_timeout=params.get('_request_timeout'),
149
+ collection_formats=collection_formats)
150
+
46
151
  def slurm_jobs_user_service_create_user_slurm_job(self, body: 'SlurmJobsBody', project_id: 'str', **kwargs) -> 'V1SLURMJob': # noqa: E501
47
152
  """CreateUserSLURMJob is used to create a new SLURM job for an authenticated user. For creating SLURM Jobs from an agent, please use the SLURMJobsAgentService. # noqa: E501
48
153
 
@@ -459,6 +564,103 @@ class SlurmJobsUserServiceApi(object):
459
564
  _request_timeout=params.get('_request_timeout'),
460
565
  collection_formats=collection_formats)
461
566
 
567
+ def slurm_jobs_user_service_list_slurm_cluster_users(self, cluster_id: 'str', **kwargs) -> 'V1ListSLURMClusterUsersResponse': # noqa: E501
568
+ """ListSLURMClusterUsers is used to list all users on a SLURM cluster. # noqa: E501
569
+
570
+ This method makes a synchronous HTTP request by default. To make an
571
+ asynchronous HTTP request, please pass async_req=True
572
+ >>> thread = api.slurm_jobs_user_service_list_slurm_cluster_users(cluster_id, async_req=True)
573
+ >>> result = thread.get()
574
+
575
+ :param async_req bool
576
+ :param str cluster_id: (required)
577
+ :param str organization_id:
578
+ :return: V1ListSLURMClusterUsersResponse
579
+ If the method is called asynchronously,
580
+ returns the request thread.
581
+ """
582
+ kwargs['_return_http_data_only'] = True
583
+ if kwargs.get('async_req'):
584
+ return self.slurm_jobs_user_service_list_slurm_cluster_users_with_http_info(cluster_id, **kwargs) # noqa: E501
585
+ else:
586
+ (data) = self.slurm_jobs_user_service_list_slurm_cluster_users_with_http_info(cluster_id, **kwargs) # noqa: E501
587
+ return data
588
+
589
+ def slurm_jobs_user_service_list_slurm_cluster_users_with_http_info(self, cluster_id: 'str', **kwargs) -> 'V1ListSLURMClusterUsersResponse': # noqa: E501
590
+ """ListSLURMClusterUsers is used to list all users on a SLURM cluster. # noqa: E501
591
+
592
+ This method makes a synchronous HTTP request by default. To make an
593
+ asynchronous HTTP request, please pass async_req=True
594
+ >>> thread = api.slurm_jobs_user_service_list_slurm_cluster_users_with_http_info(cluster_id, async_req=True)
595
+ >>> result = thread.get()
596
+
597
+ :param async_req bool
598
+ :param str cluster_id: (required)
599
+ :param str organization_id:
600
+ :return: V1ListSLURMClusterUsersResponse
601
+ If the method is called asynchronously,
602
+ returns the request thread.
603
+ """
604
+
605
+ all_params = ['cluster_id', 'organization_id'] # noqa: E501
606
+ all_params.append('async_req')
607
+ all_params.append('_return_http_data_only')
608
+ all_params.append('_preload_content')
609
+ all_params.append('_request_timeout')
610
+
611
+ params = locals()
612
+ for key, val in six.iteritems(params['kwargs']):
613
+ if key not in all_params:
614
+ raise TypeError(
615
+ "Got an unexpected keyword argument '%s'"
616
+ " to method slurm_jobs_user_service_list_slurm_cluster_users" % key
617
+ )
618
+ params[key] = val
619
+ del params['kwargs']
620
+ # verify the required parameter 'cluster_id' is set
621
+ if ('cluster_id' not in params or
622
+ params['cluster_id'] is None):
623
+ raise ValueError("Missing the required parameter `cluster_id` when calling `slurm_jobs_user_service_list_slurm_cluster_users`") # noqa: E501
624
+
625
+ collection_formats = {}
626
+
627
+ path_params = {}
628
+ if 'cluster_id' in params:
629
+ path_params['clusterId'] = params['cluster_id'] # noqa: E501
630
+
631
+ query_params = []
632
+ if 'organization_id' in params:
633
+ query_params.append(('organizationId', params['organization_id'])) # noqa: E501
634
+
635
+ header_params = {}
636
+
637
+ form_params = []
638
+ local_var_files = {}
639
+
640
+ body_params = None
641
+ # HTTP header `Accept`
642
+ header_params['Accept'] = self.api_client.select_header_accept(
643
+ ['application/json']) # noqa: E501
644
+
645
+ # Authentication setting
646
+ auth_settings = [] # noqa: E501
647
+
648
+ return self.api_client.call_api(
649
+ '/v1/core/clusters/{clusterId}/slurm-users', 'GET',
650
+ path_params,
651
+ query_params,
652
+ header_params,
653
+ body=body_params,
654
+ post_params=form_params,
655
+ files=local_var_files,
656
+ response_type='V1ListSLURMClusterUsersResponse', # noqa: E501
657
+ auth_settings=auth_settings,
658
+ async_req=params.get('async_req'),
659
+ _return_http_data_only=params.get('_return_http_data_only'),
660
+ _preload_content=params.get('_preload_content', True),
661
+ _request_timeout=params.get('_request_timeout'),
662
+ collection_formats=collection_formats)
663
+
462
664
  def slurm_jobs_user_service_list_user_slurm_jobs(self, project_id: 'str', **kwargs) -> 'V1ListUserSLURMJobsResponse': # noqa: E501
463
665
  """ListUserSLURMJobs is used to list all SLURM jobs for an authenticated user. For listing SLURM Jobs from an agent, please use the SLURMJobsAgentService. # noqa: E501
464
666
 
@@ -43,6 +43,7 @@ from lightning_sdk.lightning_cloud.openapi.models.cloudspaces_id_body import Clo
43
43
  from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityblock_body import ClusterIdCapacityblockBody
44
44
  from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityreservations_body import ClusterIdCapacityreservationsBody
45
45
  from lightning_sdk.lightning_cloud.openapi.models.cluster_id_proxies_body import ClusterIdProxiesBody
46
+ from lightning_sdk.lightning_cloud.openapi.models.cluster_id_slurmusers_body import ClusterIdSlurmusersBody
46
47
  from lightning_sdk.lightning_cloud.openapi.models.cluster_id_usagerestrictions_body import ClusterIdUsagerestrictionsBody
47
48
  from lightning_sdk.lightning_cloud.openapi.models.clusters_id_body import ClustersIdBody
48
49
  from lightning_sdk.lightning_cloud.openapi.models.clusters_id_body1 import ClustersIdBody1
@@ -52,10 +53,13 @@ from lightning_sdk.lightning_cloud.openapi.models.create import Create
52
53
  from lightning_sdk.lightning_cloud.openapi.models.create_checkout_session_request_wallet_type import CreateCheckoutSessionRequestWalletType
53
54
  from lightning_sdk.lightning_cloud.openapi.models.create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs import CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs
54
55
  from lightning_sdk.lightning_cloud.openapi.models.data_connection_mount_data_connection_mount_copy_status import DataConnectionMountDataConnectionMountCopyStatus
56
+ from lightning_sdk.lightning_cloud.openapi.models.dataset_id_versions_body import DatasetIdVersionsBody
57
+ from lightning_sdk.lightning_cloud.openapi.models.dataset_id_visibility_body import DatasetIdVisibilityBody
55
58
  from lightning_sdk.lightning_cloud.openapi.models.datasets_id_body import DatasetsIdBody
56
59
  from lightning_sdk.lightning_cloud.openapi.models.deployments_id_body import DeploymentsIdBody
57
60
  from lightning_sdk.lightning_cloud.openapi.models.deploymenttemplates_id_body import DeploymenttemplatesIdBody
58
61
  from lightning_sdk.lightning_cloud.openapi.models.endpoints_id_body import EndpointsIdBody
62
+ from lightning_sdk.lightning_cloud.openapi.models.environmenttemplates_id_body import EnvironmenttemplatesIdBody
59
63
  from lightning_sdk.lightning_cloud.openapi.models.experiment_name_variant_name_body import ExperimentNameVariantNameBody
60
64
  from lightning_sdk.lightning_cloud.openapi.models.externalv1_cloud_space_instance_status import Externalv1CloudSpaceInstanceStatus
61
65
  from lightning_sdk.lightning_cloud.openapi.models.externalv1_cluster import Externalv1Cluster
@@ -99,6 +103,7 @@ from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body import JobsIdBody
99
103
  from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body1 import JobsIdBody1
100
104
  from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body2 import JobsIdBody2
101
105
  from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body3 import JobsIdBody3
106
+ from lightning_sdk.lightning_cloud.openapi.models.litdatasets_dataset_id_body import LitdatasetsDatasetIdBody
102
107
  from lightning_sdk.lightning_cloud.openapi.models.litloggermetrics_id_body import LitloggermetricsIdBody
103
108
  from lightning_sdk.lightning_cloud.openapi.models.litpages_id_body import LitpagesIdBody
104
109
  from lightning_sdk.lightning_cloud.openapi.models.litregistry_lit_repo_name_body import LitregistryLitRepoNameBody
@@ -116,6 +121,7 @@ from lightning_sdk.lightning_cloud.openapi.models.org_id_memberships_body import
116
121
  from lightning_sdk.lightning_cloud.openapi.models.org_id_roles_body import OrgIdRolesBody
117
122
  from lightning_sdk.lightning_cloud.openapi.models.orgs_id_body import OrgsIdBody
118
123
  from lightning_sdk.lightning_cloud.openapi.models.pipelines_id_body import PipelinesIdBody
124
+ from lightning_sdk.lightning_cloud.openapi.models.pipelinetemplates_id_body import PipelinetemplatesIdBody
119
125
  from lightning_sdk.lightning_cloud.openapi.models.profiler_captures_body import ProfilerCapturesBody
120
126
  from lightning_sdk.lightning_cloud.openapi.models.profiler_enabled_body import ProfilerEnabledBody
121
127
  from lightning_sdk.lightning_cloud.openapi.models.project_id_agentmanagedendpoints_body import ProjectIdAgentmanagedendpointsBody
@@ -128,12 +134,14 @@ from lightning_sdk.lightning_cloud.openapi.models.project_id_fileendpoints_body
128
134
  from lightning_sdk.lightning_cloud.openapi.models.project_id_getapp_body import ProjectIdGetappBody
129
135
  from lightning_sdk.lightning_cloud.openapi.models.project_id_invite_body import ProjectIdInviteBody
130
136
  from lightning_sdk.lightning_cloud.openapi.models.project_id_jobs_body import ProjectIdJobsBody
137
+ from lightning_sdk.lightning_cloud.openapi.models.project_id_litdatasets_body import ProjectIdLitdatasetsBody
131
138
  from lightning_sdk.lightning_cloud.openapi.models.project_id_litregistry_body import ProjectIdLitregistryBody
132
139
  from lightning_sdk.lightning_cloud.openapi.models.project_id_memberships_body import ProjectIdMembershipsBody
133
140
  from lightning_sdk.lightning_cloud.openapi.models.project_id_models_body import ProjectIdModelsBody
134
141
  from lightning_sdk.lightning_cloud.openapi.models.project_id_multimachinejobs_body import ProjectIdMultimachinejobsBody
135
142
  from lightning_sdk.lightning_cloud.openapi.models.project_id_pipelines_body import ProjectIdPipelinesBody
136
143
  from lightning_sdk.lightning_cloud.openapi.models.project_id_projectclustersbindings_body import ProjectIdProjectclustersbindingsBody
144
+ from lightning_sdk.lightning_cloud.openapi.models.project_id_schedules_body import ProjectIdSchedulesBody
137
145
  from lightning_sdk.lightning_cloud.openapi.models.project_id_secrets_body import ProjectIdSecretsBody
138
146
  from lightning_sdk.lightning_cloud.openapi.models.project_id_serviceexecution_body import ProjectIdServiceexecutionBody
139
147
  from lightning_sdk.lightning_cloud.openapi.models.project_id_snowflake_body import ProjectIdSnowflakeBody
@@ -143,8 +151,10 @@ from lightning_sdk.lightning_cloud.openapi.models.projects_project_id_body impor
143
151
  from lightning_sdk.lightning_cloud.openapi.models.protobuf_any import ProtobufAny
144
152
  from lightning_sdk.lightning_cloud.openapi.models.query_query_id_body import QueryQueryIdBody
145
153
  from lightning_sdk.lightning_cloud.openapi.models.rpc_status import RpcStatus
154
+ from lightning_sdk.lightning_cloud.openapi.models.schedules_id_body import SchedulesIdBody
146
155
  from lightning_sdk.lightning_cloud.openapi.models.secrets_id_body import SecretsIdBody
147
156
  from lightning_sdk.lightning_cloud.openapi.models.secrets_id_body1 import SecretsIdBody1
157
+ from lightning_sdk.lightning_cloud.openapi.models.server_id_alerts_body import ServerIdAlertsBody
148
158
  from lightning_sdk.lightning_cloud.openapi.models.servers_server_id_body import ServersServerIdBody
149
159
  from lightning_sdk.lightning_cloud.openapi.models.service_artifact_artifact_kind import ServiceArtifactArtifactKind
150
160
  from lightning_sdk.lightning_cloud.openapi.models.service_health_service_status import ServiceHealthServiceStatus
@@ -160,7 +170,9 @@ from lightning_sdk.lightning_cloud.openapi.models.stream_result_of_v1_get_long_r
160
170
  from lightning_sdk.lightning_cloud.openapi.models.studioapp_jobs_body import StudioappJobsBody
161
171
  from lightning_sdk.lightning_cloud.openapi.models.update import Update
162
172
  from lightning_sdk.lightning_cloud.openapi.models.upload_id_complete_body import UploadIdCompleteBody
173
+ from lightning_sdk.lightning_cloud.openapi.models.upload_id_complete_body1 import UploadIdCompleteBody1
163
174
  from lightning_sdk.lightning_cloud.openapi.models.upload_id_parts_body import UploadIdPartsBody
175
+ from lightning_sdk.lightning_cloud.openapi.models.upload_id_parts_body1 import UploadIdPartsBody1
164
176
  from lightning_sdk.lightning_cloud.openapi.models.uploads_upload_id_body import UploadsUploadIdBody
165
177
  from lightning_sdk.lightning_cloud.openapi.models.uploads_upload_id_body1 import UploadsUploadIdBody1
166
178
  from lightning_sdk.lightning_cloud.openapi.models.usagerestrictions_id_body import UsagerestrictionsIdBody
@@ -219,6 +231,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_artifact_event_
219
231
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version import V1CloudSpaceCodeVersion
220
232
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version_status import V1CloudSpaceCodeVersionStatus
221
233
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_engagement_response import V1CloudSpaceEngagementResponse
234
+ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template import V1CloudSpaceEnvironmentTemplate
235
+ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template_config import V1CloudSpaceEnvironmentTemplateConfig
222
236
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_instance_collab_status import V1CloudSpaceInstanceCollabStatus
223
237
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_instance_config import V1CloudSpaceInstanceConfig
224
238
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_instance_startup_status import V1CloudSpaceInstanceStartupStatus
@@ -250,6 +264,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_collab_action import V1Coll
250
264
  from lightning_sdk.lightning_cloud.openapi.models.v1_collab_session import V1CollabSession
251
265
  from lightning_sdk.lightning_cloud.openapi.models.v1_command_argument import V1CommandArgument
252
266
  from lightning_sdk.lightning_cloud.openapi.models.v1_complete_agent_multipart_upload import V1CompleteAgentMultipartUpload
267
+ from lightning_sdk.lightning_cloud.openapi.models.v1_complete_lit_dataset_multi_part_upload_response import V1CompleteLitDatasetMultiPartUploadResponse
268
+ from lightning_sdk.lightning_cloud.openapi.models.v1_complete_lit_dataset_upload_response import V1CompleteLitDatasetUploadResponse
253
269
  from lightning_sdk.lightning_cloud.openapi.models.v1_complete_model_upload_response import V1CompleteModelUploadResponse
254
270
  from lightning_sdk.lightning_cloud.openapi.models.v1_complete_multi_part_upload_response import V1CompleteMultiPartUploadResponse
255
271
  from lightning_sdk.lightning_cloud.openapi.models.v1_complete_onboarding_response import V1CompleteOnboardingResponse
@@ -270,6 +286,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_billing_portal_sessi
270
286
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_checkout_session_request import V1CreateCheckoutSessionRequest
271
287
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_checkout_session_response import V1CreateCheckoutSessionResponse
272
288
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_cloud_space_app_instance_response import V1CreateCloudSpaceAppInstanceResponse
289
+ from lightning_sdk.lightning_cloud.openapi.models.v1_create_cloud_space_environment_template_request import V1CreateCloudSpaceEnvironmentTemplateRequest
273
290
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_cloud_space_instance_metric_response import V1CreateCloudSpaceInstanceMetricResponse
274
291
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_capacity_reservation_response import V1CreateClusterCapacityReservationResponse
275
292
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_encryption_keys_request import V1CreateClusterEncryptionKeysRequest
@@ -279,14 +296,17 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_response imp
279
296
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_deployment_request import V1CreateDeploymentRequest
280
297
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_deployment_template_request import V1CreateDeploymentTemplateRequest
281
298
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_job_request import V1CreateJobRequest
299
+ from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_dataset_multi_part_upload_response import V1CreateLitDatasetMultiPartUploadResponse
282
300
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_page_request import V1CreateLitPageRequest
283
301
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_page_response import V1CreateLitPageResponse
284
302
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_managed_endpoint_response import V1CreateManagedEndpointResponse
285
303
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_multi_machine_job_request import V1CreateMultiMachineJobRequest
286
304
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_multi_part_upload_response import V1CreateMultiPartUploadResponse
287
305
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_organization_request import V1CreateOrganizationRequest
306
+ from lightning_sdk.lightning_cloud.openapi.models.v1_create_pipeline_template_request import V1CreatePipelineTemplateRequest
288
307
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_project_request import V1CreateProjectRequest
289
308
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_ssh_public_key_request import V1CreateSSHPublicKeyRequest
309
+ from lightning_sdk.lightning_cloud.openapi.models.v1_create_server_alert_response import V1CreateServerAlertResponse
290
310
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_shared_metrics_stream_request import V1CreateSharedMetricsStreamRequest
291
311
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_shared_metrics_stream_response import V1CreateSharedMetricsStreamResponse
292
312
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_snowflake_connection_response import V1CreateSnowflakeConnectionResponse
@@ -304,6 +324,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_dataset_type import V1Datas
304
324
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_affiliate_link_response import V1DeleteAffiliateLinkResponse
305
325
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_assistant_response import V1DeleteAssistantResponse
306
326
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cloud_space_app_response import V1DeleteCloudSpaceAppResponse
327
+ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cloud_space_environment_template_response import V1DeleteCloudSpaceEnvironmentTemplateResponse
307
328
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cloud_space_response import V1DeleteCloudSpaceResponse
308
329
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cloud_space_session_response import V1DeleteCloudSpaceSessionResponse
309
330
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cloud_space_version_publication_response import V1DeleteCloudSpaceVersionPublicationResponse
@@ -326,6 +347,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightning_run_respon
326
347
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningapp_instance_artifact_response import V1DeleteLightningappInstanceArtifactResponse
327
348
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningapp_instance_response import V1DeleteLightningappInstanceResponse
328
349
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningwork_response import V1DeleteLightningworkResponse
350
+ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_dataset_response import V1DeleteLitDatasetResponse
351
+ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_dataset_version_response import V1DeleteLitDatasetVersionResponse
329
352
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_page_response import V1DeleteLitPageResponse
330
353
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_repository_response import V1DeleteLitRepositoryResponse
331
354
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_logger_artifact_response import V1DeleteLoggerArtifactResponse
@@ -348,6 +371,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_project_membership_r
348
371
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_project_response import V1DeleteProjectResponse
349
372
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_project_role_response import V1DeleteProjectRoleResponse
350
373
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_ssh_public_key_response import V1DeleteSSHPublicKeyResponse
374
+ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_schedule_response import V1DeleteScheduleResponse
351
375
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_secret_response import V1DeleteSecretResponse
352
376
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_service_execution_response import V1DeleteServiceExecutionResponse
353
377
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_shared_metrics_stream_response import V1DeleteSharedMetricsStreamResponse
@@ -435,6 +459,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_job_system_metrics_resp
435
459
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_lightning_run_source_code_download_url_response import V1GetLightningRunSourceCodeDownloadUrlResponse
436
460
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_lightningapp_instance_open_ports_response import V1GetLightningappInstanceOpenPortsResponse
437
461
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_lightningapp_instance_system_metrics_response import V1GetLightningappInstanceSystemMetricsResponse
462
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_lit_dataset_file_upload_urls_response import V1GetLitDatasetFileUploadUrlsResponse
463
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_lit_dataset_files_url_response import V1GetLitDatasetFilesUrlResponse
438
464
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_lit_page_response import V1GetLitPageResponse
439
465
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_logger_metrics_response import V1GetLoggerMetricsResponse
440
466
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_long_running_command_in_cloud_space_response import V1GetLongRunningCommandInCloudSpaceResponse
@@ -506,6 +532,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_agent_job_artifacts_re
506
532
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_agent_jobs_response import V1ListAgentJobsResponse
507
533
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_assistants_response import V1ListAssistantsResponse
508
534
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_apps_response import V1ListCloudSpaceAppsResponse
535
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_environment_templates_response import V1ListCloudSpaceEnvironmentTemplatesResponse
509
536
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_instances_response import V1ListCloudSpaceInstancesResponse
510
537
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_publications_response import V1ListCloudSpacePublicationsResponse
511
538
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_python_versions_response import V1ListCloudSpacePythonVersionsResponse
@@ -551,6 +578,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningapp_instance_
551
578
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningapp_instances_response import V1ListLightningappInstancesResponse
552
579
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningwork_events_response import V1ListLightningworkEventsResponse
553
580
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningwork_response import V1ListLightningworkResponse
581
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_dataset_versions_response import V1ListLitDatasetVersionsResponse
582
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_datasets_response import V1ListLitDatasetsResponse
554
583
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_pages_response import V1ListLitPagesResponse
555
584
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_registry_repository_image_artifact_versions_response import V1ListLitRegistryRepositoryImageArtifactVersionsResponse
556
585
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_logger_artifact_response import V1ListLoggerArtifactResponse
@@ -581,12 +610,17 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_project_roles_response
581
610
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_published_cloud_spaces_response import V1ListPublishedCloudSpacesResponse
582
611
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_published_deployment_templates_response import V1ListPublishedDeploymentTemplatesResponse
583
612
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_quests_response import V1ListQuestsResponse
613
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_slurm_cluster_users_response import V1ListSLURMClusterUsersResponse
584
614
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_ssh_public_keys_response import V1ListSSHPublicKeysResponse
615
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_schedules_response import V1ListSchedulesResponse
585
616
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_secrets_response import V1ListSecretsResponse
586
617
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_service_execution_lightningapp_instances_response import V1ListServiceExecutionLightningappInstancesResponse
587
618
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_service_executions_response import V1ListServiceExecutionsResponse
588
619
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_studio_jobs_response import V1ListStudioJobsResponse
589
620
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_user_slurm_jobs_response import V1ListUserSLURMJobsResponse
621
+ from lightning_sdk.lightning_cloud.openapi.models.v1_lit_dataset import V1LitDataset
622
+ from lightning_sdk.lightning_cloud.openapi.models.v1_lit_dataset_file import V1LitDatasetFile
623
+ from lightning_sdk.lightning_cloud.openapi.models.v1_lit_dataset_version_archive import V1LitDatasetVersionArchive
590
624
  from lightning_sdk.lightning_cloud.openapi.models.v1_lit_page import V1LitPage
591
625
  from lightning_sdk.lightning_cloud.openapi.models.v1_lit_page_type import V1LitPageType
592
626
  from lightning_sdk.lightning_cloud.openapi.models.v1_lit_registry_artifact import V1LitRegistryArtifact
@@ -648,10 +682,15 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_path_mapping import V1PathM
648
682
  from lightning_sdk.lightning_cloud.openapi.models.v1_path_telemetry import V1PathTelemetry
649
683
  from lightning_sdk.lightning_cloud.openapi.models.v1_phase_type import V1PhaseType
650
684
  from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline import V1Pipeline
651
- from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_schedule import V1PipelineSchedule
685
+ from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_parameter import V1PipelineParameter
686
+ from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_parameter_placement import V1PipelineParameterPlacement
687
+ from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_parameter_placement_type import V1PipelineParameterPlacementType
688
+ from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_parameter_type import V1PipelineParameterType
652
689
  from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_step import V1PipelineStep
653
690
  from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_step_status import V1PipelineStepStatus
654
691
  from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_step_type import V1PipelineStepType
692
+ from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_template import V1PipelineTemplate
693
+ from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_template_visibility_type import V1PipelineTemplateVisibilityType
655
694
  from lightning_sdk.lightning_cloud.openapi.models.v1_plugin import V1Plugin
656
695
  from lightning_sdk.lightning_cloud.openapi.models.v1_plugins_list_response import V1PluginsListResponse
657
696
  from lightning_sdk.lightning_cloud.openapi.models.v1_post_cloud_space_artifact_events_response import V1PostCloudSpaceArtifactEventsResponse
@@ -692,6 +731,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_report_restart_timings_resp
692
731
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_request import V1RequestClusterAccessRequest
693
732
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_response import V1RequestClusterAccessResponse
694
733
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_verification_code_response import V1RequestVerificationCodeResponse
734
+ from lightning_sdk.lightning_cloud.openapi.models.v1_reservation_details import V1ReservationDetails
695
735
  from lightning_sdk.lightning_cloud.openapi.models.v1_resource_tag import V1ResourceTag
696
736
  from lightning_sdk.lightning_cloud.openapi.models.v1_resource_visibility import V1ResourceVisibility
697
737
  from lightning_sdk.lightning_cloud.openapi.models.v1_resources import V1Resources
@@ -711,12 +751,17 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_s3_folder_data_connection i
711
751
  from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_job import V1SLURMJob
712
752
  from lightning_sdk.lightning_cloud.openapi.models.v1_ssh_key_pair import V1SSHKeyPair
713
753
  from lightning_sdk.lightning_cloud.openapi.models.v1_ssh_public_key import V1SSHPublicKey
754
+ from lightning_sdk.lightning_cloud.openapi.models.v1_schedule import V1Schedule
755
+ from lightning_sdk.lightning_cloud.openapi.models.v1_schedule_resource_type import V1ScheduleResourceType
714
756
  from lightning_sdk.lightning_cloud.openapi.models.v1_search_job_logs_response import V1SearchJobLogsResponse
715
757
  from lightning_sdk.lightning_cloud.openapi.models.v1_search_user import V1SearchUser
716
758
  from lightning_sdk.lightning_cloud.openapi.models.v1_search_users_response import V1SearchUsersResponse
717
759
  from lightning_sdk.lightning_cloud.openapi.models.v1_secret import V1Secret
718
760
  from lightning_sdk.lightning_cloud.openapi.models.v1_secret_type import V1SecretType
719
761
  from lightning_sdk.lightning_cloud.openapi.models.v1_select import V1Select
762
+ from lightning_sdk.lightning_cloud.openapi.models.v1_server_alert import V1ServerAlert
763
+ from lightning_sdk.lightning_cloud.openapi.models.v1_server_alert_severity import V1ServerAlertSeverity
764
+ from lightning_sdk.lightning_cloud.openapi.models.v1_server_alert_type import V1ServerAlertType
720
765
  from lightning_sdk.lightning_cloud.openapi.models.v1_server_check_in_response import V1ServerCheckInResponse
721
766
  from lightning_sdk.lightning_cloud.openapi.models.v1_service_artifact import V1ServiceArtifact
722
767
  from lightning_sdk.lightning_cloud.openapi.models.v1_service_execution import V1ServiceExecution
@@ -726,6 +771,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_setup_data_connection_respo
726
771
  from lightning_sdk.lightning_cloud.openapi.models.v1_shared_filesystem import V1SharedFilesystem
727
772
  from lightning_sdk.lightning_cloud.openapi.models.v1_should_start_syncing_response import V1ShouldStartSyncingResponse
728
773
  from lightning_sdk.lightning_cloud.openapi.models.v1_signed_url import V1SignedUrl
774
+ from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_cluster_user import V1SlurmClusterUser
729
775
  from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_node import V1SlurmNode
730
776
  from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_v1 import V1SlurmV1
731
777
  from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_v1_job_status import V1SlurmV1JobStatus
@@ -740,6 +786,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_storage_asset_type import V
740
786
  from lightning_sdk.lightning_cloud.openapi.models.v1_storage_system_metrics import V1StorageSystemMetrics
741
787
  from lightning_sdk.lightning_cloud.openapi.models.v1_studio_job import V1StudioJob
742
788
  from lightning_sdk.lightning_cloud.openapi.models.v1_studio_job_app import V1StudioJobApp
789
+ from lightning_sdk.lightning_cloud.openapi.models.v1_subnet_spec import V1SubnetSpec
743
790
  from lightning_sdk.lightning_cloud.openapi.models.v1_switch_cloud_space_instance_response import V1SwitchCloudSpaceInstanceResponse
744
791
  from lightning_sdk.lightning_cloud.openapi.models.v1_system_info import V1SystemInfo
745
792
  from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics import V1SystemMetrics
@@ -763,6 +810,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_availability
763
810
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_deployment_visibility_response import V1UpdateDeploymentVisibilityResponse
764
811
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_index_response import V1UpdateIndexResponse
765
812
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_job_visibility_response import V1UpdateJobVisibilityResponse
813
+ from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_dataset_visibility_response import V1UpdateLitDatasetVisibilityResponse
766
814
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_page_response import V1UpdateLitPageResponse
767
815
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_repository_response import V1UpdateLitRepositoryResponse
768
816
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_metrics_stream_visibility_response import V1UpdateMetricsStreamVisibilityResponse
@@ -802,13 +850,17 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_endpoint_r
802
850
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_endpoint_response import V1ValidateManagedEndpointResponse
803
851
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_model_response import V1ValidateManagedModelResponse
804
852
  from lightning_sdk.lightning_cloud.openapi.models.v1_verify_verification_response import V1VerifyVerificationResponse
853
+ from lightning_sdk.lightning_cloud.openapi.models.v1_voltage_park_direct_v1 import V1VoltageParkDirectV1
805
854
  from lightning_sdk.lightning_cloud.openapi.models.v1_volume import V1Volume
806
855
  from lightning_sdk.lightning_cloud.openapi.models.v1_vultr_direct_v1 import V1VultrDirectV1
807
856
  from lightning_sdk.lightning_cloud.openapi.models.v1_work import V1Work
808
857
  from lightning_sdk.lightning_cloud.openapi.models.validate import Validate
809
858
  from lightning_sdk.lightning_cloud.openapi.models.validateautojoindomain_domain_body import ValidateautojoindomainDomainBody
810
859
  from lightning_sdk.lightning_cloud.openapi.models.version_default_body import VersionDefaultBody
860
+ from lightning_sdk.lightning_cloud.openapi.models.version_default_body1 import VersionDefaultBody1
811
861
  from lightning_sdk.lightning_cloud.openapi.models.version_uploads_body import VersionUploadsBody
862
+ from lightning_sdk.lightning_cloud.openapi.models.version_uploads_body1 import VersionUploadsBody1
812
863
  from lightning_sdk.lightning_cloud.openapi.models.versions_id_body import VersionsIdBody
813
864
  from lightning_sdk.lightning_cloud.openapi.models.versions_version_body import VersionsVersionBody
865
+ from lightning_sdk.lightning_cloud.openapi.models.versions_version_body1 import VersionsVersionBody1
814
866
  from lightning_sdk.lightning_cloud.openapi.models.works_id_body import WorksIdBody
@@ -41,7 +41,7 @@ class ClusterIdCapacityblockBody(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
- 'capacity_block_duration_days': 'int',
44
+ 'capacity_block_duration_hours': 'int',
45
45
  'instance_count': 'int',
46
46
  'instance_type': 'str',
47
47
  'org_id': 'str',
@@ -50,7 +50,7 @@ class ClusterIdCapacityblockBody(object):
50
50
  }
51
51
 
52
52
  attribute_map = {
53
- 'capacity_block_duration_days': 'capacityBlockDurationDays',
53
+ 'capacity_block_duration_hours': 'capacityBlockDurationHours',
54
54
  'instance_count': 'instanceCount',
55
55
  'instance_type': 'instanceType',
56
56
  'org_id': 'orgId',
@@ -58,17 +58,17 @@ class ClusterIdCapacityblockBody(object):
58
58
  'timezone': 'timezone'
59
59
  }
60
60
 
61
- def __init__(self, capacity_block_duration_days: 'int' =None, instance_count: 'int' =None, instance_type: 'str' =None, org_id: 'str' =None, start_date: 'datetime' =None, timezone: 'str' =None): # noqa: E501
61
+ def __init__(self, capacity_block_duration_hours: 'int' =None, instance_count: 'int' =None, instance_type: 'str' =None, org_id: 'str' =None, start_date: 'datetime' =None, timezone: 'str' =None): # noqa: E501
62
62
  """ClusterIdCapacityblockBody - a model defined in Swagger""" # noqa: E501
63
- self._capacity_block_duration_days = None
63
+ self._capacity_block_duration_hours = None
64
64
  self._instance_count = None
65
65
  self._instance_type = None
66
66
  self._org_id = None
67
67
  self._start_date = None
68
68
  self._timezone = None
69
69
  self.discriminator = None
70
- if capacity_block_duration_days is not None:
71
- self.capacity_block_duration_days = capacity_block_duration_days
70
+ if capacity_block_duration_hours is not None:
71
+ self.capacity_block_duration_hours = capacity_block_duration_hours
72
72
  if instance_count is not None:
73
73
  self.instance_count = instance_count
74
74
  if instance_type is not None:
@@ -81,25 +81,25 @@ class ClusterIdCapacityblockBody(object):
81
81
  self.timezone = timezone
82
82
 
83
83
  @property
84
- def capacity_block_duration_days(self) -> 'int':
85
- """Gets the capacity_block_duration_days of this ClusterIdCapacityblockBody. # noqa: E501
84
+ def capacity_block_duration_hours(self) -> 'int':
85
+ """Gets the capacity_block_duration_hours of this ClusterIdCapacityblockBody. # noqa: E501
86
86
 
87
87
 
88
- :return: The capacity_block_duration_days of this ClusterIdCapacityblockBody. # noqa: E501
88
+ :return: The capacity_block_duration_hours of this ClusterIdCapacityblockBody. # noqa: E501
89
89
  :rtype: int
90
90
  """
91
- return self._capacity_block_duration_days
91
+ return self._capacity_block_duration_hours
92
92
 
93
- @capacity_block_duration_days.setter
94
- def capacity_block_duration_days(self, capacity_block_duration_days: 'int'):
95
- """Sets the capacity_block_duration_days of this ClusterIdCapacityblockBody.
93
+ @capacity_block_duration_hours.setter
94
+ def capacity_block_duration_hours(self, capacity_block_duration_hours: 'int'):
95
+ """Sets the capacity_block_duration_hours of this ClusterIdCapacityblockBody.
96
96
 
97
97
 
98
- :param capacity_block_duration_days: The capacity_block_duration_days of this ClusterIdCapacityblockBody. # noqa: E501
98
+ :param capacity_block_duration_hours: The capacity_block_duration_hours of this ClusterIdCapacityblockBody. # noqa: E501
99
99
  :type: int
100
100
  """
101
101
 
102
- self._capacity_block_duration_days = capacity_block_duration_days
102
+ self._capacity_block_duration_hours = capacity_block_duration_hours
103
103
 
104
104
  @property
105
105
  def instance_count(self) -> 'int':