lightning-sdk 0.1.38__py3-none-any.whl → 0.1.40__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 (53) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/_mmt/__init__.py +3 -0
  3. lightning_sdk/_mmt/base.py +180 -0
  4. lightning_sdk/_mmt/mmt.py +161 -0
  5. lightning_sdk/_mmt/v1.py +69 -0
  6. lightning_sdk/_mmt/v2.py +141 -0
  7. lightning_sdk/api/deployment_api.py +0 -2
  8. lightning_sdk/api/job_api.py +4 -0
  9. lightning_sdk/api/mmt_api.py +147 -0
  10. lightning_sdk/api/teamspace_api.py +4 -11
  11. lightning_sdk/api/utils.py +6 -3
  12. lightning_sdk/cli/download.py +3 -5
  13. lightning_sdk/cli/mmt.py +137 -0
  14. lightning_sdk/cli/run.py +16 -0
  15. lightning_sdk/cli/upload.py +3 -10
  16. lightning_sdk/job/base.py +24 -3
  17. lightning_sdk/job/job.py +10 -1
  18. lightning_sdk/job/v1.py +7 -1
  19. lightning_sdk/job/v2.py +18 -9
  20. lightning_sdk/lightning_cloud/openapi/__init__.py +7 -3
  21. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +90 -284
  22. lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py +6 -1
  23. lightning_sdk/lightning_cloud/openapi/api/models_store_api.py +235 -1
  24. lightning_sdk/lightning_cloud/openapi/models/__init__.py +7 -3
  25. lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py +27 -1
  26. lightning_sdk/lightning_cloud/openapi/models/id_start_body.py +29 -3
  27. lightning_sdk/lightning_cloud/openapi/models/model_id_visibility_body.py +123 -0
  28. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
  29. lightning_sdk/lightning_cloud/openapi/models/v1_aws_direct_v1.py +27 -1
  30. lightning_sdk/lightning_cloud/openapi/models/{project_id_agentmanagedmodels_body.py → v1_body.py} +21 -47
  31. lightning_sdk/lightning_cloud/openapi/models/v1_data_path.py +29 -3
  32. lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py +27 -1
  33. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_api.py +201 -0
  34. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_response.py +27 -1
  35. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +53 -53
  36. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +27 -1
  37. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model_abilities.py +175 -0
  38. lightning_sdk/lightning_cloud/openapi/models/v1_model.py +29 -3
  39. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py +53 -1
  40. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_state.py +1 -2
  41. lightning_sdk/lightning_cloud/openapi/models/v1_query_param.py +175 -0
  42. lightning_sdk/lightning_cloud/openapi/models/{v1_list_managed_models_response.py → v1_resource_visibility.py} +23 -23
  43. lightning_sdk/lightning_cloud/openapi/models/{v1_delete_managed_model_response.py → v1_update_model_visibility_response.py} +6 -6
  44. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +27 -1
  45. lightning_sdk/models.py +153 -0
  46. lightning_sdk/teamspace.py +15 -11
  47. {lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.40.dist-info}/METADATA +1 -1
  48. {lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.40.dist-info}/RECORD +52 -41
  49. {lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.40.dist-info}/entry_points.txt +1 -0
  50. lightning_sdk/cli/models.py +0 -68
  51. {lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.40.dist-info}/LICENSE +0 -0
  52. {lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.40.dist-info}/WHEEL +0 -0
  53. {lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.40.dist-info}/top_level.txt +0 -0
lightning_sdk/lightning_cloud/openapi/__init__.py CHANGED
@@ -137,6 +137,7 @@ from lightning_sdk.lightning_cloud.openapi.models.metricsstream_create_body impo
137
137
  from lightning_sdk.lightning_cloud.openapi.models.metricsstream_delete_body import MetricsstreamDeleteBody
138
138
  from lightning_sdk.lightning_cloud.openapi.models.metricsstream_id_body import MetricsstreamIdBody
139
139
  from lightning_sdk.lightning_cloud.openapi.models.model_id_versions_body import ModelIdVersionsBody
140
+ from lightning_sdk.lightning_cloud.openapi.models.model_id_visibility_body import ModelIdVisibilityBody
140
141
  from lightning_sdk.lightning_cloud.openapi.models.models_model_id_body import ModelsModelIdBody
141
142
  from lightning_sdk.lightning_cloud.openapi.models.multimachinejobs_id_body import MultimachinejobsIdBody
142
143
  from lightning_sdk.lightning_cloud.openapi.models.multipartuploads_upload_id_body import MultipartuploadsUploadIdBody
@@ -146,7 +147,6 @@ from lightning_sdk.lightning_cloud.openapi.models.orgs_id_body import OrgsIdBody
146
147
  from lightning_sdk.lightning_cloud.openapi.models.profiler_captures_body import ProfilerCapturesBody
147
148
  from lightning_sdk.lightning_cloud.openapi.models.profiler_enabled_body import ProfilerEnabledBody
148
149
  from lightning_sdk.lightning_cloud.openapi.models.project_id_agentmanagedendpoints_body import ProjectIdAgentmanagedendpointsBody
149
- from lightning_sdk.lightning_cloud.openapi.models.project_id_agentmanagedmodels_body import ProjectIdAgentmanagedmodelsBody
150
150
  from lightning_sdk.lightning_cloud.openapi.models.project_id_agents_body import ProjectIdAgentsBody
151
151
  from lightning_sdk.lightning_cloud.openapi.models.project_id_cloudspaces_body import ProjectIdCloudspacesBody
152
152
  from lightning_sdk.lightning_cloud.openapi.models.project_id_clusters_body import ProjectIdClustersBody
@@ -222,6 +222,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_billing_feature import V1Bi
222
222
  from lightning_sdk.lightning_cloud.openapi.models.v1_billing_period import V1BillingPeriod
223
223
  from lightning_sdk.lightning_cloud.openapi.models.v1_billing_subscription import V1BillingSubscription
224
224
  from lightning_sdk.lightning_cloud.openapi.models.v1_billing_tier import V1BillingTier
225
+ from lightning_sdk.lightning_cloud.openapi.models.v1_body import V1Body
225
226
  from lightning_sdk.lightning_cloud.openapi.models.v1_build_spec import V1BuildSpec
226
227
  from lightning_sdk.lightning_cloud.openapi.models.v1_cpu_system_metrics import V1CPUSystemMetrics
227
228
  from lightning_sdk.lightning_cloud.openapi.models.v1_cancel_cloud_space_instance_switch_response import V1CancelCloudSpaceInstanceSwitchResponse
@@ -339,7 +340,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningwork_respon
339
340
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_page_response import V1DeleteLitPageResponse
340
341
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_logger_artifact_response import V1DeleteLoggerArtifactResponse
341
342
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_managed_endpoint_response import V1DeleteManagedEndpointResponse
342
- from lightning_sdk.lightning_cloud.openapi.models.v1_delete_managed_model_response import V1DeleteManagedModelResponse
343
343
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_metrics_stream_response import V1DeleteMetricsStreamResponse
344
344
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_model_response import V1DeleteModelResponse
345
345
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_model_version_response import V1DeleteModelVersionResponse
@@ -365,6 +365,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_user_slurm_job_respo
365
365
  from lightning_sdk.lightning_cloud.openapi.models.v1_dependency_cache_state import V1DependencyCacheState
366
366
  from lightning_sdk.lightning_cloud.openapi.models.v1_dependency_file_info import V1DependencyFileInfo
367
367
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment import V1Deployment
368
+ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_api import V1DeploymentAPI
368
369
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_event import V1DeploymentEvent
369
370
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_event_type import V1DeploymentEventType
370
371
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_metrics import V1DeploymentMetrics
@@ -552,7 +553,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningwork_response
552
553
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_pages_response import V1ListLitPagesResponse
553
554
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_logger_artifact_response import V1ListLoggerArtifactResponse
554
555
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_managed_endpoints_response import V1ListManagedEndpointsResponse
555
- from lightning_sdk.lightning_cloud.openapi.models.v1_list_managed_models_response import V1ListManagedModelsResponse
556
556
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_memberships_response import V1ListMembershipsResponse
557
557
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_metrics_streams_response import V1ListMetricsStreamsResponse
558
558
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_model_versions_response import V1ListModelVersionsResponse
@@ -598,6 +598,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_magic_link_login_request im
598
598
  from lightning_sdk.lightning_cloud.openapi.models.v1_magic_link_login_response import V1MagicLinkLoginResponse
599
599
  from lightning_sdk.lightning_cloud.openapi.models.v1_managed_endpoint import V1ManagedEndpoint
600
600
  from lightning_sdk.lightning_cloud.openapi.models.v1_managed_model import V1ManagedModel
601
+ from lightning_sdk.lightning_cloud.openapi.models.v1_managed_model_abilities import V1ManagedModelAbilities
601
602
  from lightning_sdk.lightning_cloud.openapi.models.v1_membership import V1Membership
602
603
  from lightning_sdk.lightning_cloud.openapi.models.v1_message import V1Message
603
604
  from lightning_sdk.lightning_cloud.openapi.models.v1_message_author import V1MessageAuthor
@@ -653,6 +654,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_publish_cloud_space_respons
653
654
  from lightning_sdk.lightning_cloud.openapi.models.v1_published_cloud_space_response import V1PublishedCloudSpaceResponse
654
655
  from lightning_sdk.lightning_cloud.openapi.models.v1_purchase_capacity_block_response import V1PurchaseCapacityBlockResponse
655
656
  from lightning_sdk.lightning_cloud.openapi.models.v1_python_dependency_info import V1PythonDependencyInfo
657
+ from lightning_sdk.lightning_cloud.openapi.models.v1_query_param import V1QueryParam
656
658
  from lightning_sdk.lightning_cloud.openapi.models.v1_query_result import V1QueryResult
657
659
  from lightning_sdk.lightning_cloud.openapi.models.v1_query_result_row import V1QueryResultRow
658
660
  from lightning_sdk.lightning_cloud.openapi.models.v1_quest import V1Quest
@@ -671,6 +673,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_requ
671
673
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_response import V1RequestClusterAccessResponse
672
674
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_verification_code_response import V1RequestVerificationCodeResponse
673
675
  from lightning_sdk.lightning_cloud.openapi.models.v1_resource_tag import V1ResourceTag
676
+ from lightning_sdk.lightning_cloud.openapi.models.v1_resource_visibility import V1ResourceVisibility
674
677
  from lightning_sdk.lightning_cloud.openapi.models.v1_resources import V1Resources
675
678
  from lightning_sdk.lightning_cloud.openapi.models.v1_response_choice import V1ResponseChoice
676
679
  from lightning_sdk.lightning_cloud.openapi.models.v1_response_choice_delta import V1ResponseChoiceDelta
@@ -736,6 +739,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators
736
739
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_availability_request import V1UpdateClusterAvailabilityRequest
737
740
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_index_response import V1UpdateIndexResponse
738
741
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_page_response import V1UpdateLitPageResponse
742
+ from lightning_sdk.lightning_cloud.openapi.models.v1_update_model_visibility_response import V1UpdateModelVisibilityResponse
739
743
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_project_cluster_accelerators_response import V1UpdateProjectClusterAcceleratorsResponse
740
744
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_shared_metrics_stream_response import V1UpdateSharedMetricsStreamResponse
741
745
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_snowflake_query_response import V1UpdateSnowflakeQueryResponse