lightning-sdk 0.1.36__py3-none-any.whl → 0.1.38__py3-none-any.whl

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',
@@ -64,8 +65,10 @@ class V1UserFeatures(object):
         'deployment_data_path': 'bool',
         'deployment_gallery': 'bool',
         'deployment_persistent_disk': 'bool',
+        'deployment_version_visibility': 'bool',
         'docs_agent': 'bool',
         'drive_v2': 'bool',
+        'dynamic_workload_scheduling': 'bool',
         'enable_crypto_crackdown': 'bool',
         'enable_efs': 'bool',
         'enable_storage_limits': 'bool',
@@ -110,10 +113,12 @@ 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',
         'studio_on_stop': 'bool',
+        'studio_version_visibility': 'bool',
         'teamspace_storage_tab': 'bool',
         'use_rclone_mounts_only': 'bool',
         'writable_data_connections': 'bool'
@@ -123,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',
@@ -143,8 +149,10 @@ class V1UserFeatures(object):
         'deployment_data_path': 'deploymentDataPath',
         'deployment_gallery': 'deploymentGallery',
         'deployment_persistent_disk': 'deploymentPersistentDisk',
+        'deployment_version_visibility': 'deploymentVersionVisibility',
         'docs_agent': 'docsAgent',
         'drive_v2': 'driveV2',
+        'dynamic_workload_scheduling': 'dynamicWorkloadScheduling',
         'enable_crypto_crackdown': 'enableCryptoCrackdown',
         'enable_efs': 'enableEfs',
         'enable_storage_limits': 'enableStorageLimits',
@@ -189,20 +197,23 @@ 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',
         'studio_on_stop': 'studioOnStop',
+        'studio_version_visibility': 'studioVersionVisibility',
         'teamspace_storage_tab': 'teamspaceStorageTab',
         'use_rclone_mounts_only': 'useRcloneMountsOnly',
         '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, docs_agent: 'bool' =None, drive_v2: '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, 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
@@ -223,8 +234,10 @@ class V1UserFeatures(object):
         self._deployment_data_path = None
         self._deployment_gallery = None
         self._deployment_persistent_disk = None
+        self._deployment_version_visibility = None
         self._docs_agent = None
         self._drive_v2 = None
+        self._dynamic_workload_scheduling = None
         self._enable_crypto_crackdown = None
         self._enable_efs = None
         self._enable_storage_limits = None
@@ -269,10 +282,12 @@ 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
         self._studio_on_stop = None
+        self._studio_version_visibility = None
         self._teamspace_storage_tab = None
         self._use_rclone_mounts_only = None
         self._writable_data_connections = None
@@ -283,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:
@@ -323,10 +340,14 @@ class V1UserFeatures(object):
             self.deployment_gallery = deployment_gallery
         if deployment_persistent_disk is not None:
             self.deployment_persistent_disk = deployment_persistent_disk
+        if deployment_version_visibility is not None:
+            self.deployment_version_visibility = deployment_version_visibility
         if docs_agent is not None:
             self.docs_agent = docs_agent
         if drive_v2 is not None:
             self.drive_v2 = drive_v2
+        if dynamic_workload_scheduling is not None:
+            self.dynamic_workload_scheduling = dynamic_workload_scheduling
         if enable_crypto_crackdown is not None:
             self.enable_crypto_crackdown = enable_crypto_crackdown
         if enable_efs is not None:
@@ -415,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:
@@ -423,6 +446,8 @@ class V1UserFeatures(object):
             self.studio_config = studio_config
         if studio_on_stop is not None:
             self.studio_on_stop = studio_on_stop
+        if studio_version_visibility is not None:
+            self.studio_version_visibility = studio_version_visibility
         if teamspace_storage_tab is not None:
             self.teamspace_storage_tab = teamspace_storage_tab
         if use_rclone_mounts_only is not None:
@@ -493,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
@@ -913,6 +959,27 @@ class V1UserFeatures(object):
 
         self._deployment_persistent_disk = deployment_persistent_disk
 
+    @property
+    def deployment_version_visibility(self) -> 'bool':
+        """Gets the deployment_version_visibility of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The deployment_version_visibility of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._deployment_version_visibility
+
+    @deployment_version_visibility.setter
+    def deployment_version_visibility(self, deployment_version_visibility: 'bool'):
+        """Sets the deployment_version_visibility of this V1UserFeatures.
+
+
+        :param deployment_version_visibility: The deployment_version_visibility of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._deployment_version_visibility = deployment_version_visibility
+
     @property
     def docs_agent(self) -> 'bool':
         """Gets the docs_agent of this V1UserFeatures.  # noqa: E501
@@ -955,6 +1022,27 @@ class V1UserFeatures(object):
 
         self._drive_v2 = drive_v2
 
+    @property
+    def dynamic_workload_scheduling(self) -> 'bool':
+        """Gets the dynamic_workload_scheduling of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The dynamic_workload_scheduling of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._dynamic_workload_scheduling
+
+    @dynamic_workload_scheduling.setter
+    def dynamic_workload_scheduling(self, dynamic_workload_scheduling: 'bool'):
+        """Sets the dynamic_workload_scheduling of this V1UserFeatures.
+
+
+        :param dynamic_workload_scheduling: The dynamic_workload_scheduling of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._dynamic_workload_scheduling = dynamic_workload_scheduling
+
     @property
     def enable_crypto_crackdown(self) -> 'bool':
         """Gets the enable_crypto_crackdown of this V1UserFeatures.  # noqa: E501
@@ -1879,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
@@ -1963,6 +2072,27 @@ class V1UserFeatures(object):
 
         self._studio_on_stop = studio_on_stop
 
+    @property
+    def studio_version_visibility(self) -> 'bool':
+        """Gets the studio_version_visibility of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The studio_version_visibility of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._studio_version_visibility
+
+    @studio_version_visibility.setter
+    def studio_version_visibility(self, studio_version_visibility: 'bool'):
+        """Sets the studio_version_visibility of this V1UserFeatures.
+
+
+        :param studio_version_visibility: The studio_version_visibility of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._studio_version_visibility = studio_version_visibility
+
     @property
     def teamspace_storage_tab(self) -> 'bool':
         """Gets the teamspace_storage_tab 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

lightning_sdk/lightning_cloud/openapi/models/validate.py

@@ -44,8 +44,7 @@ class Validate(object):
         'aws': 'V1AwsDataConnection',
         'check_is_public': 'bool',
         'cluster_ids': 'list[str]',
-        'efs': 'V1EfsDataConnection',
-        'efs_folder': 'V1EFSFolderDataConnection',
+        'efs': 'V1EfsConfig',
         'gcp': 'V1GcpDataConnection',
         'gcs_folder': 'V1GCSFolderDataConnection',
         's3_folder': 'V1S3FolderDataConnection'
@@ -56,19 +55,17 @@ class Validate(object):
         'check_is_public': 'checkIsPublic',
         'cluster_ids': 'clusterIds',
         'efs': 'efs',
-        'efs_folder': 'efsFolder',
         'gcp': 'gcp',
         'gcs_folder': 'gcsFolder',
         's3_folder': 's3Folder'
     }
 
-    def __init__(self, aws: 'V1AwsDataConnection' =None, check_is_public: 'bool' =None, cluster_ids: 'list[str]' =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, check_is_public: 'bool' =None, cluster_ids: 'list[str]' =None, efs: 'V1EfsConfig' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, s3_folder: 'V1S3FolderDataConnection' =None):  # noqa: E501
         """Validate - a model defined in Swagger"""  # noqa: E501
         self._aws = None
         self._check_is_public = None
         self._cluster_ids = None
         self._efs = None
-        self._efs_folder = None
         self._gcp = None
         self._gcs_folder = None
         self._s3_folder = None
@@ -81,8 +78,6 @@ class Validate(object):
             self.cluster_ids = cluster_ids
         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:
@@ -154,47 +149,26 @@ class Validate(object):
         self._cluster_ids = cluster_ids
 
     @property
-    def efs(self) -> 'V1EfsDataConnection':
+    def efs(self) -> 'V1EfsConfig':
         """Gets the efs of this Validate.  # noqa: E501
 
 
         :return: The efs of this Validate.  # 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 Validate.
 
 
         :param efs: The efs of this Validate.  # noqa: E501
-        :type: V1EfsDataConnection
+        :type: V1EfsConfig
         """
 
         self._efs = efs
 
-    @property
-    def efs_folder(self) -> 'V1EFSFolderDataConnection':
-        """Gets the efs_folder of this Validate.  # noqa: E501
-
-
-        :return: The efs_folder of this Validate.  # noqa: E501
-        :rtype: V1EFSFolderDataConnection
-        """
-        return self._efs_folder
-
-    @efs_folder.setter
-    def efs_folder(self, efs_folder: 'V1EFSFolderDataConnection'):
-        """Sets the efs_folder of this Validate.
-
-
-        :param efs_folder: The efs_folder of this Validate.  # noqa: E501
-        :type: V1EFSFolderDataConnection
-        """
-
-        self._efs_folder = efs_folder
-
     @property
     def gcp(self) -> 'V1GcpDataConnection':
         """Gets the gcp of this Validate.  # noqa: E501

lightning_sdk/teamspace.py

@@ -245,6 +245,11 @@ class Teamspace:
         download_dir = Path(download_dir)
 
         name, version = _parse_model_and_version(name)
+        model_version = self._teamspace_api.get_model_version(name=name, version=version, teamspace_id=self.id)
+        if not model_version.upload_complete:
+            raise RuntimeError(
+                f"Model {name}:{version} is not fully uploaded yet. Please wait until the upload is complete."
+            )
         downloaded_files = self._teamspace_api.download_model_files(
             name=name,
             version=version,

{lightning_sdk-0.1.36.dist-info → lightning_sdk-0.1.38.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning_sdk
-Version: 0.1.36
+Version: 0.1.38
 Summary: SDK to develop using Lightning AI Studios
 Author-email: Lightning-AI <justus@lightning.ai>
 License: MIT License