lightning-sdk 2025.12.17__py3-none-any.whl → 2026.1.22__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_capacity_reservation_used_by.py

@@ -0,0 +1,227 @@
+# coding: utf-8
+
+"""
+    external/v1/auth_service.proto
+
+    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
+
+    OpenAPI spec version: version not set
+    
+    Generated by: https://github.com/swagger-api/swagger-codegen.git
+
+    NOTE
+    ----
+    standard swagger-codegen-cli for this python client has been modified
+    by custom templates. The purpose of these templates is to include
+    typing information in the API and Model code. Please refer to the
+    main grid repository for more info
+"""
+
+import pprint
+import re  # noqa: F401
+
+from typing import TYPE_CHECKING
+
+import six
+
+if TYPE_CHECKING:
+    from datetime import datetime
+    from lightning_sdk.lightning_cloud.openapi.models import *
+
+class V1CapacityReservationUsedBy(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+    """
+    Attributes:
+      swagger_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    swagger_types = {
+        'cloudspace_id': 'str',
+        'job_id': 'str',
+        'project_id': 'str',
+        'server_id': 'str',
+        'user_id': 'str'
+    }
+
+    attribute_map = {
+        'cloudspace_id': 'cloudspaceId',
+        'job_id': 'jobId',
+        'project_id': 'projectId',
+        'server_id': 'serverId',
+        'user_id': 'userId'
+    }
+
+    def __init__(self, cloudspace_id: 'str' =None, job_id: 'str' =None, project_id: 'str' =None, server_id: 'str' =None, user_id: 'str' =None):  # noqa: E501
+        """V1CapacityReservationUsedBy - a model defined in Swagger"""  # noqa: E501
+        self._cloudspace_id = None
+        self._job_id = None
+        self._project_id = None
+        self._server_id = None
+        self._user_id = None
+        self.discriminator = None
+        if cloudspace_id is not None:
+            self.cloudspace_id = cloudspace_id
+        if job_id is not None:
+            self.job_id = job_id
+        if project_id is not None:
+            self.project_id = project_id
+        if server_id is not None:
+            self.server_id = server_id
+        if user_id is not None:
+            self.user_id = user_id
+
+    @property
+    def cloudspace_id(self) -> 'str':
+        """Gets the cloudspace_id of this V1CapacityReservationUsedBy.  # noqa: E501
+
+
+        :return: The cloudspace_id of this V1CapacityReservationUsedBy.  # noqa: E501
+        :rtype: str
+        """
+        return self._cloudspace_id
+
+    @cloudspace_id.setter
+    def cloudspace_id(self, cloudspace_id: 'str'):
+        """Sets the cloudspace_id of this V1CapacityReservationUsedBy.
+
+
+        :param cloudspace_id: The cloudspace_id of this V1CapacityReservationUsedBy.  # noqa: E501
+        :type: str
+        """
+
+        self._cloudspace_id = cloudspace_id
+
+    @property
+    def job_id(self) -> 'str':
+        """Gets the job_id of this V1CapacityReservationUsedBy.  # noqa: E501
+
+
+        :return: The job_id of this V1CapacityReservationUsedBy.  # noqa: E501
+        :rtype: str
+        """
+        return self._job_id
+
+    @job_id.setter
+    def job_id(self, job_id: 'str'):
+        """Sets the job_id of this V1CapacityReservationUsedBy.
+
+
+        :param job_id: The job_id of this V1CapacityReservationUsedBy.  # noqa: E501
+        :type: str
+        """
+
+        self._job_id = job_id
+
+    @property
+    def project_id(self) -> 'str':
+        """Gets the project_id of this V1CapacityReservationUsedBy.  # noqa: E501
+
+
+        :return: The project_id of this V1CapacityReservationUsedBy.  # noqa: E501
+        :rtype: str
+        """
+        return self._project_id
+
+    @project_id.setter
+    def project_id(self, project_id: 'str'):
+        """Sets the project_id of this V1CapacityReservationUsedBy.
+
+
+        :param project_id: The project_id of this V1CapacityReservationUsedBy.  # noqa: E501
+        :type: str
+        """
+
+        self._project_id = project_id
+
+    @property
+    def server_id(self) -> 'str':
+        """Gets the server_id of this V1CapacityReservationUsedBy.  # noqa: E501
+
+
+        :return: The server_id of this V1CapacityReservationUsedBy.  # noqa: E501
+        :rtype: str
+        """
+        return self._server_id
+
+    @server_id.setter
+    def server_id(self, server_id: 'str'):
+        """Sets the server_id of this V1CapacityReservationUsedBy.
+
+
+        :param server_id: The server_id of this V1CapacityReservationUsedBy.  # noqa: E501
+        :type: str
+        """
+
+        self._server_id = server_id
+
+    @property
+    def user_id(self) -> 'str':
+        """Gets the user_id of this V1CapacityReservationUsedBy.  # noqa: E501
+
+
+        :return: The user_id of this V1CapacityReservationUsedBy.  # noqa: E501
+        :rtype: str
+        """
+        return self._user_id
+
+    @user_id.setter
+    def user_id(self, user_id: 'str'):
+        """Sets the user_id of this V1CapacityReservationUsedBy.
+
+
+        :param user_id: The user_id of this V1CapacityReservationUsedBy.  # noqa: E501
+        :type: str
+        """
+
+        self._user_id = user_id
+
+    def to_dict(self) -> dict:
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.swagger_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+        if issubclass(V1CapacityReservationUsedBy, dict):
+            for key, value in self.items():
+                result[key] = value
+
+        return result
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self) -> str:
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other: 'V1CapacityReservationUsedBy') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1CapacityReservationUsedBy):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1CapacityReservationUsedBy') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py

@@ -50,9 +50,9 @@ class V1CloudProvider(object):
     LIGHTNING = "LIGHTNING"
     LIGHTNING_AGGREGATE = "LIGHTNING_AGGREGATE"
     KUBERNETES = "KUBERNETES"
-    AI_POD = "AI_POD"
     MACHINE = "MACHINE"
     LIGHTNING_ELASTIC_CLUSTER_AGGREGATE = "LIGHTNING_ELASTIC_CLUSTER_AGGREGATE"
+    CUDO = "CUDO"
     """
     Attributes:
       swagger_types (dict): The key is attribute name

lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_type.py

@@ -41,6 +41,7 @@ class V1CloudSpaceEnvironmentType(object):
     DEFAULT = "CLOUD_SPACE_ENVIRONMENT_TYPE_DEFAULT"
     BLANK = "CLOUD_SPACE_ENVIRONMENT_TYPE_BLANK"
     PYTHON_EMPTY = "CLOUD_SPACE_ENVIRONMENT_TYPE_PYTHON_EMPTY"
+    COMFYUI = "CLOUD_SPACE_ENVIRONMENT_TYPE_COMFYUI"
     """
     Attributes:
       swagger_types (dict): The key is attribute name

lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py

@@ -73,6 +73,7 @@ class V1ClusterAccelerator(object):
         'max_available_quota': 'str',
         'non_spot': 'bool',
         'out_of_capacity': 'bool',
+        'partitionable_parent': 'str',
         'persistent_disk_supported': 'bool',
         'provider': 'V1CloudProvider',
         'quota_checked_at': 'datetime',
@@ -130,6 +131,7 @@ class V1ClusterAccelerator(object):
         'max_available_quota': 'maxAvailableQuota',
         'non_spot': 'nonSpot',
         'out_of_capacity': 'outOfCapacity',
+        'partitionable_parent': 'partitionableParent',
         'persistent_disk_supported': 'persistentDiskSupported',
         'provider': 'provider',
         'quota_checked_at': 'quotaCheckedAt',
@@ -154,7 +156,7 @@ class V1ClusterAccelerator(object):
         'spot_quota_page_url': 'spotQuotaPageUrl'
     }
 
-    def __init__(self, accelerator_type: 'str' =None, allowed_resources: 'list[str]' =None, availability_strategy: 'str' =None, available_in_seconds: 'str' =None, available_in_seconds_spot: 'str' =None, available_zones: 'list[str]' =None, byoc_only: 'bool' =None, capacity_block_only: 'bool' =None, capacity_block_price: 'float' =None, capacity_blocks_available: 'list[V1ClusterCapacityReservation]' =None, cluster_id: 'str' =None, cost: 'float' =None, detailed_quotas_info: 'list[V1AcceleratorQuotaInfo]' =None, device_card: 'str' =None, device_info: 'str' =None, display_name: 'str' =None, dws_cost: 'float' =None, dws_only: 'bool' =None, dws_supported: 'bool' =None, enabled: 'bool' =None, family: 'str' =None, instance_id: 'str' =None, is_custom: 'bool' =None, is_tier_restricted: 'bool' =None, lightning_interruptible: 'bool' =None, local_disk_included: 'bool' =None, local_disk_size: 'str' =None, local_disk_supported: 'bool' =None, local_disks_count: 'str' =None, max_available_quota: 'str' =None, non_spot: 'bool' =None, out_of_capacity: 'bool' =None, persistent_disk_supported: 'bool' =None, provider: 'V1CloudProvider' =None, quota_checked_at: 'datetime' =None, quota_code: 'str' =None, quota_name: 'str' =None, quota_page_url: 'str' =None, quota_service_code: 'str' =None, quota_utilization: 'str' =None, quota_value: 'str' =None, reservable: 'bool' =None, reservation_available_zones: 'list[str]' =None, reservation_quota_code: 'str' =None, reservation_quota_name: 'str' =None, reservation_quota_page_url: 'str' =None, resources: 'V1Resources' =None, secondary_instance_id: 'str' =None, slug: 'str' =None, slug_multi_cloud: 'str' =None, spot_price: 'float' =None, spot_quota_code: 'str' =None, spot_quota_name: 'str' =None, spot_quota_page_url: 'str' =None):  # noqa: E501
+    def __init__(self, accelerator_type: 'str' =None, allowed_resources: 'list[str]' =None, availability_strategy: 'str' =None, available_in_seconds: 'str' =None, available_in_seconds_spot: 'str' =None, available_zones: 'list[str]' =None, byoc_only: 'bool' =None, capacity_block_only: 'bool' =None, capacity_block_price: 'float' =None, capacity_blocks_available: 'list[V1ClusterCapacityReservation]' =None, cluster_id: 'str' =None, cost: 'float' =None, detailed_quotas_info: 'list[V1AcceleratorQuotaInfo]' =None, device_card: 'str' =None, device_info: 'str' =None, display_name: 'str' =None, dws_cost: 'float' =None, dws_only: 'bool' =None, dws_supported: 'bool' =None, enabled: 'bool' =None, family: 'str' =None, instance_id: 'str' =None, is_custom: 'bool' =None, is_tier_restricted: 'bool' =None, lightning_interruptible: 'bool' =None, local_disk_included: 'bool' =None, local_disk_size: 'str' =None, local_disk_supported: 'bool' =None, local_disks_count: 'str' =None, max_available_quota: 'str' =None, non_spot: 'bool' =None, out_of_capacity: 'bool' =None, partitionable_parent: 'str' =None, persistent_disk_supported: 'bool' =None, provider: 'V1CloudProvider' =None, quota_checked_at: 'datetime' =None, quota_code: 'str' =None, quota_name: 'str' =None, quota_page_url: 'str' =None, quota_service_code: 'str' =None, quota_utilization: 'str' =None, quota_value: 'str' =None, reservable: 'bool' =None, reservation_available_zones: 'list[str]' =None, reservation_quota_code: 'str' =None, reservation_quota_name: 'str' =None, reservation_quota_page_url: 'str' =None, resources: 'V1Resources' =None, secondary_instance_id: 'str' =None, slug: 'str' =None, slug_multi_cloud: 'str' =None, spot_price: 'float' =None, spot_quota_code: 'str' =None, spot_quota_name: 'str' =None, spot_quota_page_url: 'str' =None):  # noqa: E501
         """V1ClusterAccelerator - a model defined in Swagger"""  # noqa: E501
         self._accelerator_type = None
         self._allowed_resources = None
@@ -188,6 +190,7 @@ class V1ClusterAccelerator(object):
         self._max_available_quota = None
         self._non_spot = None
         self._out_of_capacity = None
+        self._partitionable_parent = None
         self._persistent_disk_supported = None
         self._provider = None
         self._quota_checked_at = None
@@ -275,6 +278,8 @@ class V1ClusterAccelerator(object):
             self.non_spot = non_spot
         if out_of_capacity is not None:
             self.out_of_capacity = out_of_capacity
+        if partitionable_parent is not None:
+            self.partitionable_parent = partitionable_parent
         if persistent_disk_supported is not None:
             self.persistent_disk_supported = persistent_disk_supported
         if provider is not None:
@@ -992,6 +997,27 @@ class V1ClusterAccelerator(object):
 
         self._out_of_capacity = out_of_capacity
 
+    @property
+    def partitionable_parent(self) -> 'str':
+        """Gets the partitionable_parent of this V1ClusterAccelerator.  # noqa: E501
+
+
+        :return: The partitionable_parent of this V1ClusterAccelerator.  # noqa: E501
+        :rtype: str
+        """
+        return self._partitionable_parent
+
+    @partitionable_parent.setter
+    def partitionable_parent(self, partitionable_parent: 'str'):
+        """Sets the partitionable_parent of this V1ClusterAccelerator.
+
+
+        :param partitionable_parent: The partitionable_parent of this V1ClusterAccelerator.  # noqa: E501
+        :type: str
+        """
+
+        self._partitionable_parent = partitionable_parent
+
     @property
     def persistent_disk_supported(self) -> 'bool':
         """Gets the persistent_disk_supported of this V1ClusterAccelerator.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py

@@ -57,6 +57,7 @@ class V1ClusterCapacityReservation(object):
         'project_id': 'str',
         'region': 'str',
         'start_time': 'datetime',
+        'used_by': 'list[V1CapacityReservationUsedBy]',
         'zone': 'str'
     }
 
@@ -77,10 +78,11 @@ class V1ClusterCapacityReservation(object):
         'project_id': 'projectId',
         'region': 'region',
         'start_time': 'startTime',
+        'used_by': 'usedBy',
         'zone': 'zone'
     }
 
-    def __init__(self, aggregate_availability: 'str' =None, capacity_reservation_type: 'str' =None, cloud_provider_capacity_reservation_id: 'str' =None, cluster_id: 'str' =None, end_time: 'datetime' =None, full_cloud_provider_reservation_string: 'str' =None, id: 'str' =None, in_use: 'str' =None, in_use_aggregate: 'str' =None, instance_type: 'str' =None, match_pattern: 'str' =None, node_group_name: 'str' =None, num_instances: 'str' =None, project_id: 'str' =None, region: 'str' =None, start_time: 'datetime' =None, zone: 'str' =None):  # noqa: E501
+    def __init__(self, aggregate_availability: 'str' =None, capacity_reservation_type: 'str' =None, cloud_provider_capacity_reservation_id: 'str' =None, cluster_id: 'str' =None, end_time: 'datetime' =None, full_cloud_provider_reservation_string: 'str' =None, id: 'str' =None, in_use: 'str' =None, in_use_aggregate: 'str' =None, instance_type: 'str' =None, match_pattern: 'str' =None, node_group_name: 'str' =None, num_instances: 'str' =None, project_id: 'str' =None, region: 'str' =None, start_time: 'datetime' =None, used_by: 'list[V1CapacityReservationUsedBy]' =None, zone: 'str' =None):  # noqa: E501
         """V1ClusterCapacityReservation - a model defined in Swagger"""  # noqa: E501
         self._aggregate_availability = None
         self._capacity_reservation_type = None
@@ -98,6 +100,7 @@ class V1ClusterCapacityReservation(object):
         self._project_id = None
         self._region = None
         self._start_time = None
+        self._used_by = None
         self._zone = None
         self.discriminator = None
         if aggregate_availability is not None:
@@ -132,6 +135,8 @@ class V1ClusterCapacityReservation(object):
             self.region = region
         if start_time is not None:
             self.start_time = start_time
+        if used_by is not None:
+            self.used_by = used_by
         if zone is not None:
             self.zone = zone
 
@@ -471,6 +476,27 @@ class V1ClusterCapacityReservation(object):
 
         self._start_time = start_time
 
+    @property
+    def used_by(self) -> 'list[V1CapacityReservationUsedBy]':
+        """Gets the used_by of this V1ClusterCapacityReservation.  # noqa: E501
+
+
+        :return: The used_by of this V1ClusterCapacityReservation.  # noqa: E501
+        :rtype: list[V1CapacityReservationUsedBy]
+        """
+        return self._used_by
+
+    @used_by.setter
+    def used_by(self, used_by: 'list[V1CapacityReservationUsedBy]'):
+        """Sets the used_by of this V1ClusterCapacityReservation.
+
+
+        :param used_by: The used_by of this V1ClusterCapacityReservation.  # noqa: E501
+        :type: list[V1CapacityReservationUsedBy]
+        """
+
+        self._used_by = used_by
+
     @property
     def zone(self) -> 'str':
         """Gets the zone of this V1ClusterCapacityReservation.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py

@@ -41,7 +41,6 @@ class V1ClusterSpec(object):
                             and the value is json key in definition.
     """
     swagger_types = {
-        'ai_pod_v1': 'V1AiPodV1',
         'auth_token': 'str',
         'available_accelerators': 'list[str]',
         'aws_v1': 'V1AWSDirectV1',
@@ -50,6 +49,7 @@ class V1ClusterSpec(object):
         'cluster_type': 'V1ClusterType',
         'compute_cluster_ids': 'list[str]',
         'compute_cluster_request': 'V1ComputeClusterRequest',
+        'cudo_v1': 'V1CudoDirectV1',
         'deletion_options': 'V1ClusterDeletionOptions',
         'desired_state': 'V1ClusterState',
         'domain': 'str',
@@ -68,6 +68,7 @@ class V1ClusterSpec(object):
         'overprovisioning': 'list[V1InstanceOverprovisioningSpec]',
         'parent_cluster_id': 'str',
         'parent_cluster_type': 'str',
+        'partitioning_enabled': 'bool',
         'pause_automation': 'bool',
         'reservation_details': 'V1ReservationDetails',
         'reserved_capacity_provider': 'bool',
@@ -83,7 +84,6 @@ class V1ClusterSpec(object):
     }
 
     attribute_map = {
-        'ai_pod_v1': 'aiPodV1',
         'auth_token': 'authToken',
         'available_accelerators': 'availableAccelerators',
         'aws_v1': 'awsV1',
@@ -92,6 +92,7 @@ class V1ClusterSpec(object):
         'cluster_type': 'clusterType',
         'compute_cluster_ids': 'computeClusterIds',
         'compute_cluster_request': 'computeClusterRequest',
+        'cudo_v1': 'cudoV1',
         'deletion_options': 'deletionOptions',
         'desired_state': 'desiredState',
         'domain': 'domain',
@@ -110,6 +111,7 @@ class V1ClusterSpec(object):
         'overprovisioning': 'overprovisioning',
         'parent_cluster_id': 'parentClusterId',
         'parent_cluster_type': 'parentClusterType',
+        'partitioning_enabled': 'partitioningEnabled',
         'pause_automation': 'pauseAutomation',
         'reservation_details': 'reservationDetails',
         'reserved_capacity_provider': 'reservedCapacityProvider',
@@ -124,9 +126,8 @@ class V1ClusterSpec(object):
         'vultr_v1': 'vultrV1'
     }
 
-    def __init__(self, ai_pod_v1: 'V1AiPodV1' =None, auth_token: 'str' =None, available_accelerators: 'list[str]' =None, aws_v1: 'V1AWSDirectV1' =None, cloud_pricing_enabled: 'bool' =None, cloudflare_v1: 'V1CloudflareV1' =None, cluster_type: 'V1ClusterType' =None, compute_cluster_ids: 'list[str]' =None, compute_cluster_request: 'V1ComputeClusterRequest' =None, deletion_options: 'V1ClusterDeletionOptions' =None, desired_state: 'V1ClusterState' =None, domain: 'str' =None, driver: 'V1CloudProvider' =None, freeze_accelerators: 'bool' =None, google_cloud_v1: 'V1GoogleCloudDirectV1' =None, insurer_disabled: 'bool' =None, kubernetes_v1: 'V1KubernetesDirectV1' =None, lambda_labs_v1: 'V1LambdaLabsDirectV1' =None, lightning_elastic_cluster_v1: 'V1LightningElasticClusterV1' =None, lock_overprovisioning: 'bool' =None, locked_zones: 'list[str]' =None, machine_v1: 'V1MachineDirectV1' =None, monitor_deletion_disabled: 'bool' =None, nebius_v1: 'V1NebiusDirectV1' =None, overprovisioning: 'list[V1InstanceOverprovisioningSpec]' =None, parent_cluster_id: 'str' =None, parent_cluster_type: 'str' =None, pause_automation: 'bool' =None, reservation_details: 'V1ReservationDetails' =None, reserved_capacity_provider: 'bool' =None, reserved_instances_only: 'bool' =None, security_options: 'V1ClusterSecurityOptions' =None, slurm_v1: 'V1SlurmV1' =None, tagging_options: 'V1ClusterTaggingOptions' =None, unavailability_spikes_detection_enabled: 'bool' =None, user_id: 'str' =None, vibe_coding_enabled: 'bool' =None, voltage_park_v1: 'V1VoltageParkDirectV1' =None, vultr_v1: 'V1VultrDirectV1' =None):  # noqa: E501
+    def __init__(self, auth_token: 'str' =None, available_accelerators: 'list[str]' =None, aws_v1: 'V1AWSDirectV1' =None, cloud_pricing_enabled: 'bool' =None, cloudflare_v1: 'V1CloudflareV1' =None, cluster_type: 'V1ClusterType' =None, compute_cluster_ids: 'list[str]' =None, compute_cluster_request: 'V1ComputeClusterRequest' =None, cudo_v1: 'V1CudoDirectV1' =None, deletion_options: 'V1ClusterDeletionOptions' =None, desired_state: 'V1ClusterState' =None, domain: 'str' =None, driver: 'V1CloudProvider' =None, freeze_accelerators: 'bool' =None, google_cloud_v1: 'V1GoogleCloudDirectV1' =None, insurer_disabled: 'bool' =None, kubernetes_v1: 'V1KubernetesDirectV1' =None, lambda_labs_v1: 'V1LambdaLabsDirectV1' =None, lightning_elastic_cluster_v1: 'V1LightningElasticClusterV1' =None, lock_overprovisioning: 'bool' =None, locked_zones: 'list[str]' =None, machine_v1: 'V1MachineDirectV1' =None, monitor_deletion_disabled: 'bool' =None, nebius_v1: 'V1NebiusDirectV1' =None, overprovisioning: 'list[V1InstanceOverprovisioningSpec]' =None, parent_cluster_id: 'str' =None, parent_cluster_type: 'str' =None, partitioning_enabled: 'bool' =None, pause_automation: 'bool' =None, reservation_details: 'V1ReservationDetails' =None, reserved_capacity_provider: 'bool' =None, reserved_instances_only: 'bool' =None, security_options: 'V1ClusterSecurityOptions' =None, slurm_v1: 'V1SlurmV1' =None, tagging_options: 'V1ClusterTaggingOptions' =None, unavailability_spikes_detection_enabled: 'bool' =None, user_id: 'str' =None, vibe_coding_enabled: 'bool' =None, voltage_park_v1: 'V1VoltageParkDirectV1' =None, vultr_v1: 'V1VultrDirectV1' =None):  # noqa: E501
         """V1ClusterSpec - a model defined in Swagger"""  # noqa: E501
-        self._ai_pod_v1 = None
         self._auth_token = None
         self._available_accelerators = None
         self._aws_v1 = None
@@ -135,6 +136,7 @@ class V1ClusterSpec(object):
         self._cluster_type = None
         self._compute_cluster_ids = None
         self._compute_cluster_request = None
+        self._cudo_v1 = None
         self._deletion_options = None
         self._desired_state = None
         self._domain = None
@@ -153,6 +155,7 @@ class V1ClusterSpec(object):
         self._overprovisioning = None
         self._parent_cluster_id = None
         self._parent_cluster_type = None
+        self._partitioning_enabled = None
         self._pause_automation = None
         self._reservation_details = None
         self._reserved_capacity_provider = None
@@ -166,8 +169,6 @@ class V1ClusterSpec(object):
         self._voltage_park_v1 = None
         self._vultr_v1 = None
         self.discriminator = None
-        if ai_pod_v1 is not None:
-            self.ai_pod_v1 = ai_pod_v1
         if auth_token is not None:
             self.auth_token = auth_token
         if available_accelerators is not None:
@@ -184,6 +185,8 @@ class V1ClusterSpec(object):
             self.compute_cluster_ids = compute_cluster_ids
         if compute_cluster_request is not None:
             self.compute_cluster_request = compute_cluster_request
+        if cudo_v1 is not None:
+            self.cudo_v1 = cudo_v1
         if deletion_options is not None:
             self.deletion_options = deletion_options
         if desired_state is not None:
@@ -220,6 +223,8 @@ class V1ClusterSpec(object):
             self.parent_cluster_id = parent_cluster_id
         if parent_cluster_type is not None:
             self.parent_cluster_type = parent_cluster_type
+        if partitioning_enabled is not None:
+            self.partitioning_enabled = partitioning_enabled
         if pause_automation is not None:
             self.pause_automation = pause_automation
         if reservation_details is not None:
@@ -245,27 +250,6 @@ class V1ClusterSpec(object):
         if vultr_v1 is not None:
             self.vultr_v1 = vultr_v1
 
-    @property
-    def ai_pod_v1(self) -> 'V1AiPodV1':
-        """Gets the ai_pod_v1 of this V1ClusterSpec.  # noqa: E501
-
-
-        :return: The ai_pod_v1 of this V1ClusterSpec.  # noqa: E501
-        :rtype: V1AiPodV1
-        """
-        return self._ai_pod_v1
-
-    @ai_pod_v1.setter
-    def ai_pod_v1(self, ai_pod_v1: 'V1AiPodV1'):
-        """Sets the ai_pod_v1 of this V1ClusterSpec.
-
-
-        :param ai_pod_v1: The ai_pod_v1 of this V1ClusterSpec.  # noqa: E501
-        :type: V1AiPodV1
-        """
-
-        self._ai_pod_v1 = ai_pod_v1
-
     @property
     def auth_token(self) -> 'str':
         """Gets the auth_token of this V1ClusterSpec.  # noqa: E501
@@ -434,6 +418,27 @@ class V1ClusterSpec(object):
 
         self._compute_cluster_request = compute_cluster_request
 
+    @property
+    def cudo_v1(self) -> 'V1CudoDirectV1':
+        """Gets the cudo_v1 of this V1ClusterSpec.  # noqa: E501
+
+
+        :return: The cudo_v1 of this V1ClusterSpec.  # noqa: E501
+        :rtype: V1CudoDirectV1
+        """
+        return self._cudo_v1
+
+    @cudo_v1.setter
+    def cudo_v1(self, cudo_v1: 'V1CudoDirectV1'):
+        """Sets the cudo_v1 of this V1ClusterSpec.
+
+
+        :param cudo_v1: The cudo_v1 of this V1ClusterSpec.  # noqa: E501
+        :type: V1CudoDirectV1
+        """
+
+        self._cudo_v1 = cudo_v1
+
     @property
     def deletion_options(self) -> 'V1ClusterDeletionOptions':
         """Gets the deletion_options of this V1ClusterSpec.  # noqa: E501
@@ -814,6 +819,27 @@ class V1ClusterSpec(object):
 
         self._parent_cluster_type = parent_cluster_type
 
+    @property
+    def partitioning_enabled(self) -> 'bool':
+        """Gets the partitioning_enabled of this V1ClusterSpec.  # noqa: E501
+
+
+        :return: The partitioning_enabled of this V1ClusterSpec.  # noqa: E501
+        :rtype: bool
+        """
+        return self._partitioning_enabled
+
+    @partitioning_enabled.setter
+    def partitioning_enabled(self, partitioning_enabled: 'bool'):
+        """Sets the partitioning_enabled of this V1ClusterSpec.
+
+
+        :param partitioning_enabled: The partitioning_enabled of this V1ClusterSpec.  # noqa: E501
+        :type: bool
+        """
+
+        self._partitioning_enabled = partitioning_enabled
+
     @property
     def pause_automation(self) -> 'bool':
         """Gets the pause_automation of this V1ClusterSpec.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_cluster_type.py

@@ -41,7 +41,6 @@ class V1ClusterType(object):
     GLOBAL = "CLUSTER_TYPE_GLOBAL"
     BYOC = "CLUSTER_TYPE_BYOC"
     SLURM = "CLUSTER_TYPE_SLURM"
-    AI_POD = "CLUSTER_TYPE_AI_POD"
     LEC = "CLUSTER_TYPE_LEC"
     """
     Attributes: