lightning-sdk 0.2.21rc1__py3-none-any.whl → 0.2.23__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 (70) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/license_api.py +2 -2
  3. lightning_sdk/api/llm_api.py +69 -27
  4. lightning_sdk/api/pipeline_api.py +49 -9
  5. lightning_sdk/api/studio_api.py +2 -0
  6. lightning_sdk/cli/configure.py +34 -27
  7. lightning_sdk/cli/connect.py +2 -2
  8. lightning_sdk/cli/download.py +38 -2
  9. lightning_sdk/cli/entrypoint.py +15 -13
  10. lightning_sdk/cli/start.py +5 -2
  11. lightning_sdk/lightning_cloud/openapi/__init__.py +9 -0
  12. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +206 -0
  13. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +98 -5
  14. lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py +113 -0
  15. lightning_sdk/lightning_cloud/openapi/api/pipelines_service_api.py +118 -1
  16. lightning_sdk/lightning_cloud/openapi/api/schedules_service_api.py +5 -1
  17. lightning_sdk/lightning_cloud/openapi/models/__init__.py +9 -0
  18. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +29 -3
  19. lightning_sdk/lightning_cloud/openapi/models/cloudspace_id_visibility_body.py +123 -0
  20. lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py +27 -1
  21. lightning_sdk/lightning_cloud/openapi/models/metricsstream_create_body.py +27 -1
  22. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body.py +27 -1
  23. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
  24. lightning_sdk/lightning_cloud/openapi/models/project_id_pipelines_body.py +27 -1
  25. lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py +27 -1
  26. lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py +27 -1
  27. lightning_sdk/lightning_cloud/openapi/models/v1_check_cluster_name_availability_request.py +123 -0
  28. lightning_sdk/lightning_cloud/openapi/models/v1_check_cluster_name_availability_response.py +123 -0
  29. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -0
  30. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +53 -1
  31. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_session.py +29 -3
  32. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py +27 -1
  33. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +79 -1
  34. lightning_sdk/lightning_cloud/openapi/models/v1_create_deployment_request.py +27 -1
  35. lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_registry_repository_image_artifact_version_by_digest_response.py +97 -0
  36. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster.py +253 -0
  37. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster_spec.py +827 -0
  38. lightning_sdk/lightning_cloud/openapi/models/v1_get_project_storage_metadata_response.py +27 -1
  39. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +105 -1
  40. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_storage_breakdown_response.py +53 -1
  41. lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.py +29 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_lightning_run.py +53 -1
  43. lightning_sdk/lightning_cloud/openapi/models/v1_list_clusters_response.py +6 -6
  44. lightning_sdk/lightning_cloud/openapi/models/v1_list_project_clusters_response.py +6 -6
  45. lightning_sdk/lightning_cloud/openapi/models/v1_lite_published_cloud_space_response.py +513 -0
  46. lightning_sdk/lightning_cloud/openapi/models/v1_metrics_stream.py +27 -1
  47. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py +27 -1
  48. lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py +27 -1
  49. lightning_sdk/lightning_cloud/openapi/models/v1_shared_filesystem.py +131 -1
  50. lightning_sdk/lightning_cloud/openapi/models/v1_update_cloud_space_visibility_response.py +97 -0
  51. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +79 -1
  52. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +53 -183
  53. lightning_sdk/lightning_cloud/openapi/models/v1_volume.py +78 -104
  54. lightning_sdk/lightning_cloud/openapi/models/v1_volume_state.py +104 -0
  55. lightning_sdk/llm/llm.py +6 -1
  56. lightning_sdk/pipeline/__init__.py +12 -2
  57. lightning_sdk/pipeline/pipeline.py +59 -17
  58. lightning_sdk/pipeline/printer.py +121 -0
  59. lightning_sdk/pipeline/schedule.py +8 -0
  60. lightning_sdk/pipeline/{types.py → steps.py} +77 -59
  61. lightning_sdk/pipeline/utils.py +39 -17
  62. lightning_sdk/sandbox.py +157 -0
  63. lightning_sdk/services/license.py +12 -6
  64. lightning_sdk/studio.py +7 -1
  65. {lightning_sdk-0.2.21rc1.dist-info → lightning_sdk-0.2.23.dist-info}/METADATA +1 -1
  66. {lightning_sdk-0.2.21rc1.dist-info → lightning_sdk-0.2.23.dist-info}/RECORD +70 -58
  67. {lightning_sdk-0.2.21rc1.dist-info → lightning_sdk-0.2.23.dist-info}/LICENSE +0 -0
  68. {lightning_sdk-0.2.21rc1.dist-info → lightning_sdk-0.2.23.dist-info}/WHEEL +0 -0
  69. {lightning_sdk-0.2.21rc1.dist-info → lightning_sdk-0.2.23.dist-info}/entry_points.txt +0 -0
  70. {lightning_sdk-0.2.21rc1.dist-info → lightning_sdk-0.2.23.dist-info}/top_level.txt +0 -0
@@ -47,8 +47,6 @@ class V1UserFeatures(object):
47
47
  'ai_hub_monetization': 'bool',
48
48
  'auto_fast_load': 'bool',
49
49
  'auto_join_orgs': 'bool',
50
- 'auto_top_up': 'bool',
51
- 'auto_top_up_teamspace': 'bool',
52
50
  'b2c_experience': 'bool',
53
51
  'cap_add': 'list[str]',
54
52
  'cap_drop': 'list[str]',
@@ -63,10 +61,10 @@ class V1UserFeatures(object):
63
61
  'control_center_monitoring': 'bool',
64
62
  'cost_attribution_settings': 'bool',
65
63
  'custom_app_domain': 'bool',
64
+ 'data_connection_flushing_v2': 'bool',
66
65
  'datasets': 'bool',
67
66
  'default_one_cluster': 'bool',
68
67
  'deployment_persistent_disk': 'bool',
69
- 'docs_agent': 'bool',
70
68
  'drive_v2': 'bool',
71
69
  'enterprise_compute_admin': 'bool',
72
70
  'fair_share': 'bool',
@@ -78,6 +76,7 @@ class V1UserFeatures(object):
78
76
  'lambda_labs_studios': 'bool',
79
77
  'landing_studios': 'bool',
80
78
  'lit_logger': 'bool',
79
+ 'machine_selector_v2': 'bool',
81
80
  'marketplace': 'bool',
82
81
  'mmt_fault_tolerance': 'bool',
83
82
  'mmt_strategy_selector': 'bool',
@@ -93,7 +92,6 @@ class V1UserFeatures(object):
93
92
  'org_usage_limits': 'bool',
94
93
  'paygo_free_storage_limit_check': 'bool',
95
94
  'pipelines': 'bool',
96
- 'plg_control_center': 'bool',
97
95
  'plugin_distributed': 'bool',
98
96
  'plugin_inference': 'bool',
99
97
  'plugin_label_studio': 'bool',
@@ -111,15 +109,12 @@ class V1UserFeatures(object):
111
109
  'restartable_jobs': 'bool',
112
110
  'runnable_public_studio_page': 'bool',
113
111
  'security_docs': 'bool',
114
- 'seoul_aws_region': 'bool',
115
112
  'show_dev_admin': 'bool',
116
113
  'single_wallet': 'bool',
117
114
  'slurm': 'bool',
118
- 'slurm_machine_selector': 'bool',
119
115
  'storage_overuse_deletion': 'bool',
120
116
  'studio_config': 'bool',
121
117
  'studio_version_visibility': 'bool',
122
- 'teamspace_storage_tab': 'bool',
123
118
  'trainium2': 'bool',
124
119
  'use_rclone_mounts_only': 'bool',
125
120
  'voltage_park': 'bool',
@@ -135,8 +130,6 @@ class V1UserFeatures(object):
135
130
  'ai_hub_monetization': 'aiHubMonetization',
136
131
  'auto_fast_load': 'autoFastLoad',
137
132
  'auto_join_orgs': 'autoJoinOrgs',
138
- 'auto_top_up': 'autoTopUp',
139
- 'auto_top_up_teamspace': 'autoTopUpTeamspace',
140
133
  'b2c_experience': 'b2cExperience',
141
134
  'cap_add': 'capAdd',
142
135
  'cap_drop': 'capDrop',
@@ -151,10 +144,10 @@ class V1UserFeatures(object):
151
144
  'control_center_monitoring': 'controlCenterMonitoring',
152
145
  'cost_attribution_settings': 'costAttributionSettings',
153
146
  'custom_app_domain': 'customAppDomain',
147
+ 'data_connection_flushing_v2': 'dataConnectionFlushingV2',
154
148
  'datasets': 'datasets',
155
149
  'default_one_cluster': 'defaultOneCluster',
156
150
  'deployment_persistent_disk': 'deploymentPersistentDisk',
157
- 'docs_agent': 'docsAgent',
158
151
  'drive_v2': 'driveV2',
159
152
  'enterprise_compute_admin': 'enterpriseComputeAdmin',
160
153
  'fair_share': 'fairShare',
@@ -166,6 +159,7 @@ class V1UserFeatures(object):
166
159
  'lambda_labs_studios': 'lambdaLabsStudios',
167
160
  'landing_studios': 'landingStudios',
168
161
  'lit_logger': 'litLogger',
162
+ 'machine_selector_v2': 'machineSelectorV2',
169
163
  'marketplace': 'marketplace',
170
164
  'mmt_fault_tolerance': 'mmtFaultTolerance',
171
165
  'mmt_strategy_selector': 'mmtStrategySelector',
@@ -181,7 +175,6 @@ class V1UserFeatures(object):
181
175
  'org_usage_limits': 'orgUsageLimits',
182
176
  'paygo_free_storage_limit_check': 'paygoFreeStorageLimitCheck',
183
177
  'pipelines': 'pipelines',
184
- 'plg_control_center': 'plgControlCenter',
185
178
  'plugin_distributed': 'pluginDistributed',
186
179
  'plugin_inference': 'pluginInference',
187
180
  'plugin_label_studio': 'pluginLabelStudio',
@@ -199,15 +192,12 @@ class V1UserFeatures(object):
199
192
  'restartable_jobs': 'restartableJobs',
200
193
  'runnable_public_studio_page': 'runnablePublicStudioPage',
201
194
  'security_docs': 'securityDocs',
202
- 'seoul_aws_region': 'seoulAwsRegion',
203
195
  'show_dev_admin': 'showDevAdmin',
204
196
  'single_wallet': 'singleWallet',
205
197
  'slurm': 'slurm',
206
- 'slurm_machine_selector': 'slurmMachineSelector',
207
198
  'storage_overuse_deletion': 'storageOveruseDeletion',
208
199
  'studio_config': 'studioConfig',
209
200
  'studio_version_visibility': 'studioVersionVisibility',
210
- 'teamspace_storage_tab': 'teamspaceStorageTab',
211
201
  'trainium2': 'trainium2',
212
202
  'use_rclone_mounts_only': 'useRcloneMountsOnly',
213
203
  'voltage_park': 'voltagePark',
@@ -216,7 +206,7 @@ class V1UserFeatures(object):
216
206
  'weka': 'weka'
217
207
  }
218
208
 
219
- def __init__(self, accurate_billing: 'bool' =None, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, auto_top_up: 'bool' =None, auto_top_up_teamspace: '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, cloud_space_environment_templates: 'bool' =None, cloudy_vibe_code: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, concurrent_gpu_limit: 'bool' =None, control_center_monitoring: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_persistent_disk: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcs_folders: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, lambda_labs: 'bool' =None, lambda_labs_studios: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_api_dashboard: 'bool' =None, multicloud_folders: 'bool' =None, multicloud_saas: 'bool' =None, multiple_studio_versions: 'bool' =None, nebius: 'bool' =None, nebius_cpu_studios: 'bool' =None, nebius_gpu_studios: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, paygo_free_storage_limit_check: 'bool' =None, pipelines: 'bool' =None, plg_control_center: 'bool' =None, plugin_distributed: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, product_license: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, r2_data_connections: 'bool' =None, reserved_machines_tab: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, seoul_aws_region: 'bool' =None, show_dev_admin: 'bool' =None, single_wallet: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_version_visibility: 'bool' =None, teamspace_storage_tab: 'bool' =None, trainium2: 'bool' =None, use_rclone_mounts_only: 'bool' =None, voltage_park: 'bool' =None, voltage_park_studios: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None): # noqa: E501
209
+ def __init__(self, accurate_billing: 'bool' =None, 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, cloud_space_environment_templates: 'bool' =None, cloudy_vibe_code: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, concurrent_gpu_limit: 'bool' =None, control_center_monitoring: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, data_connection_flushing_v2: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_persistent_disk: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcs_folders: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, lambda_labs: 'bool' =None, lambda_labs_studios: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, machine_selector_v2: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_api_dashboard: 'bool' =None, multicloud_folders: 'bool' =None, multicloud_saas: 'bool' =None, multiple_studio_versions: 'bool' =None, nebius: 'bool' =None, nebius_cpu_studios: 'bool' =None, nebius_gpu_studios: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, paygo_free_storage_limit_check: 'bool' =None, pipelines: 'bool' =None, plugin_distributed: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, product_license: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, r2_data_connections: 'bool' =None, reserved_machines_tab: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, show_dev_admin: 'bool' =None, single_wallet: 'bool' =None, slurm: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_version_visibility: 'bool' =None, trainium2: 'bool' =None, use_rclone_mounts_only: 'bool' =None, voltage_park: 'bool' =None, voltage_park_studios: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None): # noqa: E501
220
210
  """V1UserFeatures - a model defined in Swagger""" # noqa: E501
221
211
  self._accurate_billing = None
222
212
  self._affiliate_links = None
@@ -224,8 +214,6 @@ class V1UserFeatures(object):
224
214
  self._ai_hub_monetization = None
225
215
  self._auto_fast_load = None
226
216
  self._auto_join_orgs = None
227
- self._auto_top_up = None
228
- self._auto_top_up_teamspace = None
229
217
  self._b2c_experience = None
230
218
  self._cap_add = None
231
219
  self._cap_drop = None
@@ -240,10 +228,10 @@ class V1UserFeatures(object):
240
228
  self._control_center_monitoring = None
241
229
  self._cost_attribution_settings = None
242
230
  self._custom_app_domain = None
231
+ self._data_connection_flushing_v2 = None
243
232
  self._datasets = None
244
233
  self._default_one_cluster = None
245
234
  self._deployment_persistent_disk = None
246
- self._docs_agent = None
247
235
  self._drive_v2 = None
248
236
  self._enterprise_compute_admin = None
249
237
  self._fair_share = None
@@ -255,6 +243,7 @@ class V1UserFeatures(object):
255
243
  self._lambda_labs_studios = None
256
244
  self._landing_studios = None
257
245
  self._lit_logger = None
246
+ self._machine_selector_v2 = None
258
247
  self._marketplace = None
259
248
  self._mmt_fault_tolerance = None
260
249
  self._mmt_strategy_selector = None
@@ -270,7 +259,6 @@ class V1UserFeatures(object):
270
259
  self._org_usage_limits = None
271
260
  self._paygo_free_storage_limit_check = None
272
261
  self._pipelines = None
273
- self._plg_control_center = None
274
262
  self._plugin_distributed = None
275
263
  self._plugin_inference = None
276
264
  self._plugin_label_studio = None
@@ -288,15 +276,12 @@ class V1UserFeatures(object):
288
276
  self._restartable_jobs = None
289
277
  self._runnable_public_studio_page = None
290
278
  self._security_docs = None
291
- self._seoul_aws_region = None
292
279
  self._show_dev_admin = None
293
280
  self._single_wallet = None
294
281
  self._slurm = None
295
- self._slurm_machine_selector = None
296
282
  self._storage_overuse_deletion = None
297
283
  self._studio_config = None
298
284
  self._studio_version_visibility = None
299
- self._teamspace_storage_tab = None
300
285
  self._trainium2 = None
301
286
  self._use_rclone_mounts_only = None
302
287
  self._voltage_park = None
@@ -316,10 +301,6 @@ class V1UserFeatures(object):
316
301
  self.auto_fast_load = auto_fast_load
317
302
  if auto_join_orgs is not None:
318
303
  self.auto_join_orgs = auto_join_orgs
319
- if auto_top_up is not None:
320
- self.auto_top_up = auto_top_up
321
- if auto_top_up_teamspace is not None:
322
- self.auto_top_up_teamspace = auto_top_up_teamspace
323
304
  if b2c_experience is not None:
324
305
  self.b2c_experience = b2c_experience
325
306
  if cap_add is not None:
@@ -348,14 +329,14 @@ class V1UserFeatures(object):
348
329
  self.cost_attribution_settings = cost_attribution_settings
349
330
  if custom_app_domain is not None:
350
331
  self.custom_app_domain = custom_app_domain
332
+ if data_connection_flushing_v2 is not None:
333
+ self.data_connection_flushing_v2 = data_connection_flushing_v2
351
334
  if datasets is not None:
352
335
  self.datasets = datasets
353
336
  if default_one_cluster is not None:
354
337
  self.default_one_cluster = default_one_cluster
355
338
  if deployment_persistent_disk is not None:
356
339
  self.deployment_persistent_disk = deployment_persistent_disk
357
- if docs_agent is not None:
358
- self.docs_agent = docs_agent
359
340
  if drive_v2 is not None:
360
341
  self.drive_v2 = drive_v2
361
342
  if enterprise_compute_admin is not None:
@@ -378,6 +359,8 @@ class V1UserFeatures(object):
378
359
  self.landing_studios = landing_studios
379
360
  if lit_logger is not None:
380
361
  self.lit_logger = lit_logger
362
+ if machine_selector_v2 is not None:
363
+ self.machine_selector_v2 = machine_selector_v2
381
364
  if marketplace is not None:
382
365
  self.marketplace = marketplace
383
366
  if mmt_fault_tolerance is not None:
@@ -408,8 +391,6 @@ class V1UserFeatures(object):
408
391
  self.paygo_free_storage_limit_check = paygo_free_storage_limit_check
409
392
  if pipelines is not None:
410
393
  self.pipelines = pipelines
411
- if plg_control_center is not None:
412
- self.plg_control_center = plg_control_center
413
394
  if plugin_distributed is not None:
414
395
  self.plugin_distributed = plugin_distributed
415
396
  if plugin_inference is not None:
@@ -444,24 +425,18 @@ class V1UserFeatures(object):
444
425
  self.runnable_public_studio_page = runnable_public_studio_page
445
426
  if security_docs is not None:
446
427
  self.security_docs = security_docs
447
- if seoul_aws_region is not None:
448
- self.seoul_aws_region = seoul_aws_region
449
428
  if show_dev_admin is not None:
450
429
  self.show_dev_admin = show_dev_admin
451
430
  if single_wallet is not None:
452
431
  self.single_wallet = single_wallet
453
432
  if slurm is not None:
454
433
  self.slurm = slurm
455
- if slurm_machine_selector is not None:
456
- self.slurm_machine_selector = slurm_machine_selector
457
434
  if storage_overuse_deletion is not None:
458
435
  self.storage_overuse_deletion = storage_overuse_deletion
459
436
  if studio_config is not None:
460
437
  self.studio_config = studio_config
461
438
  if studio_version_visibility is not None:
462
439
  self.studio_version_visibility = studio_version_visibility
463
- if teamspace_storage_tab is not None:
464
- self.teamspace_storage_tab = teamspace_storage_tab
465
440
  if trainium2 is not None:
466
441
  self.trainium2 = trainium2
467
442
  if use_rclone_mounts_only is not None:
@@ -601,48 +576,6 @@ class V1UserFeatures(object):
601
576
 
602
577
  self._auto_join_orgs = auto_join_orgs
603
578
 
604
- @property
605
- def auto_top_up(self) -> 'bool':
606
- """Gets the auto_top_up of this V1UserFeatures. # noqa: E501
607
-
608
-
609
- :return: The auto_top_up of this V1UserFeatures. # noqa: E501
610
- :rtype: bool
611
- """
612
- return self._auto_top_up
613
-
614
- @auto_top_up.setter
615
- def auto_top_up(self, auto_top_up: 'bool'):
616
- """Sets the auto_top_up of this V1UserFeatures.
617
-
618
-
619
- :param auto_top_up: The auto_top_up of this V1UserFeatures. # noqa: E501
620
- :type: bool
621
- """
622
-
623
- self._auto_top_up = auto_top_up
624
-
625
- @property
626
- def auto_top_up_teamspace(self) -> 'bool':
627
- """Gets the auto_top_up_teamspace of this V1UserFeatures. # noqa: E501
628
-
629
-
630
- :return: The auto_top_up_teamspace of this V1UserFeatures. # noqa: E501
631
- :rtype: bool
632
- """
633
- return self._auto_top_up_teamspace
634
-
635
- @auto_top_up_teamspace.setter
636
- def auto_top_up_teamspace(self, auto_top_up_teamspace: 'bool'):
637
- """Sets the auto_top_up_teamspace of this V1UserFeatures.
638
-
639
-
640
- :param auto_top_up_teamspace: The auto_top_up_teamspace of this V1UserFeatures. # noqa: E501
641
- :type: bool
642
- """
643
-
644
- self._auto_top_up_teamspace = auto_top_up_teamspace
645
-
646
579
  @property
647
580
  def b2c_experience(self) -> 'bool':
648
581
  """Gets the b2c_experience of this V1UserFeatures. # noqa: E501
@@ -937,6 +870,27 @@ class V1UserFeatures(object):
937
870
 
938
871
  self._custom_app_domain = custom_app_domain
939
872
 
873
+ @property
874
+ def data_connection_flushing_v2(self) -> 'bool':
875
+ """Gets the data_connection_flushing_v2 of this V1UserFeatures. # noqa: E501
876
+
877
+
878
+ :return: The data_connection_flushing_v2 of this V1UserFeatures. # noqa: E501
879
+ :rtype: bool
880
+ """
881
+ return self._data_connection_flushing_v2
882
+
883
+ @data_connection_flushing_v2.setter
884
+ def data_connection_flushing_v2(self, data_connection_flushing_v2: 'bool'):
885
+ """Sets the data_connection_flushing_v2 of this V1UserFeatures.
886
+
887
+
888
+ :param data_connection_flushing_v2: The data_connection_flushing_v2 of this V1UserFeatures. # noqa: E501
889
+ :type: bool
890
+ """
891
+
892
+ self._data_connection_flushing_v2 = data_connection_flushing_v2
893
+
940
894
  @property
941
895
  def datasets(self) -> 'bool':
942
896
  """Gets the datasets of this V1UserFeatures. # noqa: E501
@@ -1000,27 +954,6 @@ class V1UserFeatures(object):
1000
954
 
1001
955
  self._deployment_persistent_disk = deployment_persistent_disk
1002
956
 
1003
- @property
1004
- def docs_agent(self) -> 'bool':
1005
- """Gets the docs_agent of this V1UserFeatures. # noqa: E501
1006
-
1007
-
1008
- :return: The docs_agent of this V1UserFeatures. # noqa: E501
1009
- :rtype: bool
1010
- """
1011
- return self._docs_agent
1012
-
1013
- @docs_agent.setter
1014
- def docs_agent(self, docs_agent: 'bool'):
1015
- """Sets the docs_agent of this V1UserFeatures.
1016
-
1017
-
1018
- :param docs_agent: The docs_agent of this V1UserFeatures. # noqa: E501
1019
- :type: bool
1020
- """
1021
-
1022
- self._docs_agent = docs_agent
1023
-
1024
957
  @property
1025
958
  def drive_v2(self) -> 'bool':
1026
959
  """Gets the drive_v2 of this V1UserFeatures. # noqa: E501
@@ -1252,6 +1185,27 @@ class V1UserFeatures(object):
1252
1185
 
1253
1186
  self._lit_logger = lit_logger
1254
1187
 
1188
+ @property
1189
+ def machine_selector_v2(self) -> 'bool':
1190
+ """Gets the machine_selector_v2 of this V1UserFeatures. # noqa: E501
1191
+
1192
+
1193
+ :return: The machine_selector_v2 of this V1UserFeatures. # noqa: E501
1194
+ :rtype: bool
1195
+ """
1196
+ return self._machine_selector_v2
1197
+
1198
+ @machine_selector_v2.setter
1199
+ def machine_selector_v2(self, machine_selector_v2: 'bool'):
1200
+ """Sets the machine_selector_v2 of this V1UserFeatures.
1201
+
1202
+
1203
+ :param machine_selector_v2: The machine_selector_v2 of this V1UserFeatures. # noqa: E501
1204
+ :type: bool
1205
+ """
1206
+
1207
+ self._machine_selector_v2 = machine_selector_v2
1208
+
1255
1209
  @property
1256
1210
  def marketplace(self) -> 'bool':
1257
1211
  """Gets the marketplace of this V1UserFeatures. # noqa: E501
@@ -1567,27 +1521,6 @@ class V1UserFeatures(object):
1567
1521
 
1568
1522
  self._pipelines = pipelines
1569
1523
 
1570
- @property
1571
- def plg_control_center(self) -> 'bool':
1572
- """Gets the plg_control_center of this V1UserFeatures. # noqa: E501
1573
-
1574
-
1575
- :return: The plg_control_center of this V1UserFeatures. # noqa: E501
1576
- :rtype: bool
1577
- """
1578
- return self._plg_control_center
1579
-
1580
- @plg_control_center.setter
1581
- def plg_control_center(self, plg_control_center: 'bool'):
1582
- """Sets the plg_control_center of this V1UserFeatures.
1583
-
1584
-
1585
- :param plg_control_center: The plg_control_center of this V1UserFeatures. # noqa: E501
1586
- :type: bool
1587
- """
1588
-
1589
- self._plg_control_center = plg_control_center
1590
-
1591
1524
  @property
1592
1525
  def plugin_distributed(self) -> 'bool':
1593
1526
  """Gets the plugin_distributed of this V1UserFeatures. # noqa: E501
@@ -1945,27 +1878,6 @@ class V1UserFeatures(object):
1945
1878
 
1946
1879
  self._security_docs = security_docs
1947
1880
 
1948
- @property
1949
- def seoul_aws_region(self) -> 'bool':
1950
- """Gets the seoul_aws_region of this V1UserFeatures. # noqa: E501
1951
-
1952
-
1953
- :return: The seoul_aws_region of this V1UserFeatures. # noqa: E501
1954
- :rtype: bool
1955
- """
1956
- return self._seoul_aws_region
1957
-
1958
- @seoul_aws_region.setter
1959
- def seoul_aws_region(self, seoul_aws_region: 'bool'):
1960
- """Sets the seoul_aws_region of this V1UserFeatures.
1961
-
1962
-
1963
- :param seoul_aws_region: The seoul_aws_region of this V1UserFeatures. # noqa: E501
1964
- :type: bool
1965
- """
1966
-
1967
- self._seoul_aws_region = seoul_aws_region
1968
-
1969
1881
  @property
1970
1882
  def show_dev_admin(self) -> 'bool':
1971
1883
  """Gets the show_dev_admin of this V1UserFeatures. # noqa: E501
@@ -2029,27 +1941,6 @@ class V1UserFeatures(object):
2029
1941
 
2030
1942
  self._slurm = slurm
2031
1943
 
2032
- @property
2033
- def slurm_machine_selector(self) -> 'bool':
2034
- """Gets the slurm_machine_selector of this V1UserFeatures. # noqa: E501
2035
-
2036
-
2037
- :return: The slurm_machine_selector of this V1UserFeatures. # noqa: E501
2038
- :rtype: bool
2039
- """
2040
- return self._slurm_machine_selector
2041
-
2042
- @slurm_machine_selector.setter
2043
- def slurm_machine_selector(self, slurm_machine_selector: 'bool'):
2044
- """Sets the slurm_machine_selector of this V1UserFeatures.
2045
-
2046
-
2047
- :param slurm_machine_selector: The slurm_machine_selector of this V1UserFeatures. # noqa: E501
2048
- :type: bool
2049
- """
2050
-
2051
- self._slurm_machine_selector = slurm_machine_selector
2052
-
2053
1944
  @property
2054
1945
  def storage_overuse_deletion(self) -> 'bool':
2055
1946
  """Gets the storage_overuse_deletion of this V1UserFeatures. # noqa: E501
@@ -2113,27 +2004,6 @@ class V1UserFeatures(object):
2113
2004
 
2114
2005
  self._studio_version_visibility = studio_version_visibility
2115
2006
 
2116
- @property
2117
- def teamspace_storage_tab(self) -> 'bool':
2118
- """Gets the teamspace_storage_tab of this V1UserFeatures. # noqa: E501
2119
-
2120
-
2121
- :return: The teamspace_storage_tab of this V1UserFeatures. # noqa: E501
2122
- :rtype: bool
2123
- """
2124
- return self._teamspace_storage_tab
2125
-
2126
- @teamspace_storage_tab.setter
2127
- def teamspace_storage_tab(self, teamspace_storage_tab: 'bool'):
2128
- """Sets the teamspace_storage_tab of this V1UserFeatures.
2129
-
2130
-
2131
- :param teamspace_storage_tab: The teamspace_storage_tab of this V1UserFeatures. # noqa: E501
2132
- :type: bool
2133
- """
2134
-
2135
- self._teamspace_storage_tab = teamspace_storage_tab
2136
-
2137
2007
  @property
2138
2008
  def trainium2(self) -> 'bool':
2139
2009
  """Gets the trainium2 of this V1UserFeatures. # noqa: E501