lightning-sdk 0.2.4__py3-none-any.whl → 0.2.5__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (103) 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/deployment/deployment.py +32 -4
  6. lightning_sdk/lightning_cloud/openapi/__init__.py +49 -1
  7. lightning_sdk/lightning_cloud/openapi/api/__init__.py +4 -0
  8. lightning_sdk/lightning_cloud/openapi/api/cloud_space_environment_template_service_api.py +537 -0
  9. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +10 -6
  10. lightning_sdk/lightning_cloud/openapi/api/lit_dataset_service_api.py +1973 -0
  11. lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py +5 -1
  12. lightning_sdk/lightning_cloud/openapi/api/models_store_api.py +20 -20
  13. lightning_sdk/lightning_cloud/openapi/api/pipeline_templates_service_api.py +339 -0
  14. lightning_sdk/lightning_cloud/openapi/api/pipelines_service_api.py +5 -1
  15. lightning_sdk/lightning_cloud/openapi/api/schedules_service_api.py +573 -0
  16. lightning_sdk/lightning_cloud/openapi/api/slurm_jobs_user_service_api.py +202 -0
  17. lightning_sdk/lightning_cloud/openapi/models/__init__.py +45 -1
  18. lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityblock_body.py +15 -15
  19. lightning_sdk/lightning_cloud/openapi/models/cluster_id_slurmusers_body.py +201 -0
  20. lightning_sdk/lightning_cloud/openapi/models/dataset_id_versions_body.py +149 -0
  21. lightning_sdk/lightning_cloud/openapi/models/dataset_id_visibility_body.py +149 -0
  22. lightning_sdk/lightning_cloud/openapi/models/environmenttemplates_id_body.py +227 -0
  23. lightning_sdk/lightning_cloud/openapi/models/externalv1_cloud_space_instance_status.py +69 -69
  24. lightning_sdk/lightning_cloud/openapi/models/litdatasets_dataset_id_body.py +149 -0
  25. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +27 -1
  26. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body.py +69 -17
  27. lightning_sdk/lightning_cloud/openapi/models/pipelinetemplates_id_body.py +331 -0
  28. lightning_sdk/lightning_cloud/openapi/models/project_id_litdatasets_body.py +227 -0
  29. lightning_sdk/lightning_cloud/openapi/models/project_id_pipelines_body.py +17 -17
  30. lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py +201 -0
  31. lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py +383 -0
  32. lightning_sdk/lightning_cloud/openapi/models/slurm_jobs_body.py +15 -15
  33. lightning_sdk/lightning_cloud/openapi/models/upload_id_complete_body1.py +149 -0
  34. lightning_sdk/lightning_cloud/openapi/models/upload_id_parts_body1.py +149 -0
  35. lightning_sdk/lightning_cloud/openapi/models/v1_agent_job.py +124 -20
  36. lightning_sdk/lightning_cloud/openapi/models/v1_assistant_model_status.py +2 -0
  37. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -0
  38. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +27 -1
  39. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template.py +253 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  41. lightning_sdk/lightning_cloud/openapi/models/v1_complete_lit_dataset_multi_part_upload_response.py +97 -0
  42. lightning_sdk/lightning_cloud/openapi/models/v1_complete_lit_dataset_upload_response.py +97 -0
  43. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +149 -0
  44. lightning_sdk/lightning_cloud/openapi/models/v1_create_lit_dataset_multi_part_upload_response.py +123 -0
  45. lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py +27 -1
  46. lightning_sdk/lightning_cloud/openapi/models/v1_create_pipeline_template_request.py +383 -0
  47. lightning_sdk/lightning_cloud/openapi/models/v1_create_project_request.py +27 -1
  48. lightning_sdk/lightning_cloud/openapi/models/{v1_pipeline_schedule.py → v1_delete_cloud_space_environment_template_response.py} +32 -32
  49. lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_dataset_response.py +97 -0
  50. lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_dataset_version_response.py +97 -0
  51. lightning_sdk/lightning_cloud/openapi/models/v1_delete_schedule_response.py +175 -0
  52. lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_size_response.py +79 -1
  53. lightning_sdk/lightning_cloud/openapi/models/v1_get_lit_dataset_file_upload_urls_response.py +123 -0
  54. lightning_sdk/lightning_cloud/openapi/models/v1_get_lit_dataset_files_url_response.py +149 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
  56. lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.py +79 -27
  57. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +27 -1
  58. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloud_space_environment_templates_response.py +123 -0
  59. lightning_sdk/lightning_cloud/openapi/models/v1_list_lit_dataset_versions_response.py +123 -0
  60. lightning_sdk/lightning_cloud/openapi/models/v1_list_lit_datasets_response.py +123 -0
  61. lightning_sdk/lightning_cloud/openapi/models/v1_list_schedules_response.py +123 -0
  62. lightning_sdk/lightning_cloud/openapi/models/v1_list_slurm_cluster_users_response.py +123 -0
  63. lightning_sdk/lightning_cloud/openapi/models/v1_lit_dataset.py +539 -0
  64. lightning_sdk/lightning_cloud/openapi/models/v1_lit_dataset_file.py +175 -0
  65. lightning_sdk/lightning_cloud/openapi/models/v1_lit_dataset_version_archive.py +435 -0
  66. lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_project.py +27 -1
  67. lightning_sdk/lightning_cloud/openapi/models/v1_lit_repository.py +27 -1
  68. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_request.py +27 -1
  69. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +107 -3
  70. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +53 -1
  71. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py +69 -17
  72. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter.py +435 -0
  73. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter_placement.py +149 -0
  74. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter_placement_type.py +106 -0
  75. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter_type.py +106 -0
  76. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_template.py +513 -0
  77. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_template_visibility_type.py +105 -0
  78. lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py +27 -1
  79. lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py +435 -0
  80. lightning_sdk/lightning_cloud/openapi/models/v1_schedule_resource_type.py +103 -0
  81. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_cluster_user.py +227 -0
  82. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_job.py +58 -6
  83. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_node.py +31 -291
  84. lightning_sdk/lightning_cloud/openapi/models/v1_update_lit_dataset_visibility_response.py +123 -0
  85. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +27 -1
  86. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +157 -105
  87. lightning_sdk/lightning_cloud/openapi/models/v1_validate_managed_endpoint_response.py +27 -1
  88. lightning_sdk/lightning_cloud/openapi/models/v1_voltage_park_direct_v1.py +203 -0
  89. lightning_sdk/lightning_cloud/openapi/models/version_default_body.py +29 -29
  90. lightning_sdk/lightning_cloud/openapi/models/version_default_body1.py +149 -0
  91. lightning_sdk/lightning_cloud/openapi/models/version_uploads_body1.py +123 -0
  92. lightning_sdk/lightning_cloud/openapi/models/versions_version_body1.py +123 -0
  93. lightning_sdk/lightning_cloud/rest_client.py +2 -0
  94. lightning_sdk/pipeline/types.py +28 -2
  95. lightning_sdk/pipeline/utils.py +1 -1
  96. lightning_sdk/plugin.py +0 -6
  97. lightning_sdk/utils/resolve.py +1 -0
  98. {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.5.dist-info}/METADATA +1 -1
  99. {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.5.dist-info}/RECORD +103 -55
  100. {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.5.dist-info}/WHEEL +1 -1
  101. {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.5.dist-info}/LICENSE +0 -0
  102. {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.5.dist-info}/entry_points.txt +0 -0
  103. {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.5.dist-info}/top_level.txt +0 -0
@@ -261,6 +261,7 @@ class LitRegistryServiceApi(object):
261
261
  :param str project_id: (required)
262
262
  :param bool latest_only:
263
263
  :param str filter_for_repo: Use this to filter for a single repository. Useful when selecting a subset of the same data from various context.
264
+ :param str cluster_id:
264
265
  :return: V1LitRegistryProject
265
266
  If the method is called asynchronously,
266
267
  returns the request thread.
@@ -284,12 +285,13 @@ class LitRegistryServiceApi(object):
284
285
  :param str project_id: (required)
285
286
  :param bool latest_only:
286
287
  :param str filter_for_repo: Use this to filter for a single repository. Useful when selecting a subset of the same data from various context.
288
+ :param str cluster_id:
287
289
  :return: V1LitRegistryProject
288
290
  If the method is called asynchronously,
289
291
  returns the request thread.
290
292
  """
291
293
 
292
- all_params = ['project_id', 'latest_only', 'filter_for_repo'] # noqa: E501
294
+ all_params = ['project_id', 'latest_only', 'filter_for_repo', 'cluster_id'] # noqa: E501
293
295
  all_params.append('async_req')
294
296
  all_params.append('_return_http_data_only')
295
297
  all_params.append('_preload_content')
@@ -320,6 +322,8 @@ class LitRegistryServiceApi(object):
320
322
  query_params.append(('latestOnly', params['latest_only'])) # noqa: E501
321
323
  if 'filter_for_repo' in params:
322
324
  query_params.append(('filterForRepo', params['filter_for_repo'])) # noqa: E501
325
+ if 'cluster_id' in params:
326
+ query_params.append(('clusterId', params['cluster_id'])) # noqa: E501
323
327
 
324
328
  header_params = {}
325
329
 
@@ -164,7 +164,7 @@ class ModelsStoreApi(object):
164
164
  _request_timeout=params.get('_request_timeout'),
165
165
  collection_formats=collection_formats)
166
166
 
167
- def models_store_complete_multi_part_upload(self, body: 'UploadIdCompleteBody', project_id: 'str', model_id: 'str', version: 'str', upload_id: 'str', **kwargs) -> 'V1CompleteMultiPartUploadResponse': # noqa: E501
167
+ def models_store_complete_multi_part_upload(self, body: 'UploadIdCompleteBody1', project_id: 'str', model_id: 'str', version: 'str', upload_id: 'str', **kwargs) -> 'V1CompleteMultiPartUploadResponse': # noqa: E501
168
168
  """CompleteMultiPartUpload marks the given file-parts as completed/uploaded. Once all parts of a file are completed, the file will materialize in S3 storage # noqa: E501
169
169
 
170
170
  This method makes a synchronous HTTP request by default. To make an
@@ -173,7 +173,7 @@ class ModelsStoreApi(object):
173
173
  >>> result = thread.get()
174
174
 
175
175
  :param async_req bool
176
- :param UploadIdCompleteBody body: (required)
176
+ :param UploadIdCompleteBody1 body: (required)
177
177
  :param str project_id: (required)
178
178
  :param str model_id: (required)
179
179
  :param str version: (required)
@@ -189,7 +189,7 @@ class ModelsStoreApi(object):
189
189
  (data) = self.models_store_complete_multi_part_upload_with_http_info(body, project_id, model_id, version, upload_id, **kwargs) # noqa: E501
190
190
  return data
191
191
 
192
- def models_store_complete_multi_part_upload_with_http_info(self, body: 'UploadIdCompleteBody', project_id: 'str', model_id: 'str', version: 'str', upload_id: 'str', **kwargs) -> 'V1CompleteMultiPartUploadResponse': # noqa: E501
192
+ def models_store_complete_multi_part_upload_with_http_info(self, body: 'UploadIdCompleteBody1', project_id: 'str', model_id: 'str', version: 'str', upload_id: 'str', **kwargs) -> 'V1CompleteMultiPartUploadResponse': # noqa: E501
193
193
  """CompleteMultiPartUpload marks the given file-parts as completed/uploaded. Once all parts of a file are completed, the file will materialize in S3 storage # noqa: E501
194
194
 
195
195
  This method makes a synchronous HTTP request by default. To make an
@@ -198,7 +198,7 @@ class ModelsStoreApi(object):
198
198
  >>> result = thread.get()
199
199
 
200
200
  :param async_req bool
201
- :param UploadIdCompleteBody body: (required)
201
+ :param UploadIdCompleteBody1 body: (required)
202
202
  :param str project_id: (required)
203
203
  :param str model_id: (required)
204
204
  :param str version: (required)
@@ -511,7 +511,7 @@ class ModelsStoreApi(object):
511
511
  _request_timeout=params.get('_request_timeout'),
512
512
  collection_formats=collection_formats)
513
513
 
514
- def models_store_create_multi_part_upload(self, body: 'VersionUploadsBody', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1CreateMultiPartUploadResponse': # noqa: E501
514
+ def models_store_create_multi_part_upload(self, body: 'VersionUploadsBody1', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1CreateMultiPartUploadResponse': # noqa: E501
515
515
  """CreateMultiPartUpload initiates the multi-part upload of a file. Multiple requests can be sent to upload different files. # noqa: E501
516
516
 
517
517
  This method makes a synchronous HTTP request by default. To make an
@@ -520,7 +520,7 @@ class ModelsStoreApi(object):
520
520
  >>> result = thread.get()
521
521
 
522
522
  :param async_req bool
523
- :param VersionUploadsBody body: (required)
523
+ :param VersionUploadsBody1 body: (required)
524
524
  :param str project_id: (required)
525
525
  :param str model_id: (required)
526
526
  :param str version: (required)
@@ -535,7 +535,7 @@ class ModelsStoreApi(object):
535
535
  (data) = self.models_store_create_multi_part_upload_with_http_info(body, project_id, model_id, version, **kwargs) # noqa: E501
536
536
  return data
537
537
 
538
- def models_store_create_multi_part_upload_with_http_info(self, body: 'VersionUploadsBody', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1CreateMultiPartUploadResponse': # noqa: E501
538
+ def models_store_create_multi_part_upload_with_http_info(self, body: 'VersionUploadsBody1', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1CreateMultiPartUploadResponse': # noqa: E501
539
539
  """CreateMultiPartUpload initiates the multi-part upload of a file. Multiple requests can be sent to upload different files. # noqa: E501
540
540
 
541
541
  This method makes a synchronous HTTP request by default. To make an
@@ -544,7 +544,7 @@ class ModelsStoreApi(object):
544
544
  >>> result = thread.get()
545
545
 
546
546
  :param async_req bool
547
- :param VersionUploadsBody body: (required)
547
+ :param VersionUploadsBody1 body: (required)
548
548
  :param str project_id: (required)
549
549
  :param str model_id: (required)
550
550
  :param str version: (required)
@@ -1052,7 +1052,7 @@ class ModelsStoreApi(object):
1052
1052
  _request_timeout=params.get('_request_timeout'),
1053
1053
  collection_formats=collection_formats)
1054
1054
 
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
1055
+ def models_store_get_model_file_upload_urls(self, body: 'UploadIdPartsBody1', project_id: 'str', model_id: 'str', version: 'str', upload_id: 'str', **kwargs) -> 'V1GetModelFileUploadUrlsResponse': # noqa: E501
1056
1056
  """GetModelFileUploadUrls requests pre-signed URLs for a given number of file parts. # noqa: E501
1057
1057
 
1058
1058
  This method makes a synchronous HTTP request by default. To make an
@@ -1061,7 +1061,7 @@ class ModelsStoreApi(object):
1061
1061
  >>> result = thread.get()
1062
1062
 
1063
1063
  :param async_req bool
1064
- :param UploadIdPartsBody body: (required)
1064
+ :param UploadIdPartsBody1 body: (required)
1065
1065
  :param str project_id: (required)
1066
1066
  :param str model_id: (required)
1067
1067
  :param str version: (required)
@@ -1077,7 +1077,7 @@ class ModelsStoreApi(object):
1077
1077
  (data) = self.models_store_get_model_file_upload_urls_with_http_info(body, project_id, model_id, version, upload_id, **kwargs) # noqa: E501
1078
1078
  return data
1079
1079
 
1080
- def models_store_get_model_file_upload_urls_with_http_info(self, body: 'UploadIdPartsBody', project_id: 'str', model_id: 'str', version: 'str', upload_id: 'str', **kwargs) -> 'V1GetModelFileUploadUrlsResponse': # noqa: E501
1080
+ def models_store_get_model_file_upload_urls_with_http_info(self, body: 'UploadIdPartsBody1', project_id: 'str', model_id: 'str', version: 'str', upload_id: 'str', **kwargs) -> 'V1GetModelFileUploadUrlsResponse': # noqa: E501
1081
1081
  """GetModelFileUploadUrls requests pre-signed URLs for a given number of file parts. # noqa: E501
1082
1082
 
1083
1083
  This method makes a synchronous HTTP request by default. To make an
@@ -1086,7 +1086,7 @@ class ModelsStoreApi(object):
1086
1086
  >>> result = thread.get()
1087
1087
 
1088
1088
  :param async_req bool
1089
- :param UploadIdPartsBody body: (required)
1089
+ :param UploadIdPartsBody1 body: (required)
1090
1090
  :param str project_id: (required)
1091
1091
  :param str model_id: (required)
1092
1092
  :param str version: (required)
@@ -1827,7 +1827,7 @@ class ModelsStoreApi(object):
1827
1827
  _request_timeout=params.get('_request_timeout'),
1828
1828
  collection_formats=collection_formats)
1829
1829
 
1830
- def models_store_set_model_default_version(self, body: 'VersionDefaultBody', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1Model': # noqa: E501
1830
+ def models_store_set_model_default_version(self, body: 'VersionDefaultBody1', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1Model': # noqa: E501
1831
1831
  """models_store_set_model_default_version # noqa: E501
1832
1832
 
1833
1833
  This method makes a synchronous HTTP request by default. To make an
@@ -1836,7 +1836,7 @@ class ModelsStoreApi(object):
1836
1836
  >>> result = thread.get()
1837
1837
 
1838
1838
  :param async_req bool
1839
- :param VersionDefaultBody body: (required)
1839
+ :param VersionDefaultBody1 body: (required)
1840
1840
  :param str project_id: (required)
1841
1841
  :param str model_id: (required)
1842
1842
  :param str version: (required)
@@ -1851,7 +1851,7 @@ class ModelsStoreApi(object):
1851
1851
  (data) = self.models_store_set_model_default_version_with_http_info(body, project_id, model_id, version, **kwargs) # noqa: E501
1852
1852
  return data
1853
1853
 
1854
- def models_store_set_model_default_version_with_http_info(self, body: 'VersionDefaultBody', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1Model': # noqa: E501
1854
+ def models_store_set_model_default_version_with_http_info(self, body: 'VersionDefaultBody1', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1Model': # noqa: E501
1855
1855
  """models_store_set_model_default_version # noqa: E501
1856
1856
 
1857
1857
  This method makes a synchronous HTTP request by default. To make an
@@ -1860,7 +1860,7 @@ class ModelsStoreApi(object):
1860
1860
  >>> result = thread.get()
1861
1861
 
1862
1862
  :param async_req bool
1863
- :param VersionDefaultBody body: (required)
1863
+ :param VersionDefaultBody1 body: (required)
1864
1864
  :param str project_id: (required)
1865
1865
  :param str model_id: (required)
1866
1866
  :param str version: (required)
@@ -2061,7 +2061,7 @@ class ModelsStoreApi(object):
2061
2061
  _request_timeout=params.get('_request_timeout'),
2062
2062
  collection_formats=collection_formats)
2063
2063
 
2064
- def models_store_update_model_version(self, body: 'VersionsVersionBody', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1ModelVersionArchive': # noqa: E501
2064
+ def models_store_update_model_version(self, body: 'VersionsVersionBody1', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1ModelVersionArchive': # noqa: E501
2065
2065
  """UpdateModelVersion updates the model version, in particular the model version's name. # noqa: E501
2066
2066
 
2067
2067
  This method makes a synchronous HTTP request by default. To make an
@@ -2070,7 +2070,7 @@ class ModelsStoreApi(object):
2070
2070
  >>> result = thread.get()
2071
2071
 
2072
2072
  :param async_req bool
2073
- :param VersionsVersionBody body: (required)
2073
+ :param VersionsVersionBody1 body: (required)
2074
2074
  :param str project_id: (required)
2075
2075
  :param str model_id: (required)
2076
2076
  :param str version: (required)
@@ -2085,7 +2085,7 @@ class ModelsStoreApi(object):
2085
2085
  (data) = self.models_store_update_model_version_with_http_info(body, project_id, model_id, version, **kwargs) # noqa: E501
2086
2086
  return data
2087
2087
 
2088
- def models_store_update_model_version_with_http_info(self, body: 'VersionsVersionBody', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1ModelVersionArchive': # noqa: E501
2088
+ def models_store_update_model_version_with_http_info(self, body: 'VersionsVersionBody1', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1ModelVersionArchive': # noqa: E501
2089
2089
  """UpdateModelVersion updates the model version, in particular the model version's name. # noqa: E501
2090
2090
 
2091
2091
  This method makes a synchronous HTTP request by default. To make an
@@ -2094,7 +2094,7 @@ class ModelsStoreApi(object):
2094
2094
  >>> result = thread.get()
2095
2095
 
2096
2096
  :param async_req bool
2097
- :param VersionsVersionBody body: (required)
2097
+ :param VersionsVersionBody1 body: (required)
2098
2098
  :param str project_id: (required)
2099
2099
  :param str model_id: (required)
2100
2100
  :param str version: (required)
@@ -0,0 +1,339 @@
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
+ from __future__ import absolute_import
21
+
22
+ import re # noqa: F401
23
+ from typing import TYPE_CHECKING, Any
24
+
25
+ # python 2 and python 3 compatibility library
26
+ import six
27
+
28
+ from lightning_sdk.lightning_cloud.openapi.api_client import ApiClient
29
+
30
+ if TYPE_CHECKING:
31
+ from datetime import datetime
32
+ from lightning_sdk.lightning_cloud.openapi.models import *
33
+
34
+ class PipelineTemplatesServiceApi(object):
35
+ """NOTE: This class is auto generated by the swagger code generator program.
36
+
37
+ Do not edit the class manually.
38
+ Ref: https://github.com/swagger-api/swagger-codegen
39
+ """
40
+
41
+ def __init__(self, api_client=None):
42
+ if api_client is None:
43
+ api_client = ApiClient()
44
+ self.api_client = api_client
45
+
46
+ def pipeline_templates_service_create_pipeline_template(self, body: 'V1CreatePipelineTemplateRequest', **kwargs) -> 'V1PipelineTemplate': # noqa: E501
47
+ """Pipeline templates # 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.pipeline_templates_service_create_pipeline_template(body, async_req=True)
52
+ >>> result = thread.get()
53
+
54
+ :param async_req bool
55
+ :param V1CreatePipelineTemplateRequest body: (required)
56
+ :return: V1PipelineTemplate
57
+ If the method is called asynchronously,
58
+ returns the request thread.
59
+ """
60
+ kwargs['_return_http_data_only'] = True
61
+ if kwargs.get('async_req'):
62
+ return self.pipeline_templates_service_create_pipeline_template_with_http_info(body, **kwargs) # noqa: E501
63
+ else:
64
+ (data) = self.pipeline_templates_service_create_pipeline_template_with_http_info(body, **kwargs) # noqa: E501
65
+ return data
66
+
67
+ def pipeline_templates_service_create_pipeline_template_with_http_info(self, body: 'V1CreatePipelineTemplateRequest', **kwargs) -> 'V1PipelineTemplate': # noqa: E501
68
+ """Pipeline templates # noqa: E501
69
+
70
+ This method makes a synchronous HTTP request by default. To make an
71
+ asynchronous HTTP request, please pass async_req=True
72
+ >>> thread = api.pipeline_templates_service_create_pipeline_template_with_http_info(body, async_req=True)
73
+ >>> result = thread.get()
74
+
75
+ :param async_req bool
76
+ :param V1CreatePipelineTemplateRequest body: (required)
77
+ :return: V1PipelineTemplate
78
+ If the method is called asynchronously,
79
+ returns the request thread.
80
+ """
81
+
82
+ all_params = ['body'] # noqa: E501
83
+ all_params.append('async_req')
84
+ all_params.append('_return_http_data_only')
85
+ all_params.append('_preload_content')
86
+ all_params.append('_request_timeout')
87
+
88
+ params = locals()
89
+ for key, val in six.iteritems(params['kwargs']):
90
+ if key not in all_params:
91
+ raise TypeError(
92
+ "Got an unexpected keyword argument '%s'"
93
+ " to method pipeline_templates_service_create_pipeline_template" % key
94
+ )
95
+ params[key] = val
96
+ del params['kwargs']
97
+ # verify the required parameter 'body' is set
98
+ if ('body' not in params or
99
+ params['body'] is None):
100
+ raise ValueError("Missing the required parameter `body` when calling `pipeline_templates_service_create_pipeline_template`") # noqa: E501
101
+
102
+ collection_formats = {}
103
+
104
+ path_params = {}
105
+
106
+ query_params = []
107
+
108
+ header_params = {}
109
+
110
+ form_params = []
111
+ local_var_files = {}
112
+
113
+ body_params = None
114
+ if 'body' in params:
115
+ body_params = params['body']
116
+ # HTTP header `Accept`
117
+ header_params['Accept'] = self.api_client.select_header_accept(
118
+ ['application/json']) # noqa: E501
119
+
120
+ # HTTP header `Content-Type`
121
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
122
+ ['application/json']) # noqa: E501
123
+
124
+ # Authentication setting
125
+ auth_settings = [] # noqa: E501
126
+
127
+ return self.api_client.call_api(
128
+ '/v1/pipeline-templates', 'POST',
129
+ path_params,
130
+ query_params,
131
+ header_params,
132
+ body=body_params,
133
+ post_params=form_params,
134
+ files=local_var_files,
135
+ response_type='V1PipelineTemplate', # noqa: E501
136
+ auth_settings=auth_settings,
137
+ async_req=params.get('async_req'),
138
+ _return_http_data_only=params.get('_return_http_data_only'),
139
+ _preload_content=params.get('_preload_content', True),
140
+ _request_timeout=params.get('_request_timeout'),
141
+ collection_formats=collection_formats)
142
+
143
+ def pipeline_templates_service_get_pipeline_template(self, id: 'str', **kwargs) -> 'V1PipelineTemplate': # noqa: E501
144
+ """pipeline_templates_service_get_pipeline_template # noqa: E501
145
+
146
+ This method makes a synchronous HTTP request by default. To make an
147
+ asynchronous HTTP request, please pass async_req=True
148
+ >>> thread = api.pipeline_templates_service_get_pipeline_template(id, async_req=True)
149
+ >>> result = thread.get()
150
+
151
+ :param async_req bool
152
+ :param str id: (required)
153
+ :return: V1PipelineTemplate
154
+ If the method is called asynchronously,
155
+ returns the request thread.
156
+ """
157
+ kwargs['_return_http_data_only'] = True
158
+ if kwargs.get('async_req'):
159
+ return self.pipeline_templates_service_get_pipeline_template_with_http_info(id, **kwargs) # noqa: E501
160
+ else:
161
+ (data) = self.pipeline_templates_service_get_pipeline_template_with_http_info(id, **kwargs) # noqa: E501
162
+ return data
163
+
164
+ def pipeline_templates_service_get_pipeline_template_with_http_info(self, id: 'str', **kwargs) -> 'V1PipelineTemplate': # noqa: E501
165
+ """pipeline_templates_service_get_pipeline_template # noqa: E501
166
+
167
+ This method makes a synchronous HTTP request by default. To make an
168
+ asynchronous HTTP request, please pass async_req=True
169
+ >>> thread = api.pipeline_templates_service_get_pipeline_template_with_http_info(id, async_req=True)
170
+ >>> result = thread.get()
171
+
172
+ :param async_req bool
173
+ :param str id: (required)
174
+ :return: V1PipelineTemplate
175
+ If the method is called asynchronously,
176
+ returns the request thread.
177
+ """
178
+
179
+ all_params = ['id'] # noqa: E501
180
+ all_params.append('async_req')
181
+ all_params.append('_return_http_data_only')
182
+ all_params.append('_preload_content')
183
+ all_params.append('_request_timeout')
184
+
185
+ params = locals()
186
+ for key, val in six.iteritems(params['kwargs']):
187
+ if key not in all_params:
188
+ raise TypeError(
189
+ "Got an unexpected keyword argument '%s'"
190
+ " to method pipeline_templates_service_get_pipeline_template" % key
191
+ )
192
+ params[key] = val
193
+ del params['kwargs']
194
+ # verify the required parameter 'id' is set
195
+ if ('id' not in params or
196
+ params['id'] is None):
197
+ raise ValueError("Missing the required parameter `id` when calling `pipeline_templates_service_get_pipeline_template`") # noqa: E501
198
+
199
+ collection_formats = {}
200
+
201
+ path_params = {}
202
+ if 'id' in params:
203
+ path_params['id'] = params['id'] # noqa: E501
204
+
205
+ query_params = []
206
+
207
+ header_params = {}
208
+
209
+ form_params = []
210
+ local_var_files = {}
211
+
212
+ body_params = None
213
+ # HTTP header `Accept`
214
+ header_params['Accept'] = self.api_client.select_header_accept(
215
+ ['application/json']) # noqa: E501
216
+
217
+ # Authentication setting
218
+ auth_settings = [] # noqa: E501
219
+
220
+ return self.api_client.call_api(
221
+ '/v1/pipeline-templates/{id}', 'GET',
222
+ path_params,
223
+ query_params,
224
+ header_params,
225
+ body=body_params,
226
+ post_params=form_params,
227
+ files=local_var_files,
228
+ response_type='V1PipelineTemplate', # noqa: E501
229
+ auth_settings=auth_settings,
230
+ async_req=params.get('async_req'),
231
+ _return_http_data_only=params.get('_return_http_data_only'),
232
+ _preload_content=params.get('_preload_content', True),
233
+ _request_timeout=params.get('_request_timeout'),
234
+ collection_formats=collection_formats)
235
+
236
+ def pipeline_templates_service_update_pipeline_template(self, body: 'PipelinetemplatesIdBody', id: 'str', **kwargs) -> 'V1PipelineTemplate': # noqa: E501
237
+ """pipeline_templates_service_update_pipeline_template # noqa: E501
238
+
239
+ This method makes a synchronous HTTP request by default. To make an
240
+ asynchronous HTTP request, please pass async_req=True
241
+ >>> thread = api.pipeline_templates_service_update_pipeline_template(body, id, async_req=True)
242
+ >>> result = thread.get()
243
+
244
+ :param async_req bool
245
+ :param PipelinetemplatesIdBody body: (required)
246
+ :param str id: (required)
247
+ :return: V1PipelineTemplate
248
+ If the method is called asynchronously,
249
+ returns the request thread.
250
+ """
251
+ kwargs['_return_http_data_only'] = True
252
+ if kwargs.get('async_req'):
253
+ return self.pipeline_templates_service_update_pipeline_template_with_http_info(body, id, **kwargs) # noqa: E501
254
+ else:
255
+ (data) = self.pipeline_templates_service_update_pipeline_template_with_http_info(body, id, **kwargs) # noqa: E501
256
+ return data
257
+
258
+ def pipeline_templates_service_update_pipeline_template_with_http_info(self, body: 'PipelinetemplatesIdBody', id: 'str', **kwargs) -> 'V1PipelineTemplate': # noqa: E501
259
+ """pipeline_templates_service_update_pipeline_template # noqa: E501
260
+
261
+ This method makes a synchronous HTTP request by default. To make an
262
+ asynchronous HTTP request, please pass async_req=True
263
+ >>> thread = api.pipeline_templates_service_update_pipeline_template_with_http_info(body, id, async_req=True)
264
+ >>> result = thread.get()
265
+
266
+ :param async_req bool
267
+ :param PipelinetemplatesIdBody body: (required)
268
+ :param str id: (required)
269
+ :return: V1PipelineTemplate
270
+ If the method is called asynchronously,
271
+ returns the request thread.
272
+ """
273
+
274
+ all_params = ['body', 'id'] # noqa: E501
275
+ all_params.append('async_req')
276
+ all_params.append('_return_http_data_only')
277
+ all_params.append('_preload_content')
278
+ all_params.append('_request_timeout')
279
+
280
+ params = locals()
281
+ for key, val in six.iteritems(params['kwargs']):
282
+ if key not in all_params:
283
+ raise TypeError(
284
+ "Got an unexpected keyword argument '%s'"
285
+ " to method pipeline_templates_service_update_pipeline_template" % key
286
+ )
287
+ params[key] = val
288
+ del params['kwargs']
289
+ # verify the required parameter 'body' is set
290
+ if ('body' not in params or
291
+ params['body'] is None):
292
+ raise ValueError("Missing the required parameter `body` when calling `pipeline_templates_service_update_pipeline_template`") # noqa: E501
293
+ # verify the required parameter 'id' is set
294
+ if ('id' not in params or
295
+ params['id'] is None):
296
+ raise ValueError("Missing the required parameter `id` when calling `pipeline_templates_service_update_pipeline_template`") # noqa: E501
297
+
298
+ collection_formats = {}
299
+
300
+ path_params = {}
301
+ if 'id' in params:
302
+ path_params['id'] = params['id'] # noqa: E501
303
+
304
+ query_params = []
305
+
306
+ header_params = {}
307
+
308
+ form_params = []
309
+ local_var_files = {}
310
+
311
+ body_params = None
312
+ if 'body' in params:
313
+ body_params = params['body']
314
+ # HTTP header `Accept`
315
+ header_params['Accept'] = self.api_client.select_header_accept(
316
+ ['application/json']) # noqa: E501
317
+
318
+ # HTTP header `Content-Type`
319
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
320
+ ['application/json']) # noqa: E501
321
+
322
+ # Authentication setting
323
+ auth_settings = [] # noqa: E501
324
+
325
+ return self.api_client.call_api(
326
+ '/v1/pipeline-templates/{id}', 'PUT',
327
+ path_params,
328
+ query_params,
329
+ header_params,
330
+ body=body_params,
331
+ post_params=form_params,
332
+ files=local_var_files,
333
+ response_type='V1PipelineTemplate', # noqa: E501
334
+ auth_settings=auth_settings,
335
+ async_req=params.get('async_req'),
336
+ _return_http_data_only=params.get('_return_http_data_only'),
337
+ _preload_content=params.get('_preload_content', True),
338
+ _request_timeout=params.get('_request_timeout'),
339
+ collection_formats=collection_formats)
@@ -465,6 +465,7 @@ class PipelinesServiceApi(object):
465
465
  :param str page_token:
466
466
  :param int limit:
467
467
  :param str state:
468
+ :param bool standalone:
468
469
  :return: V1ListPipelinesResponse
469
470
  If the method is called asynchronously,
470
471
  returns the request thread.
@@ -490,12 +491,13 @@ class PipelinesServiceApi(object):
490
491
  :param str page_token:
491
492
  :param int limit:
492
493
  :param str state:
494
+ :param bool standalone:
493
495
  :return: V1ListPipelinesResponse
494
496
  If the method is called asynchronously,
495
497
  returns the request thread.
496
498
  """
497
499
 
498
- all_params = ['project_id', 'parent_pipeline_id', 'page_token', 'limit', 'state'] # noqa: E501
500
+ all_params = ['project_id', 'parent_pipeline_id', 'page_token', 'limit', 'state', 'standalone'] # noqa: E501
499
501
  all_params.append('async_req')
500
502
  all_params.append('_return_http_data_only')
501
503
  all_params.append('_preload_content')
@@ -530,6 +532,8 @@ class PipelinesServiceApi(object):
530
532
  query_params.append(('limit', params['limit'])) # noqa: E501
531
533
  if 'state' in params:
532
534
  query_params.append(('state', params['state'])) # noqa: E501
535
+ if 'standalone' in params:
536
+ query_params.append(('standalone', params['standalone'])) # noqa: E501
533
537
 
534
538
  header_params = {}
535
539