lightning-sdk 0.2.3__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 (110) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/ai_hub.py +10 -17
  3. lightning_sdk/api/ai_hub_api.py +20 -3
  4. lightning_sdk/api/studio_api.py +0 -8
  5. lightning_sdk/cli/serve.py +139 -22
  6. lightning_sdk/deployment/deployment.py +32 -4
  7. lightning_sdk/lightning_cloud/openapi/__init__.py +49 -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 +10 -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 +5 -1
  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 +45 -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 +227 -0
  24. lightning_sdk/lightning_cloud/openapi/models/externalv1_cloud_space_instance_status.py +69 -69
  25. lightning_sdk/lightning_cloud/openapi/models/litdatasets_dataset_id_body.py +149 -0
  26. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +27 -1
  27. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body.py +69 -17
  28. lightning_sdk/lightning_cloud/openapi/models/pipelinetemplates_id_body.py +331 -0
  29. lightning_sdk/lightning_cloud/openapi/models/project_id_litdatasets_body.py +227 -0
  30. lightning_sdk/lightning_cloud/openapi/models/project_id_pipelines_body.py +17 -17
  31. lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py +201 -0
  32. lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py +383 -0
  33. lightning_sdk/lightning_cloud/openapi/models/slurm_jobs_body.py +15 -15
  34. lightning_sdk/lightning_cloud/openapi/models/upload_id_complete_body1.py +149 -0
  35. lightning_sdk/lightning_cloud/openapi/models/upload_id_parts_body1.py +149 -0
  36. lightning_sdk/lightning_cloud/openapi/models/v1_agent_job.py +124 -20
  37. lightning_sdk/lightning_cloud/openapi/models/v1_assistant_model_status.py +2 -0
  38. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -0
  39. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +27 -1
  40. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template.py +253 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_complete_lit_dataset_multi_part_upload_response.py +97 -0
  43. lightning_sdk/lightning_cloud/openapi/models/v1_complete_lit_dataset_upload_response.py +97 -0
  44. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +149 -0
  45. lightning_sdk/lightning_cloud/openapi/models/v1_create_lit_dataset_multi_part_upload_response.py +123 -0
  46. lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py +27 -1
  47. lightning_sdk/lightning_cloud/openapi/models/v1_create_pipeline_template_request.py +383 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_create_project_request.py +27 -1
  49. lightning_sdk/lightning_cloud/openapi/models/{v1_pipeline_schedule.py → v1_delete_cloud_space_environment_template_response.py} +32 -32
  50. lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_dataset_response.py +97 -0
  51. lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_dataset_version_response.py +97 -0
  52. lightning_sdk/lightning_cloud/openapi/models/v1_delete_schedule_response.py +175 -0
  53. lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_size_response.py +79 -1
  54. lightning_sdk/lightning_cloud/openapi/models/v1_get_lit_dataset_file_upload_urls_response.py +123 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_get_lit_dataset_files_url_response.py +149 -0
  56. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
  57. lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.py +79 -27
  58. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +27 -1
  59. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloud_space_environment_templates_response.py +123 -0
  60. lightning_sdk/lightning_cloud/openapi/models/v1_list_lit_dataset_versions_response.py +123 -0
  61. lightning_sdk/lightning_cloud/openapi/models/v1_list_lit_datasets_response.py +123 -0
  62. lightning_sdk/lightning_cloud/openapi/models/v1_list_schedules_response.py +123 -0
  63. lightning_sdk/lightning_cloud/openapi/models/v1_list_slurm_cluster_users_response.py +123 -0
  64. lightning_sdk/lightning_cloud/openapi/models/v1_lit_dataset.py +539 -0
  65. lightning_sdk/lightning_cloud/openapi/models/v1_lit_dataset_file.py +175 -0
  66. lightning_sdk/lightning_cloud/openapi/models/v1_lit_dataset_version_archive.py +435 -0
  67. lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_project.py +27 -1
  68. lightning_sdk/lightning_cloud/openapi/models/v1_lit_repository.py +27 -1
  69. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_request.py +27 -1
  70. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +107 -3
  71. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +53 -1
  72. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py +69 -17
  73. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter.py +435 -0
  74. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter_placement.py +149 -0
  75. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter_placement_type.py +106 -0
  76. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter_type.py +106 -0
  77. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_template.py +513 -0
  78. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_template_visibility_type.py +105 -0
  79. lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py +27 -1
  80. lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py +435 -0
  81. lightning_sdk/lightning_cloud/openapi/models/v1_schedule_resource_type.py +103 -0
  82. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_cluster_user.py +227 -0
  83. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_job.py +58 -6
  84. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_node.py +31 -291
  85. lightning_sdk/lightning_cloud/openapi/models/v1_update_lit_dataset_visibility_response.py +123 -0
  86. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +27 -1
  87. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +157 -105
  88. lightning_sdk/lightning_cloud/openapi/models/v1_validate_managed_endpoint_response.py +27 -1
  89. lightning_sdk/lightning_cloud/openapi/models/v1_voltage_park_direct_v1.py +203 -0
  90. lightning_sdk/lightning_cloud/openapi/models/version_default_body.py +29 -29
  91. lightning_sdk/lightning_cloud/openapi/models/version_default_body1.py +149 -0
  92. lightning_sdk/lightning_cloud/openapi/models/version_uploads_body1.py +123 -0
  93. lightning_sdk/lightning_cloud/openapi/models/versions_version_body1.py +123 -0
  94. lightning_sdk/lightning_cloud/rest_client.py +2 -0
  95. lightning_sdk/lit_container.py +8 -1
  96. lightning_sdk/mmt/mmt.py +1 -18
  97. lightning_sdk/mmt/v1.py +1 -28
  98. lightning_sdk/models.py +15 -6
  99. lightning_sdk/pipeline/pipeline.py +2 -2
  100. lightning_sdk/pipeline/types.py +28 -2
  101. lightning_sdk/pipeline/utils.py +1 -1
  102. lightning_sdk/plugin.py +0 -6
  103. lightning_sdk/serve.py +55 -22
  104. lightning_sdk/utils/resolve.py +1 -0
  105. {lightning_sdk-0.2.3.dist-info → lightning_sdk-0.2.5.dist-info}/METADATA +1 -1
  106. {lightning_sdk-0.2.3.dist-info → lightning_sdk-0.2.5.dist-info}/RECORD +110 -62
  107. {lightning_sdk-0.2.3.dist-info → lightning_sdk-0.2.5.dist-info}/WHEEL +1 -1
  108. {lightning_sdk-0.2.3.dist-info → lightning_sdk-0.2.5.dist-info}/LICENSE +0 -0
  109. {lightning_sdk-0.2.3.dist-info → lightning_sdk-0.2.5.dist-info}/entry_points.txt +0 -0
  110. {lightning_sdk-0.2.3.dist-info → lightning_sdk-0.2.5.dist-info}/top_level.txt +0 -0
@@ -47,16 +47,19 @@ class V1UserFeatures(object):
47
47
  'auto_fast_load': 'bool',
48
48
  'auto_join_orgs': 'bool',
49
49
  'b2c_experience': 'bool',
50
+ 'byoc_litcr': 'bool',
50
51
  'cap_add': 'list[str]',
51
52
  'cap_drop': 'list[str]',
52
53
  'capacity_reservation_byoc': 'bool',
53
54
  'capacity_reservation_dry_run': 'bool',
54
55
  'chat_models': 'bool',
56
+ 'cloud_space_environment_templates': 'bool',
55
57
  'code_tab': 'bool',
56
58
  'collab_screen_sharing': 'bool',
57
59
  'cost_attribution_settings': 'bool',
58
60
  'custom_app_domain': 'bool',
59
61
  'custom_instance_types': 'bool',
62
+ 'datasets': 'bool',
60
63
  'default_one_cluster': 'bool',
61
64
  'deployment_alerts': 'bool',
62
65
  'deployment_persistent_disk': 'bool',
@@ -66,6 +69,7 @@ class V1UserFeatures(object):
66
69
  'drive_v2': 'bool',
67
70
  'enable_crypto_crackdown': 'bool',
68
71
  'enable_storage_limits': 'bool',
72
+ 'enterprise_compute_admin': 'bool',
69
73
  'fair_share': 'bool',
70
74
  'featured_studios_admin': 'bool',
71
75
  'filestore': 'bool',
@@ -75,12 +79,11 @@ class V1UserFeatures(object):
75
79
  'lambda_labs': 'bool',
76
80
  'landing_studios': 'bool',
77
81
  'lit_logger': 'bool',
82
+ 'marketplace': 'bool',
78
83
  'mmt_fault_tolerance': 'bool',
79
84
  'mmt_strategy_selector': 'bool',
80
- 'mmt_v2': 'bool',
81
85
  'multicloud_saas': 'bool',
82
86
  'multiple_studio_versions': 'bool',
83
- 'open_api_in_studio': 'bool',
84
87
  'org_level_member_permissions': 'bool',
85
88
  'pipelines': 'bool',
86
89
  'plugin_distributed': 'bool',
@@ -90,7 +93,6 @@ class V1UserFeatures(object):
90
93
  'plugin_langflow': 'bool',
91
94
  'plugin_lightning_apps': 'bool',
92
95
  'plugin_lightning_apps_distributed': 'bool',
93
- 'plugin_mage_ai': 'bool',
94
96
  'plugin_milvus': 'bool',
95
97
  'plugin_python_profiler': 'bool',
96
98
  'plugin_react': 'bool',
@@ -100,6 +102,7 @@ class V1UserFeatures(object):
100
102
  'pricing_updates': 'bool',
101
103
  'product_generator': 'bool',
102
104
  'project_selector': 'bool',
105
+ 'publish_pipelines': 'bool',
103
106
  'restartable_jobs': 'bool',
104
107
  'runnable_public_studio_page': 'bool',
105
108
  'security_docs': 'bool',
@@ -107,7 +110,6 @@ class V1UserFeatures(object):
107
110
  'slurm': 'bool',
108
111
  'slurm_machine_selector': 'bool',
109
112
  'snapshotter_service': 'bool',
110
- 'snowflake_connection': 'bool',
111
113
  'stop_ide_container_on_shutdown': 'bool',
112
114
  'studio_config': 'bool',
113
115
  'studio_on_stop': 'bool',
@@ -125,16 +127,19 @@ class V1UserFeatures(object):
125
127
  'auto_fast_load': 'autoFastLoad',
126
128
  'auto_join_orgs': 'autoJoinOrgs',
127
129
  'b2c_experience': 'b2cExperience',
130
+ 'byoc_litcr': 'byocLitcr',
128
131
  'cap_add': 'capAdd',
129
132
  'cap_drop': 'capDrop',
130
133
  'capacity_reservation_byoc': 'capacityReservationByoc',
131
134
  'capacity_reservation_dry_run': 'capacityReservationDryRun',
132
135
  'chat_models': 'chatModels',
136
+ 'cloud_space_environment_templates': 'cloudSpaceEnvironmentTemplates',
133
137
  'code_tab': 'codeTab',
134
138
  'collab_screen_sharing': 'collabScreenSharing',
135
139
  'cost_attribution_settings': 'costAttributionSettings',
136
140
  'custom_app_domain': 'customAppDomain',
137
141
  'custom_instance_types': 'customInstanceTypes',
142
+ 'datasets': 'datasets',
138
143
  'default_one_cluster': 'defaultOneCluster',
139
144
  'deployment_alerts': 'deploymentAlerts',
140
145
  'deployment_persistent_disk': 'deploymentPersistentDisk',
@@ -144,6 +149,7 @@ class V1UserFeatures(object):
144
149
  'drive_v2': 'driveV2',
145
150
  'enable_crypto_crackdown': 'enableCryptoCrackdown',
146
151
  'enable_storage_limits': 'enableStorageLimits',
152
+ 'enterprise_compute_admin': 'enterpriseComputeAdmin',
147
153
  'fair_share': 'fairShare',
148
154
  'featured_studios_admin': 'featuredStudiosAdmin',
149
155
  'filestore': 'filestore',
@@ -153,12 +159,11 @@ class V1UserFeatures(object):
153
159
  'lambda_labs': 'lambdaLabs',
154
160
  'landing_studios': 'landingStudios',
155
161
  'lit_logger': 'litLogger',
162
+ 'marketplace': 'marketplace',
156
163
  'mmt_fault_tolerance': 'mmtFaultTolerance',
157
164
  'mmt_strategy_selector': 'mmtStrategySelector',
158
- 'mmt_v2': 'mmtV2',
159
165
  'multicloud_saas': 'multicloudSaas',
160
166
  'multiple_studio_versions': 'multipleStudioVersions',
161
- 'open_api_in_studio': 'openApiInStudio',
162
167
  'org_level_member_permissions': 'orgLevelMemberPermissions',
163
168
  'pipelines': 'pipelines',
164
169
  'plugin_distributed': 'pluginDistributed',
@@ -168,7 +173,6 @@ class V1UserFeatures(object):
168
173
  'plugin_langflow': 'pluginLangflow',
169
174
  'plugin_lightning_apps': 'pluginLightningApps',
170
175
  'plugin_lightning_apps_distributed': 'pluginLightningAppsDistributed',
171
- 'plugin_mage_ai': 'pluginMageAi',
172
176
  'plugin_milvus': 'pluginMilvus',
173
177
  'plugin_python_profiler': 'pluginPythonProfiler',
174
178
  'plugin_react': 'pluginReact',
@@ -178,6 +182,7 @@ class V1UserFeatures(object):
178
182
  'pricing_updates': 'pricingUpdates',
179
183
  'product_generator': 'productGenerator',
180
184
  'project_selector': 'projectSelector',
185
+ 'publish_pipelines': 'publishPipelines',
181
186
  'restartable_jobs': 'restartableJobs',
182
187
  'runnable_public_studio_page': 'runnablePublicStudioPage',
183
188
  'security_docs': 'securityDocs',
@@ -185,7 +190,6 @@ class V1UserFeatures(object):
185
190
  'slurm': 'slurm',
186
191
  'slurm_machine_selector': 'slurmMachineSelector',
187
192
  'snapshotter_service': 'snapshotterService',
188
- 'snowflake_connection': 'snowflakeConnection',
189
193
  'stop_ide_container_on_shutdown': 'stopIdeContainerOnShutdown',
190
194
  'studio_config': 'studioConfig',
191
195
  'studio_on_stop': 'studioOnStop',
@@ -196,7 +200,7 @@ class V1UserFeatures(object):
196
200
  'vultr': 'vultr'
197
201
  }
198
202
 
199
- def __init__(self, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, b2c_experience: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, chat_models: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, custom_instance_types: 'bool' =None, default_one_cluster: 'bool' =None, deployment_alerts: 'bool' =None, deployment_persistent_disk: 'bool' =None, deployment_reservations: 'bool' =None, dgx_cloud: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, enable_crypto_crackdown: 'bool' =None, enable_storage_limits: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, filestore: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, jobs_v2: 'bool' =None, lambda_labs: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, mmt_v2: 'bool' =None, multicloud_saas: 'bool' =None, multiple_studio_versions: 'bool' =None, open_api_in_studio: 'bool' =None, org_level_member_permissions: 'bool' =None, pipelines: 'bool' =None, plugin_distributed: 'bool' =None, plugin_fiftyone: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_lightning_apps: 'bool' =None, plugin_lightning_apps_distributed: 'bool' =None, plugin_mage_ai: 'bool' =None, plugin_milvus: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_react: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, plugin_weviate: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, project_selector: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, show_dev_admin: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, snapshotter_service: 'bool' =None, snowflake_connection: 'bool' =None, stop_ide_container_on_shutdown: 'bool' =None, studio_config: 'bool' =None, studio_on_stop: 'bool' =None, studio_version_visibility: 'bool' =None, teamspace_storage_tab: 'bool' =None, trainium2: 'bool' =None, use_rclone_mounts_only: 'bool' =None, vultr: 'bool' =None): # noqa: E501
203
+ def __init__(self, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, b2c_experience: 'bool' =None, byoc_litcr: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, chat_models: 'bool' =None, cloud_space_environment_templates: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, custom_instance_types: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_alerts: 'bool' =None, deployment_persistent_disk: 'bool' =None, deployment_reservations: 'bool' =None, dgx_cloud: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, enable_crypto_crackdown: 'bool' =None, enable_storage_limits: 'bool' =None, enterprise_compute_admin: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, filestore: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, jobs_v2: 'bool' =None, lambda_labs: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, multicloud_saas: 'bool' =None, multiple_studio_versions: 'bool' =None, org_level_member_permissions: 'bool' =None, pipelines: 'bool' =None, plugin_distributed: 'bool' =None, plugin_fiftyone: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_lightning_apps: 'bool' =None, plugin_lightning_apps_distributed: 'bool' =None, plugin_milvus: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_react: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, plugin_weviate: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, show_dev_admin: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, snapshotter_service: 'bool' =None, stop_ide_container_on_shutdown: 'bool' =None, studio_config: 'bool' =None, studio_on_stop: 'bool' =None, studio_version_visibility: 'bool' =None, teamspace_storage_tab: 'bool' =None, trainium2: 'bool' =None, use_rclone_mounts_only: 'bool' =None, vultr: 'bool' =None): # noqa: E501
200
204
  """V1UserFeatures - a model defined in Swagger""" # noqa: E501
201
205
  self._affiliate_links = None
202
206
  self._agents_v2 = None
@@ -204,16 +208,19 @@ class V1UserFeatures(object):
204
208
  self._auto_fast_load = None
205
209
  self._auto_join_orgs = None
206
210
  self._b2c_experience = None
211
+ self._byoc_litcr = None
207
212
  self._cap_add = None
208
213
  self._cap_drop = None
209
214
  self._capacity_reservation_byoc = None
210
215
  self._capacity_reservation_dry_run = None
211
216
  self._chat_models = None
217
+ self._cloud_space_environment_templates = None
212
218
  self._code_tab = None
213
219
  self._collab_screen_sharing = None
214
220
  self._cost_attribution_settings = None
215
221
  self._custom_app_domain = None
216
222
  self._custom_instance_types = None
223
+ self._datasets = None
217
224
  self._default_one_cluster = None
218
225
  self._deployment_alerts = None
219
226
  self._deployment_persistent_disk = None
@@ -223,6 +230,7 @@ class V1UserFeatures(object):
223
230
  self._drive_v2 = None
224
231
  self._enable_crypto_crackdown = None
225
232
  self._enable_storage_limits = None
233
+ self._enterprise_compute_admin = None
226
234
  self._fair_share = None
227
235
  self._featured_studios_admin = None
228
236
  self._filestore = None
@@ -232,12 +240,11 @@ class V1UserFeatures(object):
232
240
  self._lambda_labs = None
233
241
  self._landing_studios = None
234
242
  self._lit_logger = None
243
+ self._marketplace = None
235
244
  self._mmt_fault_tolerance = None
236
245
  self._mmt_strategy_selector = None
237
- self._mmt_v2 = None
238
246
  self._multicloud_saas = None
239
247
  self._multiple_studio_versions = None
240
- self._open_api_in_studio = None
241
248
  self._org_level_member_permissions = None
242
249
  self._pipelines = None
243
250
  self._plugin_distributed = None
@@ -247,7 +254,6 @@ class V1UserFeatures(object):
247
254
  self._plugin_langflow = None
248
255
  self._plugin_lightning_apps = None
249
256
  self._plugin_lightning_apps_distributed = None
250
- self._plugin_mage_ai = None
251
257
  self._plugin_milvus = None
252
258
  self._plugin_python_profiler = None
253
259
  self._plugin_react = None
@@ -257,6 +263,7 @@ class V1UserFeatures(object):
257
263
  self._pricing_updates = None
258
264
  self._product_generator = None
259
265
  self._project_selector = None
266
+ self._publish_pipelines = None
260
267
  self._restartable_jobs = None
261
268
  self._runnable_public_studio_page = None
262
269
  self._security_docs = None
@@ -264,7 +271,6 @@ class V1UserFeatures(object):
264
271
  self._slurm = None
265
272
  self._slurm_machine_selector = None
266
273
  self._snapshotter_service = None
267
- self._snowflake_connection = None
268
274
  self._stop_ide_container_on_shutdown = None
269
275
  self._studio_config = None
270
276
  self._studio_on_stop = None
@@ -286,6 +292,8 @@ class V1UserFeatures(object):
286
292
  self.auto_join_orgs = auto_join_orgs
287
293
  if b2c_experience is not None:
288
294
  self.b2c_experience = b2c_experience
295
+ if byoc_litcr is not None:
296
+ self.byoc_litcr = byoc_litcr
289
297
  if cap_add is not None:
290
298
  self.cap_add = cap_add
291
299
  if cap_drop is not None:
@@ -296,6 +304,8 @@ class V1UserFeatures(object):
296
304
  self.capacity_reservation_dry_run = capacity_reservation_dry_run
297
305
  if chat_models is not None:
298
306
  self.chat_models = chat_models
307
+ if cloud_space_environment_templates is not None:
308
+ self.cloud_space_environment_templates = cloud_space_environment_templates
299
309
  if code_tab is not None:
300
310
  self.code_tab = code_tab
301
311
  if collab_screen_sharing is not None:
@@ -306,6 +316,8 @@ class V1UserFeatures(object):
306
316
  self.custom_app_domain = custom_app_domain
307
317
  if custom_instance_types is not None:
308
318
  self.custom_instance_types = custom_instance_types
319
+ if datasets is not None:
320
+ self.datasets = datasets
309
321
  if default_one_cluster is not None:
310
322
  self.default_one_cluster = default_one_cluster
311
323
  if deployment_alerts is not None:
@@ -324,6 +336,8 @@ class V1UserFeatures(object):
324
336
  self.enable_crypto_crackdown = enable_crypto_crackdown
325
337
  if enable_storage_limits is not None:
326
338
  self.enable_storage_limits = enable_storage_limits
339
+ if enterprise_compute_admin is not None:
340
+ self.enterprise_compute_admin = enterprise_compute_admin
327
341
  if fair_share is not None:
328
342
  self.fair_share = fair_share
329
343
  if featured_studios_admin is not None:
@@ -342,18 +356,16 @@ class V1UserFeatures(object):
342
356
  self.landing_studios = landing_studios
343
357
  if lit_logger is not None:
344
358
  self.lit_logger = lit_logger
359
+ if marketplace is not None:
360
+ self.marketplace = marketplace
345
361
  if mmt_fault_tolerance is not None:
346
362
  self.mmt_fault_tolerance = mmt_fault_tolerance
347
363
  if mmt_strategy_selector is not None:
348
364
  self.mmt_strategy_selector = mmt_strategy_selector
349
- if mmt_v2 is not None:
350
- self.mmt_v2 = mmt_v2
351
365
  if multicloud_saas is not None:
352
366
  self.multicloud_saas = multicloud_saas
353
367
  if multiple_studio_versions is not None:
354
368
  self.multiple_studio_versions = multiple_studio_versions
355
- if open_api_in_studio is not None:
356
- self.open_api_in_studio = open_api_in_studio
357
369
  if org_level_member_permissions is not None:
358
370
  self.org_level_member_permissions = org_level_member_permissions
359
371
  if pipelines is not None:
@@ -372,8 +384,6 @@ class V1UserFeatures(object):
372
384
  self.plugin_lightning_apps = plugin_lightning_apps
373
385
  if plugin_lightning_apps_distributed is not None:
374
386
  self.plugin_lightning_apps_distributed = plugin_lightning_apps_distributed
375
- if plugin_mage_ai is not None:
376
- self.plugin_mage_ai = plugin_mage_ai
377
387
  if plugin_milvus is not None:
378
388
  self.plugin_milvus = plugin_milvus
379
389
  if plugin_python_profiler is not None:
@@ -392,6 +402,8 @@ class V1UserFeatures(object):
392
402
  self.product_generator = product_generator
393
403
  if project_selector is not None:
394
404
  self.project_selector = project_selector
405
+ if publish_pipelines is not None:
406
+ self.publish_pipelines = publish_pipelines
395
407
  if restartable_jobs is not None:
396
408
  self.restartable_jobs = restartable_jobs
397
409
  if runnable_public_studio_page is not None:
@@ -406,8 +418,6 @@ class V1UserFeatures(object):
406
418
  self.slurm_machine_selector = slurm_machine_selector
407
419
  if snapshotter_service is not None:
408
420
  self.snapshotter_service = snapshotter_service
409
- if snowflake_connection is not None:
410
- self.snowflake_connection = snowflake_connection
411
421
  if stop_ide_container_on_shutdown is not None:
412
422
  self.stop_ide_container_on_shutdown = stop_ide_container_on_shutdown
413
423
  if studio_config is not None:
@@ -551,6 +561,27 @@ class V1UserFeatures(object):
551
561
 
552
562
  self._b2c_experience = b2c_experience
553
563
 
564
+ @property
565
+ def byoc_litcr(self) -> 'bool':
566
+ """Gets the byoc_litcr of this V1UserFeatures. # noqa: E501
567
+
568
+
569
+ :return: The byoc_litcr of this V1UserFeatures. # noqa: E501
570
+ :rtype: bool
571
+ """
572
+ return self._byoc_litcr
573
+
574
+ @byoc_litcr.setter
575
+ def byoc_litcr(self, byoc_litcr: 'bool'):
576
+ """Sets the byoc_litcr of this V1UserFeatures.
577
+
578
+
579
+ :param byoc_litcr: The byoc_litcr of this V1UserFeatures. # noqa: E501
580
+ :type: bool
581
+ """
582
+
583
+ self._byoc_litcr = byoc_litcr
584
+
554
585
  @property
555
586
  def cap_add(self) -> 'list[str]':
556
587
  """Gets the cap_add of this V1UserFeatures. # noqa: E501
@@ -656,6 +687,27 @@ class V1UserFeatures(object):
656
687
 
657
688
  self._chat_models = chat_models
658
689
 
690
+ @property
691
+ def cloud_space_environment_templates(self) -> 'bool':
692
+ """Gets the cloud_space_environment_templates of this V1UserFeatures. # noqa: E501
693
+
694
+
695
+ :return: The cloud_space_environment_templates of this V1UserFeatures. # noqa: E501
696
+ :rtype: bool
697
+ """
698
+ return self._cloud_space_environment_templates
699
+
700
+ @cloud_space_environment_templates.setter
701
+ def cloud_space_environment_templates(self, cloud_space_environment_templates: 'bool'):
702
+ """Sets the cloud_space_environment_templates of this V1UserFeatures.
703
+
704
+
705
+ :param cloud_space_environment_templates: The cloud_space_environment_templates of this V1UserFeatures. # noqa: E501
706
+ :type: bool
707
+ """
708
+
709
+ self._cloud_space_environment_templates = cloud_space_environment_templates
710
+
659
711
  @property
660
712
  def code_tab(self) -> 'bool':
661
713
  """Gets the code_tab of this V1UserFeatures. # noqa: E501
@@ -761,6 +813,27 @@ class V1UserFeatures(object):
761
813
 
762
814
  self._custom_instance_types = custom_instance_types
763
815
 
816
+ @property
817
+ def datasets(self) -> 'bool':
818
+ """Gets the datasets of this V1UserFeatures. # noqa: E501
819
+
820
+
821
+ :return: The datasets of this V1UserFeatures. # noqa: E501
822
+ :rtype: bool
823
+ """
824
+ return self._datasets
825
+
826
+ @datasets.setter
827
+ def datasets(self, datasets: 'bool'):
828
+ """Sets the datasets of this V1UserFeatures.
829
+
830
+
831
+ :param datasets: The datasets of this V1UserFeatures. # noqa: E501
832
+ :type: bool
833
+ """
834
+
835
+ self._datasets = datasets
836
+
764
837
  @property
765
838
  def default_one_cluster(self) -> 'bool':
766
839
  """Gets the default_one_cluster of this V1UserFeatures. # noqa: E501
@@ -950,6 +1023,27 @@ class V1UserFeatures(object):
950
1023
 
951
1024
  self._enable_storage_limits = enable_storage_limits
952
1025
 
1026
+ @property
1027
+ def enterprise_compute_admin(self) -> 'bool':
1028
+ """Gets the enterprise_compute_admin of this V1UserFeatures. # noqa: E501
1029
+
1030
+
1031
+ :return: The enterprise_compute_admin of this V1UserFeatures. # noqa: E501
1032
+ :rtype: bool
1033
+ """
1034
+ return self._enterprise_compute_admin
1035
+
1036
+ @enterprise_compute_admin.setter
1037
+ def enterprise_compute_admin(self, enterprise_compute_admin: 'bool'):
1038
+ """Sets the enterprise_compute_admin of this V1UserFeatures.
1039
+
1040
+
1041
+ :param enterprise_compute_admin: The enterprise_compute_admin of this V1UserFeatures. # noqa: E501
1042
+ :type: bool
1043
+ """
1044
+
1045
+ self._enterprise_compute_admin = enterprise_compute_admin
1046
+
953
1047
  @property
954
1048
  def fair_share(self) -> 'bool':
955
1049
  """Gets the fair_share of this V1UserFeatures. # noqa: E501
@@ -1139,6 +1233,27 @@ class V1UserFeatures(object):
1139
1233
 
1140
1234
  self._lit_logger = lit_logger
1141
1235
 
1236
+ @property
1237
+ def marketplace(self) -> 'bool':
1238
+ """Gets the marketplace of this V1UserFeatures. # noqa: E501
1239
+
1240
+
1241
+ :return: The marketplace of this V1UserFeatures. # noqa: E501
1242
+ :rtype: bool
1243
+ """
1244
+ return self._marketplace
1245
+
1246
+ @marketplace.setter
1247
+ def marketplace(self, marketplace: 'bool'):
1248
+ """Sets the marketplace of this V1UserFeatures.
1249
+
1250
+
1251
+ :param marketplace: The marketplace of this V1UserFeatures. # noqa: E501
1252
+ :type: bool
1253
+ """
1254
+
1255
+ self._marketplace = marketplace
1256
+
1142
1257
  @property
1143
1258
  def mmt_fault_tolerance(self) -> 'bool':
1144
1259
  """Gets the mmt_fault_tolerance of this V1UserFeatures. # noqa: E501
@@ -1181,27 +1296,6 @@ class V1UserFeatures(object):
1181
1296
 
1182
1297
  self._mmt_strategy_selector = mmt_strategy_selector
1183
1298
 
1184
- @property
1185
- def mmt_v2(self) -> 'bool':
1186
- """Gets the mmt_v2 of this V1UserFeatures. # noqa: E501
1187
-
1188
-
1189
- :return: The mmt_v2 of this V1UserFeatures. # noqa: E501
1190
- :rtype: bool
1191
- """
1192
- return self._mmt_v2
1193
-
1194
- @mmt_v2.setter
1195
- def mmt_v2(self, mmt_v2: 'bool'):
1196
- """Sets the mmt_v2 of this V1UserFeatures.
1197
-
1198
-
1199
- :param mmt_v2: The mmt_v2 of this V1UserFeatures. # noqa: E501
1200
- :type: bool
1201
- """
1202
-
1203
- self._mmt_v2 = mmt_v2
1204
-
1205
1299
  @property
1206
1300
  def multicloud_saas(self) -> 'bool':
1207
1301
  """Gets the multicloud_saas of this V1UserFeatures. # noqa: E501
@@ -1244,27 +1338,6 @@ class V1UserFeatures(object):
1244
1338
 
1245
1339
  self._multiple_studio_versions = multiple_studio_versions
1246
1340
 
1247
- @property
1248
- def open_api_in_studio(self) -> 'bool':
1249
- """Gets the open_api_in_studio of this V1UserFeatures. # noqa: E501
1250
-
1251
-
1252
- :return: The open_api_in_studio of this V1UserFeatures. # noqa: E501
1253
- :rtype: bool
1254
- """
1255
- return self._open_api_in_studio
1256
-
1257
- @open_api_in_studio.setter
1258
- def open_api_in_studio(self, open_api_in_studio: 'bool'):
1259
- """Sets the open_api_in_studio of this V1UserFeatures.
1260
-
1261
-
1262
- :param open_api_in_studio: The open_api_in_studio of this V1UserFeatures. # noqa: E501
1263
- :type: bool
1264
- """
1265
-
1266
- self._open_api_in_studio = open_api_in_studio
1267
-
1268
1341
  @property
1269
1342
  def org_level_member_permissions(self) -> 'bool':
1270
1343
  """Gets the org_level_member_permissions of this V1UserFeatures. # noqa: E501
@@ -1454,27 +1527,6 @@ class V1UserFeatures(object):
1454
1527
 
1455
1528
  self._plugin_lightning_apps_distributed = plugin_lightning_apps_distributed
1456
1529
 
1457
- @property
1458
- def plugin_mage_ai(self) -> 'bool':
1459
- """Gets the plugin_mage_ai of this V1UserFeatures. # noqa: E501
1460
-
1461
-
1462
- :return: The plugin_mage_ai of this V1UserFeatures. # noqa: E501
1463
- :rtype: bool
1464
- """
1465
- return self._plugin_mage_ai
1466
-
1467
- @plugin_mage_ai.setter
1468
- def plugin_mage_ai(self, plugin_mage_ai: 'bool'):
1469
- """Sets the plugin_mage_ai of this V1UserFeatures.
1470
-
1471
-
1472
- :param plugin_mage_ai: The plugin_mage_ai of this V1UserFeatures. # noqa: E501
1473
- :type: bool
1474
- """
1475
-
1476
- self._plugin_mage_ai = plugin_mage_ai
1477
-
1478
1530
  @property
1479
1531
  def plugin_milvus(self) -> 'bool':
1480
1532
  """Gets the plugin_milvus of this V1UserFeatures. # noqa: E501
@@ -1664,6 +1716,27 @@ class V1UserFeatures(object):
1664
1716
 
1665
1717
  self._project_selector = project_selector
1666
1718
 
1719
+ @property
1720
+ def publish_pipelines(self) -> 'bool':
1721
+ """Gets the publish_pipelines of this V1UserFeatures. # noqa: E501
1722
+
1723
+
1724
+ :return: The publish_pipelines of this V1UserFeatures. # noqa: E501
1725
+ :rtype: bool
1726
+ """
1727
+ return self._publish_pipelines
1728
+
1729
+ @publish_pipelines.setter
1730
+ def publish_pipelines(self, publish_pipelines: 'bool'):
1731
+ """Sets the publish_pipelines of this V1UserFeatures.
1732
+
1733
+
1734
+ :param publish_pipelines: The publish_pipelines of this V1UserFeatures. # noqa: E501
1735
+ :type: bool
1736
+ """
1737
+
1738
+ self._publish_pipelines = publish_pipelines
1739
+
1667
1740
  @property
1668
1741
  def restartable_jobs(self) -> 'bool':
1669
1742
  """Gets the restartable_jobs of this V1UserFeatures. # noqa: E501
@@ -1811,27 +1884,6 @@ class V1UserFeatures(object):
1811
1884
 
1812
1885
  self._snapshotter_service = snapshotter_service
1813
1886
 
1814
- @property
1815
- def snowflake_connection(self) -> 'bool':
1816
- """Gets the snowflake_connection of this V1UserFeatures. # noqa: E501
1817
-
1818
-
1819
- :return: The snowflake_connection of this V1UserFeatures. # noqa: E501
1820
- :rtype: bool
1821
- """
1822
- return self._snowflake_connection
1823
-
1824
- @snowflake_connection.setter
1825
- def snowflake_connection(self, snowflake_connection: 'bool'):
1826
- """Sets the snowflake_connection of this V1UserFeatures.
1827
-
1828
-
1829
- :param snowflake_connection: The snowflake_connection of this V1UserFeatures. # noqa: E501
1830
- :type: bool
1831
- """
1832
-
1833
- self._snowflake_connection = snowflake_connection
1834
-
1835
1887
  @property
1836
1888
  def stop_ide_container_on_shutdown(self) -> 'bool':
1837
1889
  """Gets the stop_ide_container_on_shutdown of this V1UserFeatures. # noqa: E501
@@ -41,20 +41,46 @@ class V1ValidateManagedEndpointResponse(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
+ 'lai_resource_status': 'V1AssistantModelStatus',
44
45
  'valid': 'bool'
45
46
  }
46
47
 
47
48
  attribute_map = {
49
+ 'lai_resource_status': 'laiResourceStatus',
48
50
  'valid': 'valid'
49
51
  }
50
52
 
51
- def __init__(self, valid: 'bool' =None): # noqa: E501
53
+ def __init__(self, lai_resource_status: 'V1AssistantModelStatus' =None, valid: 'bool' =None): # noqa: E501
52
54
  """V1ValidateManagedEndpointResponse - a model defined in Swagger""" # noqa: E501
55
+ self._lai_resource_status = None
53
56
  self._valid = None
54
57
  self.discriminator = None
58
+ if lai_resource_status is not None:
59
+ self.lai_resource_status = lai_resource_status
55
60
  if valid is not None:
56
61
  self.valid = valid
57
62
 
63
+ @property
64
+ def lai_resource_status(self) -> 'V1AssistantModelStatus':
65
+ """Gets the lai_resource_status of this V1ValidateManagedEndpointResponse. # noqa: E501
66
+
67
+
68
+ :return: The lai_resource_status of this V1ValidateManagedEndpointResponse. # noqa: E501
69
+ :rtype: V1AssistantModelStatus
70
+ """
71
+ return self._lai_resource_status
72
+
73
+ @lai_resource_status.setter
74
+ def lai_resource_status(self, lai_resource_status: 'V1AssistantModelStatus'):
75
+ """Sets the lai_resource_status of this V1ValidateManagedEndpointResponse.
76
+
77
+
78
+ :param lai_resource_status: The lai_resource_status of this V1ValidateManagedEndpointResponse. # noqa: E501
79
+ :type: V1AssistantModelStatus
80
+ """
81
+
82
+ self._lai_resource_status = lai_resource_status
83
+
58
84
  @property
59
85
  def valid(self) -> 'bool':
60
86
  """Gets the valid of this V1ValidateManagedEndpointResponse. # noqa: E501