lightning-sdk 0.1.37__py3-none-any.whl → 0.1.39__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_status.py

@@ -0,0 +1,279 @@
+# 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 V1MultiMachineJobStatus(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 = {
+        'deleting_machines': 'str',
+        'failing_machines': 'str',
+        'message': 'str',
+        'pending_machines': 'str',
+        'ready_machines': 'str',
+        'started_at': 'datetime',
+        'stopped_at': 'datetime'
+    }
+
+    attribute_map = {
+        'deleting_machines': 'deletingMachines',
+        'failing_machines': 'failingMachines',
+        'message': 'message',
+        'pending_machines': 'pendingMachines',
+        'ready_machines': 'readyMachines',
+        'started_at': 'startedAt',
+        'stopped_at': 'stoppedAt'
+    }
+
+    def __init__(self, deleting_machines: 'str' =None, failing_machines: 'str' =None, message: 'str' =None, pending_machines: 'str' =None, ready_machines: 'str' =None, started_at: 'datetime' =None, stopped_at: 'datetime' =None):  # noqa: E501
+        """V1MultiMachineJobStatus - a model defined in Swagger"""  # noqa: E501
+        self._deleting_machines = None
+        self._failing_machines = None
+        self._message = None
+        self._pending_machines = None
+        self._ready_machines = None
+        self._started_at = None
+        self._stopped_at = None
+        self.discriminator = None
+        if deleting_machines is not None:
+            self.deleting_machines = deleting_machines
+        if failing_machines is not None:
+            self.failing_machines = failing_machines
+        if message is not None:
+            self.message = message
+        if pending_machines is not None:
+            self.pending_machines = pending_machines
+        if ready_machines is not None:
+            self.ready_machines = ready_machines
+        if started_at is not None:
+            self.started_at = started_at
+        if stopped_at is not None:
+            self.stopped_at = stopped_at
+
+    @property
+    def deleting_machines(self) -> 'str':
+        """Gets the deleting_machines of this V1MultiMachineJobStatus.  # noqa: E501
+
+
+        :return: The deleting_machines of this V1MultiMachineJobStatus.  # noqa: E501
+        :rtype: str
+        """
+        return self._deleting_machines
+
+    @deleting_machines.setter
+    def deleting_machines(self, deleting_machines: 'str'):
+        """Sets the deleting_machines of this V1MultiMachineJobStatus.
+
+
+        :param deleting_machines: The deleting_machines of this V1MultiMachineJobStatus.  # noqa: E501
+        :type: str
+        """
+
+        self._deleting_machines = deleting_machines
+
+    @property
+    def failing_machines(self) -> 'str':
+        """Gets the failing_machines of this V1MultiMachineJobStatus.  # noqa: E501
+
+
+        :return: The failing_machines of this V1MultiMachineJobStatus.  # noqa: E501
+        :rtype: str
+        """
+        return self._failing_machines
+
+    @failing_machines.setter
+    def failing_machines(self, failing_machines: 'str'):
+        """Sets the failing_machines of this V1MultiMachineJobStatus.
+
+
+        :param failing_machines: The failing_machines of this V1MultiMachineJobStatus.  # noqa: E501
+        :type: str
+        """
+
+        self._failing_machines = failing_machines
+
+    @property
+    def message(self) -> 'str':
+        """Gets the message of this V1MultiMachineJobStatus.  # noqa: E501
+
+
+        :return: The message of this V1MultiMachineJobStatus.  # noqa: E501
+        :rtype: str
+        """
+        return self._message
+
+    @message.setter
+    def message(self, message: 'str'):
+        """Sets the message of this V1MultiMachineJobStatus.
+
+
+        :param message: The message of this V1MultiMachineJobStatus.  # noqa: E501
+        :type: str
+        """
+
+        self._message = message
+
+    @property
+    def pending_machines(self) -> 'str':
+        """Gets the pending_machines of this V1MultiMachineJobStatus.  # noqa: E501
+
+
+        :return: The pending_machines of this V1MultiMachineJobStatus.  # noqa: E501
+        :rtype: str
+        """
+        return self._pending_machines
+
+    @pending_machines.setter
+    def pending_machines(self, pending_machines: 'str'):
+        """Sets the pending_machines of this V1MultiMachineJobStatus.
+
+
+        :param pending_machines: The pending_machines of this V1MultiMachineJobStatus.  # noqa: E501
+        :type: str
+        """
+
+        self._pending_machines = pending_machines
+
+    @property
+    def ready_machines(self) -> 'str':
+        """Gets the ready_machines of this V1MultiMachineJobStatus.  # noqa: E501
+
+
+        :return: The ready_machines of this V1MultiMachineJobStatus.  # noqa: E501
+        :rtype: str
+        """
+        return self._ready_machines
+
+    @ready_machines.setter
+    def ready_machines(self, ready_machines: 'str'):
+        """Sets the ready_machines of this V1MultiMachineJobStatus.
+
+
+        :param ready_machines: The ready_machines of this V1MultiMachineJobStatus.  # noqa: E501
+        :type: str
+        """
+
+        self._ready_machines = ready_machines
+
+    @property
+    def started_at(self) -> 'datetime':
+        """Gets the started_at of this V1MultiMachineJobStatus.  # noqa: E501
+
+
+        :return: The started_at of this V1MultiMachineJobStatus.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._started_at
+
+    @started_at.setter
+    def started_at(self, started_at: 'datetime'):
+        """Sets the started_at of this V1MultiMachineJobStatus.
+
+
+        :param started_at: The started_at of this V1MultiMachineJobStatus.  # noqa: E501
+        :type: datetime
+        """
+
+        self._started_at = started_at
+
+    @property
+    def stopped_at(self) -> 'datetime':
+        """Gets the stopped_at of this V1MultiMachineJobStatus.  # noqa: E501
+
+
+        :return: The stopped_at of this V1MultiMachineJobStatus.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._stopped_at
+
+    @stopped_at.setter
+    def stopped_at(self, stopped_at: 'datetime'):
+        """Sets the stopped_at of this V1MultiMachineJobStatus.
+
+
+        :param stopped_at: The stopped_at of this V1MultiMachineJobStatus.  # noqa: E501
+        :type: datetime
+        """
+
+        self._stopped_at = stopped_at
+
+    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(V1MultiMachineJobStatus, 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: 'V1MultiMachineJobStatus') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1MultiMachineJobStatus):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1MultiMachineJobStatus') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py

@@ -70,6 +70,8 @@ class V1RuleResource(object):
     MANAGEDENDPOINT = "managedEndpoint"
     JOB = "job"
     DEPLOYMENT = "deployment"
+    MODEL = "model"
+    SLURMJOB = "slurmJob"
     """
     Attributes:
       swagger_types (dict): The key is attribute name

lightning_sdk/lightning_cloud/openapi/models/v1_system_info.py

@@ -54,6 +54,7 @@ class V1SystemInfo(object):
         'gpu_count': 'int',
         'gpu_memory_gb': 'int',
         'gpu_name': 'str',
+        'hostname': 'str',
         'hyperparameters': 'str',
         'installed_packages': 'str',
         'litlogger_version': 'str',
@@ -76,6 +77,7 @@ class V1SystemInfo(object):
         'gpu_count': 'gpuCount',
         'gpu_memory_gb': 'gpuMemoryGb',
         'gpu_name': 'gpuName',
+        'hostname': 'hostname',
         'hyperparameters': 'hyperparameters',
         'installed_packages': 'installedPackages',
         'litlogger_version': 'litloggerVersion',
@@ -84,7 +86,7 @@ class V1SystemInfo(object):
         'system_memory_gb': 'systemMemoryGb'
     }
 
-    def __init__(self, cli_args: 'str' =None, cpu_count_logical: 'int' =None, cpu_count_physical: 'int' =None, cpu_name: 'str' =None, cuda_version: 'str' =None, cudnn_version: 'str' =None, execution_command: 'str' =None, git_branch: 'str' =None, git_commit_hash: 'str' =None, git_repo_name: 'str' =None, gpu_count: 'int' =None, gpu_memory_gb: 'int' =None, gpu_name: 'str' =None, hyperparameters: 'str' =None, installed_packages: 'str' =None, litlogger_version: 'str' =None, os_name: 'str' =None, python_version: 'str' =None, system_memory_gb: 'int' =None):  # noqa: E501
+    def __init__(self, cli_args: 'str' =None, cpu_count_logical: 'int' =None, cpu_count_physical: 'int' =None, cpu_name: 'str' =None, cuda_version: 'str' =None, cudnn_version: 'str' =None, execution_command: 'str' =None, git_branch: 'str' =None, git_commit_hash: 'str' =None, git_repo_name: 'str' =None, gpu_count: 'int' =None, gpu_memory_gb: 'int' =None, gpu_name: 'str' =None, hostname: 'str' =None, hyperparameters: 'str' =None, installed_packages: 'str' =None, litlogger_version: 'str' =None, os_name: 'str' =None, python_version: 'str' =None, system_memory_gb: 'int' =None):  # noqa: E501
         """V1SystemInfo - a model defined in Swagger"""  # noqa: E501
         self._cli_args = None
         self._cpu_count_logical = None
@@ -99,6 +101,7 @@ class V1SystemInfo(object):
         self._gpu_count = None
         self._gpu_memory_gb = None
         self._gpu_name = None
+        self._hostname = None
         self._hyperparameters = None
         self._installed_packages = None
         self._litlogger_version = None
@@ -132,6 +135,8 @@ class V1SystemInfo(object):
             self.gpu_memory_gb = gpu_memory_gb
         if gpu_name is not None:
             self.gpu_name = gpu_name
+        if hostname is not None:
+            self.hostname = hostname
         if hyperparameters is not None:
             self.hyperparameters = hyperparameters
         if installed_packages is not None:
@@ -418,6 +423,27 @@ class V1SystemInfo(object):
 
         self._gpu_name = gpu_name
 
+    @property
+    def hostname(self) -> 'str':
+        """Gets the hostname of this V1SystemInfo.  # noqa: E501
+
+
+        :return: The hostname of this V1SystemInfo.  # noqa: E501
+        :rtype: str
+        """
+        return self._hostname
+
+    @hostname.setter
+    def hostname(self, hostname: 'str'):
+        """Sets the hostname of this V1SystemInfo.
+
+
+        :param hostname: The hostname of this V1SystemInfo.  # noqa: E501
+        :type: str
+        """
+
+        self._hostname = hostname
+
     @property
     def hyperparameters(self) -> 'str':
         """Gets the hyperparameters of this V1SystemInfo.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -44,6 +44,7 @@ class V1UserFeatures(object):
         'advanced_deployment_autoscaling': 'bool',
         'affiliate_links': 'bool',
         'agents_v2': 'bool',
+        'ai_hub_monetization': 'bool',
         'auto_fast_load': 'bool',
         'auto_join_orgs': 'bool',
         'aws_trainium': 'bool',
@@ -112,6 +113,7 @@ class V1UserFeatures(object):
         'show_dev_admin': 'bool',
         'slurm': 'bool',
         'slurm_machine_selector': 'bool',
+        'snapshotter_service': 'bool',
         'snowflake_connection': 'bool',
         'spot_v2': 'bool',
         'studio_config': 'bool',
@@ -126,6 +128,7 @@ class V1UserFeatures(object):
         'advanced_deployment_autoscaling': 'advancedDeploymentAutoscaling',
         'affiliate_links': 'affiliateLinks',
         'agents_v2': 'agentsV2',
+        'ai_hub_monetization': 'aiHubMonetization',
         'auto_fast_load': 'autoFastLoad',
         'auto_join_orgs': 'autoJoinOrgs',
         'aws_trainium': 'awsTrainium',
@@ -194,6 +197,7 @@ class V1UserFeatures(object):
         'show_dev_admin': 'showDevAdmin',
         'slurm': 'slurm',
         'slurm_machine_selector': 'slurmMachineSelector',
+        'snapshotter_service': 'snapshotterService',
         'snowflake_connection': 'snowflakeConnection',
         'spot_v2': 'spotV2',
         'studio_config': 'studioConfig',
@@ -204,11 +208,12 @@ class V1UserFeatures(object):
         'writable_data_connections': 'writableDataConnections'
     }
 
-    def __init__(self, advanced_deployment_autoscaling: 'bool' =None, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, aws_trainium: 'bool' =None, b2c_experience: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, cluster_proxy: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, cost_attribution_settings: 'bool' =None, crypto_monitoring: 'bool' =None, custom_app_domain: 'bool' =None, custom_instance_types: 'bool' =None, default_one_cluster: 'bool' =None, deployment_customize_api: 'bool' =None, deployment_data_path: 'bool' =None, deployment_gallery: 'bool' =None, deployment_persistent_disk: 'bool' =None, deployment_version_visibility: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, dynamic_workload_scheduling: 'bool' =None, enable_crypto_crackdown: 'bool' =None, enable_efs: 'bool' =None, enable_storage_limits: 'bool' =None, featured_studios_admin: 'bool' =None, filesystem_optimisation: 'bool' =None, gcp: 'bool' =None, inference_job_deployment_plugin: 'bool' =None, jobs_init: 'bool' =None, jobs_v2: 'bool' =None, landing_studios: 'bool' =None, lightning_registry: 'bool' =None, lit_logger: 'bool' =None, lit_logger_storage_v2: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_store: 'bool' =None, multiple_deployment_versions: 'bool' =None, multiple_studio_versions: 'bool' =None, org_level_member_permissions: 'bool' =None, plugin_biz_chat: 'bool' =None, plugin_distributed: 'bool' =None, plugin_fiftyone: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_lightning_apps: 'bool' =None, plugin_lightning_apps_distributed: 'bool' =None, plugin_mage_ai: 'bool' =None, plugin_milvus: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_react: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, plugin_weviate: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, project_selector: 'bool' =None, restart_ide_on_hang: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, s3_folders: 'bool' =None, show_dev_admin: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, snowflake_connection: 'bool' =None, spot_v2: 'bool' =None, studio_config: 'bool' =None, studio_on_stop: 'bool' =None, studio_version_visibility: 'bool' =None, teamspace_storage_tab: 'bool' =None, use_rclone_mounts_only: 'bool' =None, writable_data_connections: 'bool' =None):  # noqa: E501
+    def __init__(self, advanced_deployment_autoscaling: '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, aws_trainium: 'bool' =None, b2c_experience: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, cluster_proxy: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, cost_attribution_settings: 'bool' =None, crypto_monitoring: 'bool' =None, custom_app_domain: 'bool' =None, custom_instance_types: 'bool' =None, default_one_cluster: 'bool' =None, deployment_customize_api: 'bool' =None, deployment_data_path: 'bool' =None, deployment_gallery: 'bool' =None, deployment_persistent_disk: 'bool' =None, deployment_version_visibility: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, dynamic_workload_scheduling: 'bool' =None, enable_crypto_crackdown: 'bool' =None, enable_efs: 'bool' =None, enable_storage_limits: 'bool' =None, featured_studios_admin: 'bool' =None, filesystem_optimisation: 'bool' =None, gcp: 'bool' =None, inference_job_deployment_plugin: 'bool' =None, jobs_init: 'bool' =None, jobs_v2: 'bool' =None, landing_studios: 'bool' =None, lightning_registry: 'bool' =None, lit_logger: 'bool' =None, lit_logger_storage_v2: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_store: 'bool' =None, multiple_deployment_versions: 'bool' =None, multiple_studio_versions: 'bool' =None, org_level_member_permissions: 'bool' =None, plugin_biz_chat: 'bool' =None, plugin_distributed: 'bool' =None, plugin_fiftyone: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_lightning_apps: 'bool' =None, plugin_lightning_apps_distributed: 'bool' =None, plugin_mage_ai: 'bool' =None, plugin_milvus: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_react: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, plugin_weviate: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, project_selector: 'bool' =None, restart_ide_on_hang: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, s3_folders: 'bool' =None, show_dev_admin: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, snapshotter_service: 'bool' =None, snowflake_connection: 'bool' =None, spot_v2: 'bool' =None, studio_config: 'bool' =None, studio_on_stop: 'bool' =None, studio_version_visibility: 'bool' =None, teamspace_storage_tab: 'bool' =None, use_rclone_mounts_only: 'bool' =None, writable_data_connections: 'bool' =None):  # noqa: E501
         """V1UserFeatures - a model defined in Swagger"""  # noqa: E501
         self._advanced_deployment_autoscaling = None
         self._affiliate_links = None
         self._agents_v2 = None
+        self._ai_hub_monetization = None
         self._auto_fast_load = None
         self._auto_join_orgs = None
         self._aws_trainium = None
@@ -277,6 +282,7 @@ class V1UserFeatures(object):
         self._show_dev_admin = None
         self._slurm = None
         self._slurm_machine_selector = None
+        self._snapshotter_service = None
         self._snowflake_connection = None
         self._spot_v2 = None
         self._studio_config = None
@@ -292,6 +298,8 @@ class V1UserFeatures(object):
             self.affiliate_links = affiliate_links
         if agents_v2 is not None:
             self.agents_v2 = agents_v2
+        if ai_hub_monetization is not None:
+            self.ai_hub_monetization = ai_hub_monetization
         if auto_fast_load is not None:
             self.auto_fast_load = auto_fast_load
         if auto_join_orgs is not None:
@@ -428,6 +436,8 @@ class V1UserFeatures(object):
             self.slurm = slurm
         if slurm_machine_selector is not None:
             self.slurm_machine_selector = slurm_machine_selector
+        if snapshotter_service is not None:
+            self.snapshotter_service = snapshotter_service
         if snowflake_connection is not None:
             self.snowflake_connection = snowflake_connection
         if spot_v2 is not None:
@@ -508,6 +518,27 @@ class V1UserFeatures(object):
 
         self._agents_v2 = agents_v2
 
+    @property
+    def ai_hub_monetization(self) -> 'bool':
+        """Gets the ai_hub_monetization of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The ai_hub_monetization of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._ai_hub_monetization
+
+    @ai_hub_monetization.setter
+    def ai_hub_monetization(self, ai_hub_monetization: 'bool'):
+        """Sets the ai_hub_monetization of this V1UserFeatures.
+
+
+        :param ai_hub_monetization: The ai_hub_monetization of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._ai_hub_monetization = ai_hub_monetization
+
     @property
     def auto_fast_load(self) -> 'bool':
         """Gets the auto_fast_load of this V1UserFeatures.  # noqa: E501
@@ -1936,6 +1967,27 @@ class V1UserFeatures(object):
 
         self._slurm_machine_selector = slurm_machine_selector
 
+    @property
+    def snapshotter_service(self) -> 'bool':
+        """Gets the snapshotter_service of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The snapshotter_service of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._snapshotter_service
+
+    @snapshotter_service.setter
+    def snapshotter_service(self, snapshotter_service: 'bool'):
+        """Sets the snapshotter_service of this V1UserFeatures.
+
+
+        :param snapshotter_service: The snapshotter_service of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._snapshotter_service = snapshotter_service
+
     @property
     def snowflake_connection(self) -> 'bool':
         """Gets the snowflake_connection of this V1UserFeatures.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_compute_config.py

@@ -48,6 +48,7 @@ class V1UserRequestedComputeConfig(object):
         'gpu_image_override': 'str',
         'id': 'str',
         'name': 'str',
+        'requested_run_duration_seconds': 'str',
         'same_compute_on_resume': 'bool',
         'shm_size': 'int',
         'spot': 'bool'
@@ -61,12 +62,13 @@ class V1UserRequestedComputeConfig(object):
         'gpu_image_override': 'gpuImageOverride',
         'id': 'id',
         'name': 'name',
+        'requested_run_duration_seconds': 'requestedRunDurationSeconds',
         'same_compute_on_resume': 'sameComputeOnResume',
         'shm_size': 'shmSize',
         'spot': 'spot'
     }
 
-    def __init__(self, affinity_identifier: 'str' =None, count: 'int' =None, cpu_image_override: 'str' =None, disk_size: 'int' =None, gpu_image_override: 'str' =None, id: 'str' =None, name: 'str' =None, same_compute_on_resume: 'bool' =None, shm_size: 'int' =None, spot: 'bool' =None):  # noqa: E501
+    def __init__(self, affinity_identifier: 'str' =None, count: 'int' =None, cpu_image_override: 'str' =None, disk_size: 'int' =None, gpu_image_override: 'str' =None, id: 'str' =None, name: 'str' =None, requested_run_duration_seconds: 'str' =None, same_compute_on_resume: 'bool' =None, shm_size: 'int' =None, spot: 'bool' =None):  # noqa: E501
         """V1UserRequestedComputeConfig - a model defined in Swagger"""  # noqa: E501
         self._affinity_identifier = None
         self._count = None
@@ -75,6 +77,7 @@ class V1UserRequestedComputeConfig(object):
         self._gpu_image_override = None
         self._id = None
         self._name = None
+        self._requested_run_duration_seconds = None
         self._same_compute_on_resume = None
         self._shm_size = None
         self._spot = None
@@ -93,6 +96,8 @@ class V1UserRequestedComputeConfig(object):
             self.id = id
         if name is not None:
             self.name = name
+        if requested_run_duration_seconds is not None:
+            self.requested_run_duration_seconds = requested_run_duration_seconds
         if same_compute_on_resume is not None:
             self.same_compute_on_resume = same_compute_on_resume
         if shm_size is not None:
@@ -253,6 +258,27 @@ class V1UserRequestedComputeConfig(object):
 
         self._name = name
 
+    @property
+    def requested_run_duration_seconds(self) -> 'str':
+        """Gets the requested_run_duration_seconds of this V1UserRequestedComputeConfig.  # noqa: E501
+
+
+        :return: The requested_run_duration_seconds of this V1UserRequestedComputeConfig.  # noqa: E501
+        :rtype: str
+        """
+        return self._requested_run_duration_seconds
+
+    @requested_run_duration_seconds.setter
+    def requested_run_duration_seconds(self, requested_run_duration_seconds: 'str'):
+        """Sets the requested_run_duration_seconds of this V1UserRequestedComputeConfig.
+
+
+        :param requested_run_duration_seconds: The requested_run_duration_seconds of this V1UserRequestedComputeConfig.  # noqa: E501
+        :type: str
+        """
+
+        self._requested_run_duration_seconds = requested_run_duration_seconds
+
     @property
     def same_compute_on_resume(self) -> 'bool':
         """Gets the same_compute_on_resume of this V1UserRequestedComputeConfig.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_validate_data_connection_response.py

@@ -42,8 +42,7 @@ class V1ValidateDataConnectionResponse(object):
     """
     swagger_types = {
         'aws': 'V1AwsDataConnection',
-        'efs': 'V1EfsDataConnection',
-        'efs_folder': 'V1EFSFolderDataConnection',
+        'efs': 'V1EfsConfig',
         'gcp': 'V1GcpDataConnection',
         'gcs_folder': 'V1GCSFolderDataConnection',
         's3_folder': 'V1S3FolderDataConnection'
@@ -52,17 +51,15 @@ class V1ValidateDataConnectionResponse(object):
     attribute_map = {
         'aws': 'aws',
         'efs': 'efs',
-        'efs_folder': 'efsFolder',
         'gcp': 'gcp',
         'gcs_folder': 'gcsFolder',
         's3_folder': 's3Folder'
     }
 
-    def __init__(self, aws: 'V1AwsDataConnection' =None, efs: 'V1EfsDataConnection' =None, efs_folder: 'V1EFSFolderDataConnection' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, s3_folder: 'V1S3FolderDataConnection' =None):  # noqa: E501
+    def __init__(self, aws: 'V1AwsDataConnection' =None, efs: 'V1EfsConfig' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, s3_folder: 'V1S3FolderDataConnection' =None):  # noqa: E501
         """V1ValidateDataConnectionResponse - a model defined in Swagger"""  # noqa: E501
         self._aws = None
         self._efs = None
-        self._efs_folder = None
         self._gcp = None
         self._gcs_folder = None
         self._s3_folder = None
@@ -71,8 +68,6 @@ class V1ValidateDataConnectionResponse(object):
             self.aws = aws
         if efs is not None:
             self.efs = efs
-        if efs_folder is not None:
-            self.efs_folder = efs_folder
         if gcp is not None:
             self.gcp = gcp
         if gcs_folder is not None:
@@ -102,47 +97,26 @@ class V1ValidateDataConnectionResponse(object):
         self._aws = aws
 
     @property
-    def efs(self) -> 'V1EfsDataConnection':
+    def efs(self) -> 'V1EfsConfig':
         """Gets the efs of this V1ValidateDataConnectionResponse.  # noqa: E501
 
 
         :return: The efs of this V1ValidateDataConnectionResponse.  # noqa: E501
-        :rtype: V1EfsDataConnection
+        :rtype: V1EfsConfig
         """
         return self._efs
 
     @efs.setter
-    def efs(self, efs: 'V1EfsDataConnection'):
+    def efs(self, efs: 'V1EfsConfig'):
         """Sets the efs of this V1ValidateDataConnectionResponse.
 
 
         :param efs: The efs of this V1ValidateDataConnectionResponse.  # noqa: E501
-        :type: V1EfsDataConnection
+        :type: V1EfsConfig
         """
 
         self._efs = efs
 
-    @property
-    def efs_folder(self) -> 'V1EFSFolderDataConnection':
-        """Gets the efs_folder of this V1ValidateDataConnectionResponse.  # noqa: E501
-
-
-        :return: The efs_folder of this V1ValidateDataConnectionResponse.  # noqa: E501
-        :rtype: V1EFSFolderDataConnection
-        """
-        return self._efs_folder
-
-    @efs_folder.setter
-    def efs_folder(self, efs_folder: 'V1EFSFolderDataConnection'):
-        """Sets the efs_folder of this V1ValidateDataConnectionResponse.
-
-
-        :param efs_folder: The efs_folder of this V1ValidateDataConnectionResponse.  # noqa: E501
-        :type: V1EFSFolderDataConnection
-        """
-
-        self._efs_folder = efs_folder
-
     @property
     def gcp(self) -> 'V1GcpDataConnection':
         """Gets the gcp of this V1ValidateDataConnectionResponse.  # noqa: E501