lightning-sdk 0.2.22__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 (57) 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 +2 -0
  4. lightning_sdk/api/pipeline_api.py +1 -0
  5. lightning_sdk/api/studio_api.py +2 -0
  6. lightning_sdk/cli/entrypoint.py +15 -13
  7. lightning_sdk/cli/start.py +5 -2
  8. lightning_sdk/lightning_cloud/openapi/__init__.py +8 -0
  9. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +206 -0
  10. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +97 -0
  11. lightning_sdk/lightning_cloud/openapi/api/pipelines_service_api.py +118 -1
  12. lightning_sdk/lightning_cloud/openapi/configuration.py +1 -1
  13. lightning_sdk/lightning_cloud/openapi/models/__init__.py +8 -0
  14. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +29 -3
  15. lightning_sdk/lightning_cloud/openapi/models/cloudspace_id_visibility_body.py +123 -0
  16. lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py +27 -1
  17. lightning_sdk/lightning_cloud/openapi/models/metricsstream_create_body.py +27 -1
  18. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body.py +27 -1
  19. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
  20. lightning_sdk/lightning_cloud/openapi/models/v1_check_cluster_name_availability_request.py +123 -0
  21. lightning_sdk/lightning_cloud/openapi/models/v1_check_cluster_name_availability_response.py +123 -0
  22. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -0
  23. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +53 -1
  24. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_session.py +29 -3
  25. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py +27 -1
  26. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +79 -1
  27. lightning_sdk/lightning_cloud/openapi/models/v1_create_deployment_request.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster.py +253 -0
  29. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster_spec.py +827 -0
  30. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
  31. lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.py +29 -1
  32. lightning_sdk/lightning_cloud/openapi/models/v1_lightning_run.py +53 -1
  33. lightning_sdk/lightning_cloud/openapi/models/v1_list_clusters_response.py +6 -6
  34. lightning_sdk/lightning_cloud/openapi/models/v1_list_project_clusters_response.py +6 -6
  35. lightning_sdk/lightning_cloud/openapi/models/v1_lite_published_cloud_space_response.py +513 -0
  36. lightning_sdk/lightning_cloud/openapi/models/v1_metrics_stream.py +27 -1
  37. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py +27 -1
  38. lightning_sdk/lightning_cloud/openapi/models/v1_shared_filesystem.py +131 -1
  39. lightning_sdk/lightning_cloud/openapi/models/v1_update_cloud_space_visibility_response.py +97 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +53 -79
  41. lightning_sdk/lightning_cloud/openapi/models/v1_volume.py +78 -104
  42. lightning_sdk/lightning_cloud/openapi/models/v1_volume_state.py +104 -0
  43. lightning_sdk/pipeline/__init__.py +11 -2
  44. lightning_sdk/pipeline/pipeline.py +38 -13
  45. lightning_sdk/pipeline/printer.py +29 -10
  46. lightning_sdk/pipeline/schedule.py +2 -1
  47. lightning_sdk/pipeline/{types.py → steps.py} +74 -56
  48. lightning_sdk/pipeline/utils.py +39 -2
  49. lightning_sdk/sandbox.py +157 -0
  50. lightning_sdk/services/license.py +12 -6
  51. lightning_sdk/studio.py +7 -1
  52. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.23.dist-info}/METADATA +1 -1
  53. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.23.dist-info}/RECORD +57 -48
  54. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.23.dist-info}/LICENSE +0 -0
  55. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.23.dist-info}/WHEEL +0 -0
  56. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.23.dist-info}/entry_points.txt +0 -0
  57. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.23.dist-info}/top_level.txt +0 -0
@@ -41,35 +41,81 @@ class V1SharedFilesystem(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
+ 'efs': 'bool',
44
45
  'enabled': 'bool',
46
+ 'filestore': 'bool',
47
+ 'gcs_folder': 'bool',
48
+ 'r2_folder': 'bool',
45
49
  'resource_id': 'str',
46
50
  'resource_name': 'str',
51
+ 's3_folder': 'bool',
47
52
  'state': 'str'
48
53
  }
49
54
 
50
55
  attribute_map = {
56
+ 'efs': 'efs',
51
57
  'enabled': 'enabled',
58
+ 'filestore': 'filestore',
59
+ 'gcs_folder': 'gcsFolder',
60
+ 'r2_folder': 'r2Folder',
52
61
  'resource_id': 'resourceId',
53
62
  'resource_name': 'resourceName',
63
+ 's3_folder': 's3Folder',
54
64
  'state': 'state'
55
65
  }
56
66
 
57
- def __init__(self, enabled: 'bool' =None, resource_id: 'str' =None, resource_name: 'str' =None, state: 'str' =None): # noqa: E501
67
+ def __init__(self, efs: 'bool' =None, enabled: 'bool' =None, filestore: 'bool' =None, gcs_folder: 'bool' =None, r2_folder: 'bool' =None, resource_id: 'str' =None, resource_name: 'str' =None, s3_folder: 'bool' =None, state: 'str' =None): # noqa: E501
58
68
  """V1SharedFilesystem - a model defined in Swagger""" # noqa: E501
69
+ self._efs = None
59
70
  self._enabled = None
71
+ self._filestore = None
72
+ self._gcs_folder = None
73
+ self._r2_folder = None
60
74
  self._resource_id = None
61
75
  self._resource_name = None
76
+ self._s3_folder = None
62
77
  self._state = None
63
78
  self.discriminator = None
79
+ if efs is not None:
80
+ self.efs = efs
64
81
  if enabled is not None:
65
82
  self.enabled = enabled
83
+ if filestore is not None:
84
+ self.filestore = filestore
85
+ if gcs_folder is not None:
86
+ self.gcs_folder = gcs_folder
87
+ if r2_folder is not None:
88
+ self.r2_folder = r2_folder
66
89
  if resource_id is not None:
67
90
  self.resource_id = resource_id
68
91
  if resource_name is not None:
69
92
  self.resource_name = resource_name
93
+ if s3_folder is not None:
94
+ self.s3_folder = s3_folder
70
95
  if state is not None:
71
96
  self.state = state
72
97
 
98
+ @property
99
+ def efs(self) -> 'bool':
100
+ """Gets the efs of this V1SharedFilesystem. # noqa: E501
101
+
102
+
103
+ :return: The efs of this V1SharedFilesystem. # noqa: E501
104
+ :rtype: bool
105
+ """
106
+ return self._efs
107
+
108
+ @efs.setter
109
+ def efs(self, efs: 'bool'):
110
+ """Sets the efs of this V1SharedFilesystem.
111
+
112
+
113
+ :param efs: The efs of this V1SharedFilesystem. # noqa: E501
114
+ :type: bool
115
+ """
116
+
117
+ self._efs = efs
118
+
73
119
  @property
74
120
  def enabled(self) -> 'bool':
75
121
  """Gets the enabled of this V1SharedFilesystem. # noqa: E501
@@ -91,6 +137,69 @@ class V1SharedFilesystem(object):
91
137
 
92
138
  self._enabled = enabled
93
139
 
140
+ @property
141
+ def filestore(self) -> 'bool':
142
+ """Gets the filestore of this V1SharedFilesystem. # noqa: E501
143
+
144
+
145
+ :return: The filestore of this V1SharedFilesystem. # noqa: E501
146
+ :rtype: bool
147
+ """
148
+ return self._filestore
149
+
150
+ @filestore.setter
151
+ def filestore(self, filestore: 'bool'):
152
+ """Sets the filestore of this V1SharedFilesystem.
153
+
154
+
155
+ :param filestore: The filestore of this V1SharedFilesystem. # noqa: E501
156
+ :type: bool
157
+ """
158
+
159
+ self._filestore = filestore
160
+
161
+ @property
162
+ def gcs_folder(self) -> 'bool':
163
+ """Gets the gcs_folder of this V1SharedFilesystem. # noqa: E501
164
+
165
+
166
+ :return: The gcs_folder of this V1SharedFilesystem. # noqa: E501
167
+ :rtype: bool
168
+ """
169
+ return self._gcs_folder
170
+
171
+ @gcs_folder.setter
172
+ def gcs_folder(self, gcs_folder: 'bool'):
173
+ """Sets the gcs_folder of this V1SharedFilesystem.
174
+
175
+
176
+ :param gcs_folder: The gcs_folder of this V1SharedFilesystem. # noqa: E501
177
+ :type: bool
178
+ """
179
+
180
+ self._gcs_folder = gcs_folder
181
+
182
+ @property
183
+ def r2_folder(self) -> 'bool':
184
+ """Gets the r2_folder of this V1SharedFilesystem. # noqa: E501
185
+
186
+
187
+ :return: The r2_folder of this V1SharedFilesystem. # noqa: E501
188
+ :rtype: bool
189
+ """
190
+ return self._r2_folder
191
+
192
+ @r2_folder.setter
193
+ def r2_folder(self, r2_folder: 'bool'):
194
+ """Sets the r2_folder of this V1SharedFilesystem.
195
+
196
+
197
+ :param r2_folder: The r2_folder of this V1SharedFilesystem. # noqa: E501
198
+ :type: bool
199
+ """
200
+
201
+ self._r2_folder = r2_folder
202
+
94
203
  @property
95
204
  def resource_id(self) -> 'str':
96
205
  """Gets the resource_id of this V1SharedFilesystem. # noqa: E501
@@ -133,6 +242,27 @@ class V1SharedFilesystem(object):
133
242
 
134
243
  self._resource_name = resource_name
135
244
 
245
+ @property
246
+ def s3_folder(self) -> 'bool':
247
+ """Gets the s3_folder of this V1SharedFilesystem. # noqa: E501
248
+
249
+
250
+ :return: The s3_folder of this V1SharedFilesystem. # noqa: E501
251
+ :rtype: bool
252
+ """
253
+ return self._s3_folder
254
+
255
+ @s3_folder.setter
256
+ def s3_folder(self, s3_folder: 'bool'):
257
+ """Sets the s3_folder of this V1SharedFilesystem.
258
+
259
+
260
+ :param s3_folder: The s3_folder of this V1SharedFilesystem. # noqa: E501
261
+ :type: bool
262
+ """
263
+
264
+ self._s3_folder = s3_folder
265
+
136
266
  @property
137
267
  def state(self) -> 'str':
138
268
  """Gets the state of this V1SharedFilesystem. # noqa: E501
@@ -0,0 +1,97 @@
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
+ import pprint
21
+ import re # noqa: F401
22
+
23
+ from typing import TYPE_CHECKING
24
+
25
+ import six
26
+
27
+ if TYPE_CHECKING:
28
+ from datetime import datetime
29
+ from lightning_sdk.lightning_cloud.openapi.models import *
30
+
31
+ class V1UpdateCloudSpaceVisibilityResponse(object):
32
+ """NOTE: This class is auto generated by the swagger code generator program.
33
+
34
+ Do not edit the class manually.
35
+ """
36
+ """
37
+ Attributes:
38
+ swagger_types (dict): The key is attribute name
39
+ and the value is attribute type.
40
+ attribute_map (dict): The key is attribute name
41
+ and the value is json key in definition.
42
+ """
43
+ swagger_types = {
44
+ }
45
+
46
+ attribute_map = {
47
+ }
48
+
49
+ def __init__(self): # noqa: E501
50
+ """V1UpdateCloudSpaceVisibilityResponse - a model defined in Swagger""" # noqa: E501
51
+ self.discriminator = None
52
+
53
+ def to_dict(self) -> dict:
54
+ """Returns the model properties as a dict"""
55
+ result = {}
56
+
57
+ for attr, _ in six.iteritems(self.swagger_types):
58
+ value = getattr(self, attr)
59
+ if isinstance(value, list):
60
+ result[attr] = list(map(
61
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
62
+ value
63
+ ))
64
+ elif hasattr(value, "to_dict"):
65
+ result[attr] = value.to_dict()
66
+ elif isinstance(value, dict):
67
+ result[attr] = dict(map(
68
+ lambda item: (item[0], item[1].to_dict())
69
+ if hasattr(item[1], "to_dict") else item,
70
+ value.items()
71
+ ))
72
+ else:
73
+ result[attr] = value
74
+ if issubclass(V1UpdateCloudSpaceVisibilityResponse, dict):
75
+ for key, value in self.items():
76
+ result[key] = value
77
+
78
+ return result
79
+
80
+ def to_str(self) -> str:
81
+ """Returns the string representation of the model"""
82
+ return pprint.pformat(self.to_dict())
83
+
84
+ def __repr__(self) -> str:
85
+ """For `print` and `pprint`"""
86
+ return self.to_str()
87
+
88
+ def __eq__(self, other: 'V1UpdateCloudSpaceVisibilityResponse') -> bool:
89
+ """Returns true if both objects are equal"""
90
+ if not isinstance(other, V1UpdateCloudSpaceVisibilityResponse):
91
+ return False
92
+
93
+ return self.__dict__ == other.__dict__
94
+
95
+ def __ne__(self, other: 'V1UpdateCloudSpaceVisibilityResponse') -> bool:
96
+ """Returns true if both objects are not equal"""
97
+ return not self == other
@@ -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,6 +61,7 @@ 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',
@@ -77,6 +76,7 @@ class V1UserFeatures(object):
77
76
  'lambda_labs_studios': 'bool',
78
77
  'landing_studios': 'bool',
79
78
  'lit_logger': 'bool',
79
+ 'machine_selector_v2': 'bool',
80
80
  'marketplace': 'bool',
81
81
  'mmt_fault_tolerance': 'bool',
82
82
  'mmt_strategy_selector': 'bool',
@@ -92,7 +92,6 @@ class V1UserFeatures(object):
92
92
  'org_usage_limits': 'bool',
93
93
  'paygo_free_storage_limit_check': 'bool',
94
94
  'pipelines': 'bool',
95
- 'plg_control_center': 'bool',
96
95
  'plugin_distributed': 'bool',
97
96
  'plugin_inference': 'bool',
98
97
  'plugin_label_studio': 'bool',
@@ -131,8 +130,6 @@ class V1UserFeatures(object):
131
130
  'ai_hub_monetization': 'aiHubMonetization',
132
131
  'auto_fast_load': 'autoFastLoad',
133
132
  'auto_join_orgs': 'autoJoinOrgs',
134
- 'auto_top_up': 'autoTopUp',
135
- 'auto_top_up_teamspace': 'autoTopUpTeamspace',
136
133
  'b2c_experience': 'b2cExperience',
137
134
  'cap_add': 'capAdd',
138
135
  'cap_drop': 'capDrop',
@@ -147,6 +144,7 @@ class V1UserFeatures(object):
147
144
  'control_center_monitoring': 'controlCenterMonitoring',
148
145
  'cost_attribution_settings': 'costAttributionSettings',
149
146
  'custom_app_domain': 'customAppDomain',
147
+ 'data_connection_flushing_v2': 'dataConnectionFlushingV2',
150
148
  'datasets': 'datasets',
151
149
  'default_one_cluster': 'defaultOneCluster',
152
150
  'deployment_persistent_disk': 'deploymentPersistentDisk',
@@ -161,6 +159,7 @@ class V1UserFeatures(object):
161
159
  'lambda_labs_studios': 'lambdaLabsStudios',
162
160
  'landing_studios': 'landingStudios',
163
161
  'lit_logger': 'litLogger',
162
+ 'machine_selector_v2': 'machineSelectorV2',
164
163
  'marketplace': 'marketplace',
165
164
  'mmt_fault_tolerance': 'mmtFaultTolerance',
166
165
  'mmt_strategy_selector': 'mmtStrategySelector',
@@ -176,7 +175,6 @@ class V1UserFeatures(object):
176
175
  'org_usage_limits': 'orgUsageLimits',
177
176
  'paygo_free_storage_limit_check': 'paygoFreeStorageLimitCheck',
178
177
  'pipelines': 'pipelines',
179
- 'plg_control_center': 'plgControlCenter',
180
178
  'plugin_distributed': 'pluginDistributed',
181
179
  'plugin_inference': 'pluginInference',
182
180
  'plugin_label_studio': 'pluginLabelStudio',
@@ -208,7 +206,7 @@ class V1UserFeatures(object):
208
206
  'weka': 'weka'
209
207
  }
210
208
 
211
- 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, 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, 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
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
212
210
  """V1UserFeatures - a model defined in Swagger""" # noqa: E501
213
211
  self._accurate_billing = None
214
212
  self._affiliate_links = None
@@ -216,8 +214,6 @@ class V1UserFeatures(object):
216
214
  self._ai_hub_monetization = None
217
215
  self._auto_fast_load = None
218
216
  self._auto_join_orgs = None
219
- self._auto_top_up = None
220
- self._auto_top_up_teamspace = None
221
217
  self._b2c_experience = None
222
218
  self._cap_add = None
223
219
  self._cap_drop = None
@@ -232,6 +228,7 @@ class V1UserFeatures(object):
232
228
  self._control_center_monitoring = None
233
229
  self._cost_attribution_settings = None
234
230
  self._custom_app_domain = None
231
+ self._data_connection_flushing_v2 = None
235
232
  self._datasets = None
236
233
  self._default_one_cluster = None
237
234
  self._deployment_persistent_disk = None
@@ -246,6 +243,7 @@ class V1UserFeatures(object):
246
243
  self._lambda_labs_studios = None
247
244
  self._landing_studios = None
248
245
  self._lit_logger = None
246
+ self._machine_selector_v2 = None
249
247
  self._marketplace = None
250
248
  self._mmt_fault_tolerance = None
251
249
  self._mmt_strategy_selector = None
@@ -261,7 +259,6 @@ class V1UserFeatures(object):
261
259
  self._org_usage_limits = None
262
260
  self._paygo_free_storage_limit_check = None
263
261
  self._pipelines = None
264
- self._plg_control_center = None
265
262
  self._plugin_distributed = None
266
263
  self._plugin_inference = None
267
264
  self._plugin_label_studio = None
@@ -304,10 +301,6 @@ class V1UserFeatures(object):
304
301
  self.auto_fast_load = auto_fast_load
305
302
  if auto_join_orgs is not None:
306
303
  self.auto_join_orgs = auto_join_orgs
307
- if auto_top_up is not None:
308
- self.auto_top_up = auto_top_up
309
- if auto_top_up_teamspace is not None:
310
- self.auto_top_up_teamspace = auto_top_up_teamspace
311
304
  if b2c_experience is not None:
312
305
  self.b2c_experience = b2c_experience
313
306
  if cap_add is not None:
@@ -336,6 +329,8 @@ class V1UserFeatures(object):
336
329
  self.cost_attribution_settings = cost_attribution_settings
337
330
  if custom_app_domain is not None:
338
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
339
334
  if datasets is not None:
340
335
  self.datasets = datasets
341
336
  if default_one_cluster is not None:
@@ -364,6 +359,8 @@ class V1UserFeatures(object):
364
359
  self.landing_studios = landing_studios
365
360
  if lit_logger is not None:
366
361
  self.lit_logger = lit_logger
362
+ if machine_selector_v2 is not None:
363
+ self.machine_selector_v2 = machine_selector_v2
367
364
  if marketplace is not None:
368
365
  self.marketplace = marketplace
369
366
  if mmt_fault_tolerance is not None:
@@ -394,8 +391,6 @@ class V1UserFeatures(object):
394
391
  self.paygo_free_storage_limit_check = paygo_free_storage_limit_check
395
392
  if pipelines is not None:
396
393
  self.pipelines = pipelines
397
- if plg_control_center is not None:
398
- self.plg_control_center = plg_control_center
399
394
  if plugin_distributed is not None:
400
395
  self.plugin_distributed = plugin_distributed
401
396
  if plugin_inference is not None:
@@ -581,48 +576,6 @@ class V1UserFeatures(object):
581
576
 
582
577
  self._auto_join_orgs = auto_join_orgs
583
578
 
584
- @property
585
- def auto_top_up(self) -> 'bool':
586
- """Gets the auto_top_up of this V1UserFeatures. # noqa: E501
587
-
588
-
589
- :return: The auto_top_up of this V1UserFeatures. # noqa: E501
590
- :rtype: bool
591
- """
592
- return self._auto_top_up
593
-
594
- @auto_top_up.setter
595
- def auto_top_up(self, auto_top_up: 'bool'):
596
- """Sets the auto_top_up of this V1UserFeatures.
597
-
598
-
599
- :param auto_top_up: The auto_top_up of this V1UserFeatures. # noqa: E501
600
- :type: bool
601
- """
602
-
603
- self._auto_top_up = auto_top_up
604
-
605
- @property
606
- def auto_top_up_teamspace(self) -> 'bool':
607
- """Gets the auto_top_up_teamspace of this V1UserFeatures. # noqa: E501
608
-
609
-
610
- :return: The auto_top_up_teamspace of this V1UserFeatures. # noqa: E501
611
- :rtype: bool
612
- """
613
- return self._auto_top_up_teamspace
614
-
615
- @auto_top_up_teamspace.setter
616
- def auto_top_up_teamspace(self, auto_top_up_teamspace: 'bool'):
617
- """Sets the auto_top_up_teamspace of this V1UserFeatures.
618
-
619
-
620
- :param auto_top_up_teamspace: The auto_top_up_teamspace of this V1UserFeatures. # noqa: E501
621
- :type: bool
622
- """
623
-
624
- self._auto_top_up_teamspace = auto_top_up_teamspace
625
-
626
579
  @property
627
580
  def b2c_experience(self) -> 'bool':
628
581
  """Gets the b2c_experience of this V1UserFeatures. # noqa: E501
@@ -917,6 +870,27 @@ class V1UserFeatures(object):
917
870
 
918
871
  self._custom_app_domain = custom_app_domain
919
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
+
920
894
  @property
921
895
  def datasets(self) -> 'bool':
922
896
  """Gets the datasets of this V1UserFeatures. # noqa: E501
@@ -1211,6 +1185,27 @@ class V1UserFeatures(object):
1211
1185
 
1212
1186
  self._lit_logger = lit_logger
1213
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
+
1214
1209
  @property
1215
1210
  def marketplace(self) -> 'bool':
1216
1211
  """Gets the marketplace of this V1UserFeatures. # noqa: E501
@@ -1526,27 +1521,6 @@ class V1UserFeatures(object):
1526
1521
 
1527
1522
  self._pipelines = pipelines
1528
1523
 
1529
- @property
1530
- def plg_control_center(self) -> 'bool':
1531
- """Gets the plg_control_center of this V1UserFeatures. # noqa: E501
1532
-
1533
-
1534
- :return: The plg_control_center of this V1UserFeatures. # noqa: E501
1535
- :rtype: bool
1536
- """
1537
- return self._plg_control_center
1538
-
1539
- @plg_control_center.setter
1540
- def plg_control_center(self, plg_control_center: 'bool'):
1541
- """Sets the plg_control_center of this V1UserFeatures.
1542
-
1543
-
1544
- :param plg_control_center: The plg_control_center of this V1UserFeatures. # noqa: E501
1545
- :type: bool
1546
- """
1547
-
1548
- self._plg_control_center = plg_control_center
1549
-
1550
1524
  @property
1551
1525
  def plugin_distributed(self) -> 'bool':
1552
1526
  """Gets the plugin_distributed of this V1UserFeatures. # noqa: E501