lightning-sdk 2025.8.18__py3-none-any.whl → 2025.8.19.post0__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_node_metrics.py

@@ -44,12 +44,15 @@ class V1NodeMetrics(object):
         'cpu_capacity_cores': 'float',
         'cpu_util': 'float',
         'filesystem': 'dict(str, V1FilesystemMetrics)',
+        'max_power_per_gpu': 'float',
         'mem_total': 'float',
         'mem_util': 'float',
         'node_name': 'str',
         'per_gpu_mem_free': 'dict(str, float)',
         'per_gpu_mem_used': 'dict(str, float)',
         'per_gpu_power_usage_watts': 'dict(str, float)',
+        'per_gpu_sm_active': 'dict(str, float)',
+        'per_gpu_sm_occupancy': 'dict(str, float)',
         'per_gpu_temperature_c': 'dict(str, float)',
         'per_gpu_util': 'dict(str, float)',
         'timestamp': 'datetime'
@@ -59,28 +62,34 @@ class V1NodeMetrics(object):
         'cpu_capacity_cores': 'cpuCapacityCores',
         'cpu_util': 'cpuUtil',
         'filesystem': 'filesystem',
+        'max_power_per_gpu': 'maxPowerPerGpu',
         'mem_total': 'memTotal',
         'mem_util': 'memUtil',
         'node_name': 'nodeName',
         'per_gpu_mem_free': 'perGpuMemFree',
         'per_gpu_mem_used': 'perGpuMemUsed',
         'per_gpu_power_usage_watts': 'perGpuPowerUsageWatts',
+        'per_gpu_sm_active': 'perGpuSmActive',
+        'per_gpu_sm_occupancy': 'perGpuSmOccupancy',
         'per_gpu_temperature_c': 'perGpuTemperatureC',
         'per_gpu_util': 'perGpuUtil',
         'timestamp': 'timestamp'
     }
 
-    def __init__(self, cpu_capacity_cores: 'float' =None, cpu_util: 'float' =None, filesystem: 'dict(str, V1FilesystemMetrics)' =None, mem_total: 'float' =None, mem_util: 'float' =None, node_name: 'str' =None, per_gpu_mem_free: 'dict(str, float)' =None, per_gpu_mem_used: 'dict(str, float)' =None, per_gpu_power_usage_watts: 'dict(str, float)' =None, per_gpu_temperature_c: 'dict(str, float)' =None, per_gpu_util: 'dict(str, float)' =None, timestamp: 'datetime' =None):  # noqa: E501
+    def __init__(self, cpu_capacity_cores: 'float' =None, cpu_util: 'float' =None, filesystem: 'dict(str, V1FilesystemMetrics)' =None, max_power_per_gpu: 'float' =None, mem_total: 'float' =None, mem_util: 'float' =None, node_name: 'str' =None, per_gpu_mem_free: 'dict(str, float)' =None, per_gpu_mem_used: 'dict(str, float)' =None, per_gpu_power_usage_watts: 'dict(str, float)' =None, per_gpu_sm_active: 'dict(str, float)' =None, per_gpu_sm_occupancy: 'dict(str, float)' =None, per_gpu_temperature_c: 'dict(str, float)' =None, per_gpu_util: 'dict(str, float)' =None, timestamp: 'datetime' =None):  # noqa: E501
         """V1NodeMetrics - a model defined in Swagger"""  # noqa: E501
         self._cpu_capacity_cores = None
         self._cpu_util = None
         self._filesystem = None
+        self._max_power_per_gpu = None
         self._mem_total = None
         self._mem_util = None
         self._node_name = None
         self._per_gpu_mem_free = None
         self._per_gpu_mem_used = None
         self._per_gpu_power_usage_watts = None
+        self._per_gpu_sm_active = None
+        self._per_gpu_sm_occupancy = None
         self._per_gpu_temperature_c = None
         self._per_gpu_util = None
         self._timestamp = None
@@ -91,6 +100,8 @@ class V1NodeMetrics(object):
             self.cpu_util = cpu_util
         if filesystem is not None:
             self.filesystem = filesystem
+        if max_power_per_gpu is not None:
+            self.max_power_per_gpu = max_power_per_gpu
         if mem_total is not None:
             self.mem_total = mem_total
         if mem_util is not None:
@@ -103,6 +114,10 @@ class V1NodeMetrics(object):
             self.per_gpu_mem_used = per_gpu_mem_used
         if per_gpu_power_usage_watts is not None:
             self.per_gpu_power_usage_watts = per_gpu_power_usage_watts
+        if per_gpu_sm_active is not None:
+            self.per_gpu_sm_active = per_gpu_sm_active
+        if per_gpu_sm_occupancy is not None:
+            self.per_gpu_sm_occupancy = per_gpu_sm_occupancy
         if per_gpu_temperature_c is not None:
             self.per_gpu_temperature_c = per_gpu_temperature_c
         if per_gpu_util is not None:
@@ -173,6 +188,27 @@ class V1NodeMetrics(object):
 
         self._filesystem = filesystem
 
+    @property
+    def max_power_per_gpu(self) -> 'float':
+        """Gets the max_power_per_gpu of this V1NodeMetrics.  # noqa: E501
+
+
+        :return: The max_power_per_gpu of this V1NodeMetrics.  # noqa: E501
+        :rtype: float
+        """
+        return self._max_power_per_gpu
+
+    @max_power_per_gpu.setter
+    def max_power_per_gpu(self, max_power_per_gpu: 'float'):
+        """Sets the max_power_per_gpu of this V1NodeMetrics.
+
+
+        :param max_power_per_gpu: The max_power_per_gpu of this V1NodeMetrics.  # noqa: E501
+        :type: float
+        """
+
+        self._max_power_per_gpu = max_power_per_gpu
+
     @property
     def mem_total(self) -> 'float':
         """Gets the mem_total of this V1NodeMetrics.  # noqa: E501
@@ -299,6 +335,48 @@ class V1NodeMetrics(object):
 
         self._per_gpu_power_usage_watts = per_gpu_power_usage_watts
 
+    @property
+    def per_gpu_sm_active(self) -> 'dict(str, float)':
+        """Gets the per_gpu_sm_active of this V1NodeMetrics.  # noqa: E501
+
+
+        :return: The per_gpu_sm_active of this V1NodeMetrics.  # noqa: E501
+        :rtype: dict(str, float)
+        """
+        return self._per_gpu_sm_active
+
+    @per_gpu_sm_active.setter
+    def per_gpu_sm_active(self, per_gpu_sm_active: 'dict(str, float)'):
+        """Sets the per_gpu_sm_active of this V1NodeMetrics.
+
+
+        :param per_gpu_sm_active: The per_gpu_sm_active of this V1NodeMetrics.  # noqa: E501
+        :type: dict(str, float)
+        """
+
+        self._per_gpu_sm_active = per_gpu_sm_active
+
+    @property
+    def per_gpu_sm_occupancy(self) -> 'dict(str, float)':
+        """Gets the per_gpu_sm_occupancy of this V1NodeMetrics.  # noqa: E501
+
+
+        :return: The per_gpu_sm_occupancy of this V1NodeMetrics.  # noqa: E501
+        :rtype: dict(str, float)
+        """
+        return self._per_gpu_sm_occupancy
+
+    @per_gpu_sm_occupancy.setter
+    def per_gpu_sm_occupancy(self, per_gpu_sm_occupancy: 'dict(str, float)'):
+        """Sets the per_gpu_sm_occupancy of this V1NodeMetrics.
+
+
+        :param per_gpu_sm_occupancy: The per_gpu_sm_occupancy of this V1NodeMetrics.  # noqa: E501
+        :type: dict(str, float)
+        """
+
+        self._per_gpu_sm_occupancy = per_gpu_sm_occupancy
+
     @property
     def per_gpu_temperature_c(self) -> 'dict(str, float)':
         """Gets the per_gpu_temperature_c of this V1NodeMetrics.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_organization.py

@@ -82,6 +82,7 @@ class V1Organization(object):
         'preferred_cluster': 'str',
         'preferred_deployment_provider': 'str',
         'preferred_studio_provider': 'str',
+        'show_model_apis_tab': 'bool',
         'start_studios_on_spot_instance': 'bool',
         'storage_overuse_bytes': 'str',
         'storage_overuse_deletion_at': 'datetime',
@@ -135,6 +136,7 @@ class V1Organization(object):
         'preferred_cluster': 'preferredCluster',
         'preferred_deployment_provider': 'preferredDeploymentProvider',
         'preferred_studio_provider': 'preferredStudioProvider',
+        'show_model_apis_tab': 'showModelApisTab',
         'start_studios_on_spot_instance': 'startStudiosOnSpotInstance',
         'storage_overuse_bytes': 'storageOveruseBytes',
         'storage_overuse_deletion_at': 'storageOveruseDeletionAt',
@@ -146,7 +148,7 @@ class V1Organization(object):
         'workload_max_run_duration': 'workloadMaxRunDuration'
     }
 
-    def __init__(self, alerts_config: 'V1AlertsConfig' =None, allow_budgeting: 'bool' =None, allow_credits_auto_replenish: 'bool' =None, allow_external_project_duplication: 'bool' =None, allow_guest: 'bool' =None, allow_marketplace: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domain_validations: 'dict(str, V1AutoJoinDomainValidation)' =None, auto_join_domains: 'list[str]' =None, auto_replenish_amount: 'float' =None, auto_replenish_threshold: 'float' =None, auto_switch_machine: 'bool' =None, created_at: 'datetime' =None, default_machine_image_version: 'str' =None, default_machine_type: 'str' =None, default_project_id: 'str' =None, description: 'str' =None, disallow_aws_saas: 'bool' =None, disallow_dgx_saas: 'bool' =None, disallow_gcp_saas: 'bool' =None, disallow_lambda_saas: 'bool' =None, disallow_lightning_saas: 'bool' =None, disallow_nebius_saas: 'bool' =None, disallow_voltage_park_saas: 'bool' =None, disallow_vultr_saas: 'bool' =None, display_name: 'str' =None, domain: 'str' =None, email: 'str' =None, featured_gallery: 'bool' =None, full_story_end_date: 'datetime' =None, full_story_start_date: 'datetime' =None, general_teamspace: 'bool' =None, id: 'str' =None, last_storage_overuse_notification_sent_at: 'datetime' =None, location: 'str' =None, name: 'str' =None, preferred_cluster: 'str' =None, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, start_studios_on_spot_instance: 'bool' =None, storage_overuse_bytes: 'str' =None, storage_overuse_deletion_at: 'datetime' =None, storage_overuse_notification_count: 'int' =None, switch_to_default_machine_on_idle: 'bool' =None, teamspace_default_credits: 'float' =None, twitter_username: 'str' =None, updated_at: 'datetime' =None, workload_max_run_duration: 'str' =None):  # noqa: E501
+    def __init__(self, alerts_config: 'V1AlertsConfig' =None, allow_budgeting: 'bool' =None, allow_credits_auto_replenish: 'bool' =None, allow_external_project_duplication: 'bool' =None, allow_guest: 'bool' =None, allow_marketplace: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domain_validations: 'dict(str, V1AutoJoinDomainValidation)' =None, auto_join_domains: 'list[str]' =None, auto_replenish_amount: 'float' =None, auto_replenish_threshold: 'float' =None, auto_switch_machine: 'bool' =None, created_at: 'datetime' =None, default_machine_image_version: 'str' =None, default_machine_type: 'str' =None, default_project_id: 'str' =None, description: 'str' =None, disallow_aws_saas: 'bool' =None, disallow_dgx_saas: 'bool' =None, disallow_gcp_saas: 'bool' =None, disallow_lambda_saas: 'bool' =None, disallow_lightning_saas: 'bool' =None, disallow_nebius_saas: 'bool' =None, disallow_voltage_park_saas: 'bool' =None, disallow_vultr_saas: 'bool' =None, display_name: 'str' =None, domain: 'str' =None, email: 'str' =None, featured_gallery: 'bool' =None, full_story_end_date: 'datetime' =None, full_story_start_date: 'datetime' =None, general_teamspace: 'bool' =None, id: 'str' =None, last_storage_overuse_notification_sent_at: 'datetime' =None, location: 'str' =None, name: 'str' =None, preferred_cluster: 'str' =None, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, show_model_apis_tab: 'bool' =None, start_studios_on_spot_instance: 'bool' =None, storage_overuse_bytes: 'str' =None, storage_overuse_deletion_at: 'datetime' =None, storage_overuse_notification_count: 'int' =None, switch_to_default_machine_on_idle: 'bool' =None, teamspace_default_credits: 'float' =None, twitter_username: 'str' =None, updated_at: 'datetime' =None, workload_max_run_duration: 'str' =None):  # noqa: E501
         """V1Organization - a model defined in Swagger"""  # noqa: E501
         self._alerts_config = None
         self._allow_budgeting = None
@@ -189,6 +191,7 @@ class V1Organization(object):
         self._preferred_cluster = None
         self._preferred_deployment_provider = None
         self._preferred_studio_provider = None
+        self._show_model_apis_tab = None
         self._start_studios_on_spot_instance = None
         self._storage_overuse_bytes = None
         self._storage_overuse_deletion_at = None
@@ -281,6 +284,8 @@ class V1Organization(object):
             self.preferred_deployment_provider = preferred_deployment_provider
         if preferred_studio_provider is not None:
             self.preferred_studio_provider = preferred_studio_provider
+        if show_model_apis_tab is not None:
+            self.show_model_apis_tab = show_model_apis_tab
         if start_studios_on_spot_instance is not None:
             self.start_studios_on_spot_instance = start_studios_on_spot_instance
         if storage_overuse_bytes is not None:
@@ -1163,6 +1168,27 @@ class V1Organization(object):
 
         self._preferred_studio_provider = preferred_studio_provider
 
+    @property
+    def show_model_apis_tab(self) -> 'bool':
+        """Gets the show_model_apis_tab of this V1Organization.  # noqa: E501
+
+
+        :return: The show_model_apis_tab of this V1Organization.  # noqa: E501
+        :rtype: bool
+        """
+        return self._show_model_apis_tab
+
+    @show_model_apis_tab.setter
+    def show_model_apis_tab(self, show_model_apis_tab: 'bool'):
+        """Sets the show_model_apis_tab of this V1Organization.
+
+
+        :param show_model_apis_tab: The show_model_apis_tab of this V1Organization.  # noqa: E501
+        :type: bool
+        """
+
+        self._show_model_apis_tab = show_model_apis_tab
+
     @property
     def start_studios_on_spot_instance(self) -> 'bool':
         """Gets the start_studios_on_spot_instance of this V1Organization.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_update_cloud_space_instance_config_request.py

@@ -43,57 +43,42 @@ class V1UpdateCloudSpaceInstanceConfigRequest(object):
     swagger_types = {
         'compute_config': 'V1UserRequestedComputeConfig',
         'data_connection_mounts': 'list[V1DataConnectionMount]',
-        'disable_auto_shutdown': 'bool',
         'id': 'str',
         'ide': 'str',
-        'idle_shutdown_seconds': 'int',
         'port_plugin_port': 'str',
-        'project_id': 'str',
-        'switch_to_default_machine_on_idle': 'bool'
+        'project_id': 'str'
     }
 
     attribute_map = {
         'compute_config': 'computeConfig',
         'data_connection_mounts': 'dataConnectionMounts',
-        'disable_auto_shutdown': 'disableAutoShutdown',
         'id': 'id',
         'ide': 'ide',
-        'idle_shutdown_seconds': 'idleShutdownSeconds',
         'port_plugin_port': 'portPluginPort',
-        'project_id': 'projectId',
-        'switch_to_default_machine_on_idle': 'switchToDefaultMachineOnIdle'
+        'project_id': 'projectId'
     }
 
-    def __init__(self, compute_config: 'V1UserRequestedComputeConfig' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, disable_auto_shutdown: 'bool' =None, id: 'str' =None, ide: 'str' =None, idle_shutdown_seconds: 'int' =None, port_plugin_port: 'str' =None, project_id: 'str' =None, switch_to_default_machine_on_idle: 'bool' =None):  # noqa: E501
+    def __init__(self, compute_config: 'V1UserRequestedComputeConfig' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, id: 'str' =None, ide: 'str' =None, port_plugin_port: 'str' =None, project_id: 'str' =None):  # noqa: E501
         """V1UpdateCloudSpaceInstanceConfigRequest - a model defined in Swagger"""  # noqa: E501
         self._compute_config = None
         self._data_connection_mounts = None
-        self._disable_auto_shutdown = None
         self._id = None
         self._ide = None
-        self._idle_shutdown_seconds = None
         self._port_plugin_port = None
         self._project_id = None
-        self._switch_to_default_machine_on_idle = None
         self.discriminator = None
         if compute_config is not None:
             self.compute_config = compute_config
         if data_connection_mounts is not None:
             self.data_connection_mounts = data_connection_mounts
-        if disable_auto_shutdown is not None:
-            self.disable_auto_shutdown = disable_auto_shutdown
         if id is not None:
             self.id = id
         if ide is not None:
             self.ide = ide
-        if idle_shutdown_seconds is not None:
-            self.idle_shutdown_seconds = idle_shutdown_seconds
         if port_plugin_port is not None:
             self.port_plugin_port = port_plugin_port
         if project_id is not None:
             self.project_id = project_id
-        if switch_to_default_machine_on_idle is not None:
-            self.switch_to_default_machine_on_idle = switch_to_default_machine_on_idle
 
     @property
     def compute_config(self) -> 'V1UserRequestedComputeConfig':
@@ -137,27 +122,6 @@ class V1UpdateCloudSpaceInstanceConfigRequest(object):
 
         self._data_connection_mounts = data_connection_mounts
 
-    @property
-    def disable_auto_shutdown(self) -> 'bool':
-        """Gets the disable_auto_shutdown of this V1UpdateCloudSpaceInstanceConfigRequest.  # noqa: E501
-
-
-        :return: The disable_auto_shutdown of this V1UpdateCloudSpaceInstanceConfigRequest.  # noqa: E501
-        :rtype: bool
-        """
-        return self._disable_auto_shutdown
-
-    @disable_auto_shutdown.setter
-    def disable_auto_shutdown(self, disable_auto_shutdown: 'bool'):
-        """Sets the disable_auto_shutdown of this V1UpdateCloudSpaceInstanceConfigRequest.
-
-
-        :param disable_auto_shutdown: The disable_auto_shutdown of this V1UpdateCloudSpaceInstanceConfigRequest.  # noqa: E501
-        :type: bool
-        """
-
-        self._disable_auto_shutdown = disable_auto_shutdown
-
     @property
     def id(self) -> 'str':
         """Gets the id of this V1UpdateCloudSpaceInstanceConfigRequest.  # noqa: E501
@@ -200,27 +164,6 @@ class V1UpdateCloudSpaceInstanceConfigRequest(object):
 
         self._ide = ide
 
-    @property
-    def idle_shutdown_seconds(self) -> 'int':
-        """Gets the idle_shutdown_seconds of this V1UpdateCloudSpaceInstanceConfigRequest.  # noqa: E501
-
-
-        :return: The idle_shutdown_seconds of this V1UpdateCloudSpaceInstanceConfigRequest.  # noqa: E501
-        :rtype: int
-        """
-        return self._idle_shutdown_seconds
-
-    @idle_shutdown_seconds.setter
-    def idle_shutdown_seconds(self, idle_shutdown_seconds: 'int'):
-        """Sets the idle_shutdown_seconds of this V1UpdateCloudSpaceInstanceConfigRequest.
-
-
-        :param idle_shutdown_seconds: The idle_shutdown_seconds of this V1UpdateCloudSpaceInstanceConfigRequest.  # noqa: E501
-        :type: int
-        """
-
-        self._idle_shutdown_seconds = idle_shutdown_seconds
-
     @property
     def port_plugin_port(self) -> 'str':
         """Gets the port_plugin_port of this V1UpdateCloudSpaceInstanceConfigRequest.  # noqa: E501
@@ -263,27 +206,6 @@ class V1UpdateCloudSpaceInstanceConfigRequest(object):
 
         self._project_id = project_id
 
-    @property
-    def switch_to_default_machine_on_idle(self) -> 'bool':
-        """Gets the switch_to_default_machine_on_idle of this V1UpdateCloudSpaceInstanceConfigRequest.  # noqa: E501
-
-
-        :return: The switch_to_default_machine_on_idle of this V1UpdateCloudSpaceInstanceConfigRequest.  # noqa: E501
-        :rtype: bool
-        """
-        return self._switch_to_default_machine_on_idle
-
-    @switch_to_default_machine_on_idle.setter
-    def switch_to_default_machine_on_idle(self, switch_to_default_machine_on_idle: 'bool'):
-        """Sets the switch_to_default_machine_on_idle of this V1UpdateCloudSpaceInstanceConfigRequest.
-
-
-        :param switch_to_default_machine_on_idle: The switch_to_default_machine_on_idle of this V1UpdateCloudSpaceInstanceConfigRequest.  # noqa: E501
-        :type: bool
-        """
-
-        self._switch_to_default_machine_on_idle = switch_to_default_machine_on_idle
-
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -41,7 +41,6 @@ class V1UserFeatures(object):
                             and the value is json key in definition.
     """
     swagger_types = {
-        'add_data_v2': 'bool',
         'affiliate_links': 'bool',
         'agents_v2': 'bool',
         'ai_hub_monetization': 'bool',
@@ -75,7 +74,6 @@ class V1UserFeatures(object):
         'f241': 'bool',
         'f242': 'bool',
         'f243': 'bool',
-        'f244': 'bool',
         'f245': 'bool',
         'fair_share': 'bool',
         'featured_studios_admin': 'bool',
@@ -109,7 +107,6 @@ class V1UserFeatures(object):
         'product_license': 'bool',
         'project_selector': 'bool',
         'publish_pipelines': 'bool',
-        'r2_data_connections': 'bool',
         'reserved_machines_tab': 'bool',
         'restartable_jobs': 'bool',
         'runnable_public_studio_page': 'bool',
@@ -123,7 +120,6 @@ class V1UserFeatures(object):
         'studio_sharing_v2': 'bool',
         'studio_version_visibility': 'bool',
         'trainium2': 'bool',
-        'use_internal_data_connection_mounts': 'bool',
         'use_rclone_mounts_only': 'bool',
         'vultr': 'bool',
         'weka': 'bool',
@@ -131,7 +127,6 @@ class V1UserFeatures(object):
     }
 
     attribute_map = {
-        'add_data_v2': 'addDataV2',
         'affiliate_links': 'affiliateLinks',
         'agents_v2': 'agentsV2',
         'ai_hub_monetization': 'aiHubMonetization',
@@ -165,7 +160,6 @@ class V1UserFeatures(object):
         'f241': 'f241',
         'f242': 'f242',
         'f243': 'f243',
-        'f244': 'f244',
         'f245': 'f245',
         'fair_share': 'fairShare',
         'featured_studios_admin': 'featuredStudiosAdmin',
@@ -199,7 +193,6 @@ class V1UserFeatures(object):
         'product_license': 'productLicense',
         'project_selector': 'projectSelector',
         'publish_pipelines': 'publishPipelines',
-        'r2_data_connections': 'r2DataConnections',
         'reserved_machines_tab': 'reservedMachinesTab',
         'restartable_jobs': 'restartableJobs',
         'runnable_public_studio_page': 'runnablePublicStudioPage',
@@ -213,16 +206,14 @@ class V1UserFeatures(object):
         'studio_sharing_v2': 'studioSharingV2',
         'studio_version_visibility': 'studioVersionVisibility',
         'trainium2': 'trainium2',
-        'use_internal_data_connection_mounts': 'useInternalDataConnectionMounts',
         'use_rclone_mounts_only': 'useRcloneMountsOnly',
         'vultr': 'vultr',
         'weka': 'weka',
         'writable_s3_connections': 'writableS3Connections'
     }
 
-    def __init__(self, add_data_v2: '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, byo_machine_type: '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, cloudspace_schedules: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: '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, f227: 'bool' =None, f234: 'bool' =None, f236: 'bool' =None, f237: 'bool' =None, f238: 'bool' =None, f239: 'bool' =None, f240: 'bool' =None, f241: 'bool' =None, f242: 'bool' =None, f243: 'bool' =None, f244: 'bool' =None, f245: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcp_overprovisioning: 'bool' =None, gcs_connections_optimized: 'bool' =None, gcs_folders: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, kubernetes_cluster_ui: 'bool' =None, kubernetes_clusters: '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, multiple_studio_versions: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, onboarding_v2: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, persistent_disk: '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_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, specialised_studios: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_sharing_v2: 'bool' =None, studio_version_visibility: 'bool' =None, trainium2: 'bool' =None, use_internal_data_connection_mounts: 'bool' =None, use_rclone_mounts_only: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None, writable_s3_connections: 'bool' =None):  # noqa: E501
+    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, byo_machine_type: '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, cloudspace_schedules: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: '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, f227: 'bool' =None, f234: 'bool' =None, f236: 'bool' =None, f237: 'bool' =None, f238: 'bool' =None, f239: 'bool' =None, f240: 'bool' =None, f241: 'bool' =None, f242: 'bool' =None, f243: 'bool' =None, f245: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcp_overprovisioning: 'bool' =None, gcs_connections_optimized: 'bool' =None, gcs_folders: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, kubernetes_cluster_ui: 'bool' =None, kubernetes_clusters: '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, multiple_studio_versions: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, onboarding_v2: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, persistent_disk: '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_sweeps: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, product_license: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: '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, specialised_studios: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_sharing_v2: 'bool' =None, studio_version_visibility: 'bool' =None, trainium2: 'bool' =None, use_rclone_mounts_only: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None, writable_s3_connections: 'bool' =None):  # noqa: E501
         """V1UserFeatures - a model defined in Swagger"""  # noqa: E501
-        self._add_data_v2 = None
         self._affiliate_links = None
         self._agents_v2 = None
         self._ai_hub_monetization = None
@@ -256,7 +247,6 @@ class V1UserFeatures(object):
         self._f241 = None
         self._f242 = None
         self._f243 = None
-        self._f244 = None
         self._f245 = None
         self._fair_share = None
         self._featured_studios_admin = None
@@ -290,7 +280,6 @@ class V1UserFeatures(object):
         self._product_license = None
         self._project_selector = None
         self._publish_pipelines = None
-        self._r2_data_connections = None
         self._reserved_machines_tab = None
         self._restartable_jobs = None
         self._runnable_public_studio_page = None
@@ -304,14 +293,11 @@ class V1UserFeatures(object):
         self._studio_sharing_v2 = None
         self._studio_version_visibility = None
         self._trainium2 = None
-        self._use_internal_data_connection_mounts = None
         self._use_rclone_mounts_only = None
         self._vultr = None
         self._weka = None
         self._writable_s3_connections = None
         self.discriminator = None
-        if add_data_v2 is not None:
-            self.add_data_v2 = add_data_v2
         if affiliate_links is not None:
             self.affiliate_links = affiliate_links
         if agents_v2 is not None:
@@ -378,8 +364,6 @@ class V1UserFeatures(object):
             self.f242 = f242
         if f243 is not None:
             self.f243 = f243
-        if f244 is not None:
-            self.f244 = f244
         if f245 is not None:
             self.f245 = f245
         if fair_share is not None:
@@ -446,8 +430,6 @@ class V1UserFeatures(object):
             self.project_selector = project_selector
         if publish_pipelines is not None:
             self.publish_pipelines = publish_pipelines
-        if r2_data_connections is not None:
-            self.r2_data_connections = r2_data_connections
         if reserved_machines_tab is not None:
             self.reserved_machines_tab = reserved_machines_tab
         if restartable_jobs is not None:
@@ -474,8 +456,6 @@ class V1UserFeatures(object):
             self.studio_version_visibility = studio_version_visibility
         if trainium2 is not None:
             self.trainium2 = trainium2
-        if use_internal_data_connection_mounts is not None:
-            self.use_internal_data_connection_mounts = use_internal_data_connection_mounts
         if use_rclone_mounts_only is not None:
             self.use_rclone_mounts_only = use_rclone_mounts_only
         if vultr is not None:
@@ -485,27 +465,6 @@ class V1UserFeatures(object):
         if writable_s3_connections is not None:
             self.writable_s3_connections = writable_s3_connections
 
-    @property
-    def add_data_v2(self) -> 'bool':
-        """Gets the add_data_v2 of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The add_data_v2 of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._add_data_v2
-
-    @add_data_v2.setter
-    def add_data_v2(self, add_data_v2: 'bool'):
-        """Sets the add_data_v2 of this V1UserFeatures.
-
-
-        :param add_data_v2: The add_data_v2 of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._add_data_v2 = add_data_v2
-
     @property
     def affiliate_links(self) -> 'bool':
         """Gets the affiliate_links of this V1UserFeatures.  # noqa: E501
@@ -1199,27 +1158,6 @@ class V1UserFeatures(object):
 
         self._f243 = f243
 
-    @property
-    def f244(self) -> 'bool':
-        """Gets the f244 of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The f244 of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._f244
-
-    @f244.setter
-    def f244(self, f244: 'bool'):
-        """Sets the f244 of this V1UserFeatures.
-
-
-        :param f244: The f244 of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._f244 = f244
-
     @property
     def f245(self) -> 'bool':
         """Gets the f245 of this V1UserFeatures.  # noqa: E501
@@ -1913,27 +1851,6 @@ class V1UserFeatures(object):
 
         self._publish_pipelines = publish_pipelines
 
-    @property
-    def r2_data_connections(self) -> 'bool':
-        """Gets the r2_data_connections of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The r2_data_connections of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._r2_data_connections
-
-    @r2_data_connections.setter
-    def r2_data_connections(self, r2_data_connections: 'bool'):
-        """Sets the r2_data_connections of this V1UserFeatures.
-
-
-        :param r2_data_connections: The r2_data_connections of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._r2_data_connections = r2_data_connections
-
     @property
     def reserved_machines_tab(self) -> 'bool':
         """Gets the reserved_machines_tab of this V1UserFeatures.  # noqa: E501
@@ -2207,27 +2124,6 @@ class V1UserFeatures(object):
 
         self._trainium2 = trainium2
 
-    @property
-    def use_internal_data_connection_mounts(self) -> 'bool':
-        """Gets the use_internal_data_connection_mounts of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The use_internal_data_connection_mounts of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._use_internal_data_connection_mounts
-
-    @use_internal_data_connection_mounts.setter
-    def use_internal_data_connection_mounts(self, use_internal_data_connection_mounts: 'bool'):
-        """Sets the use_internal_data_connection_mounts of this V1UserFeatures.
-
-
-        :param use_internal_data_connection_mounts: The use_internal_data_connection_mounts of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._use_internal_data_connection_mounts = use_internal_data_connection_mounts
-
     @property
     def use_rclone_mounts_only(self) -> 'bool':
         """Gets the use_rclone_mounts_only of this V1UserFeatures.  # noqa: E501

lightning_sdk/studio.py

@@ -81,7 +81,6 @@ class Studio:
             raise ValueError("Couldn't resolve teamspace from the provided name, org, or user")
 
         self._teamspace = _teamspace
-        self._cloud_account = _resolve_deprecated_cluster(cloud_account, cluster)
 
         self._setup_done = self._skip_setup
         self._disable_secrets = disable_secrets

lightning_sdk/utils/config.py

@@ -23,6 +23,9 @@ class DefaultConfigKeys:
 
     studio: str = "studio.name"
 
+    cloud_account: str = "cloud_account.name"
+    cloud_provider: str = "cloud_provider.name"
+
 
 class ConfigProxy:
     def __init__(self, root: "Config", *path: str) -> None:

lightning_sdk/utils/resolve.py

@@ -66,6 +66,12 @@ def _resolve_deprecated_provider(
         )
         return provider
 
+    if cloud_provider is None:
+        from lightning_sdk.utils.config import Config, DefaultConfigKeys
+
+        config = Config()
+        cloud_provider = config.get_value(DefaultConfigKeys.cloud_provider)
+
     return cloud_provider
 
 
@@ -84,6 +90,12 @@ def _resolve_deprecated_cluster(cloud_account: Optional[str], cluster: Optional[
         )
         return cluster
 
+    if cloud_account is None:
+        from lightning_sdk.utils.config import Config, DefaultConfigKeys
+
+        config = Config()
+        cloud_account = config.get_value(DefaultConfigKeys.cloud_account)
+
     return cloud_account