lightning-sdk 2025.10.22__py3-none-any.whl → 2025.10.27__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_reset_api_key_response.py

@@ -0,0 +1,123 @@
+# 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 V1ResetAPIKeyResponse(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 = {
+        'api_key': 'str'
+    }
+
+    attribute_map = {
+        'api_key': 'apiKey'
+    }
+
+    def __init__(self, api_key: 'str' =None):  # noqa: E501
+        """V1ResetAPIKeyResponse - a model defined in Swagger"""  # noqa: E501
+        self._api_key = None
+        self.discriminator = None
+        if api_key is not None:
+            self.api_key = api_key
+
+    @property
+    def api_key(self) -> 'str':
+        """Gets the api_key of this V1ResetAPIKeyResponse.  # noqa: E501
+
+
+        :return: The api_key of this V1ResetAPIKeyResponse.  # noqa: E501
+        :rtype: str
+        """
+        return self._api_key
+
+    @api_key.setter
+    def api_key(self, api_key: 'str'):
+        """Sets the api_key of this V1ResetAPIKeyResponse.
+
+
+        :param api_key: The api_key of this V1ResetAPIKeyResponse.  # noqa: E501
+        :type: str
+        """
+
+        self._api_key = api_key
+
+    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(V1ResetAPIKeyResponse, 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: 'V1ResetAPIKeyResponse') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1ResetAPIKeyResponse):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1ResetAPIKeyResponse') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -63,7 +63,6 @@ class V1UserFeatures(object):
         'deployment_persistent_disk': 'bool',
         'drive_v2': 'bool',
         'enterprise_compute_admin': 'bool',
-        'f227': 'bool',
         'f234': 'bool',
         'f236': 'bool',
         'f239': 'bool',
@@ -72,25 +71,25 @@ class V1UserFeatures(object):
         'f243': 'bool',
         'f245': 'bool',
         'f247': 'bool',
-        'f248': 'bool',
         'f250': 'bool',
         'f252': 'bool',
         'f253': 'bool',
         'f254': 'bool',
         'f255': 'bool',
-        'f257': 'bool',
         'f258': 'bool',
         'f259': 'bool',
         'f260': 'bool',
         'f261': 'bool',
         'f262': 'bool',
         'f263': 'bool',
-        'f264': 'bool',
         'f265': 'bool',
         'f266': 'bool',
+        'f267': 'bool',
+        'f268': 'bool',
+        'f269': 'bool',
+        'f270': 'bool',
         'fair_share': 'bool',
         'featured_studios_admin': 'bool',
-        'gcs_connections_optimized': 'bool',
         'job_artifacts_v2': 'bool',
         'kubernetes_cluster_ui': 'bool',
         'kubernetes_clusters': 'bool',
@@ -155,7 +154,6 @@ class V1UserFeatures(object):
         'deployment_persistent_disk': 'deploymentPersistentDisk',
         'drive_v2': 'driveV2',
         'enterprise_compute_admin': 'enterpriseComputeAdmin',
-        'f227': 'f227',
         'f234': 'f234',
         'f236': 'f236',
         'f239': 'f239',
@@ -164,25 +162,25 @@ class V1UserFeatures(object):
         'f243': 'f243',
         'f245': 'f245',
         'f247': 'f247',
-        'f248': 'f248',
         'f250': 'f250',
         'f252': 'f252',
         'f253': 'f253',
         'f254': 'f254',
         'f255': 'f255',
-        'f257': 'f257',
         'f258': 'f258',
         'f259': 'f259',
         'f260': 'f260',
         'f261': 'f261',
         'f262': 'f262',
         'f263': 'f263',
-        'f264': 'f264',
         'f265': 'f265',
         'f266': 'f266',
+        'f267': 'f267',
+        'f268': 'f268',
+        'f269': 'f269',
+        'f270': 'f270',
         'fair_share': 'fairShare',
         'featured_studios_admin': 'featuredStudiosAdmin',
-        'gcs_connections_optimized': 'gcsConnectionsOptimized',
         'job_artifacts_v2': 'jobArtifactsV2',
         'kubernetes_cluster_ui': 'kubernetesClusterUi',
         'kubernetes_clusters': 'kubernetesClusters',
@@ -224,7 +222,7 @@ class V1UserFeatures(object):
         'writable_s3_connections': 'writableS3Connections'
     }
 
-    def __init__(self, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: '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, f239: 'bool' =None, f240: 'bool' =None, f241: 'bool' =None, f243: 'bool' =None, f245: 'bool' =None, f247: 'bool' =None, f248: 'bool' =None, f250: 'bool' =None, f252: 'bool' =None, f253: 'bool' =None, f254: 'bool' =None, f255: 'bool' =None, f257: 'bool' =None, f258: 'bool' =None, f259: 'bool' =None, f260: 'bool' =None, f261: 'bool' =None, f262: 'bool' =None, f263: 'bool' =None, f264: 'bool' =None, f265: 'bool' =None, f266: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcs_connections_optimized: '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, 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, 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, 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, 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, f234: 'bool' =None, f236: 'bool' =None, f239: 'bool' =None, f240: 'bool' =None, f241: 'bool' =None, f243: 'bool' =None, f245: 'bool' =None, f247: 'bool' =None, f250: 'bool' =None, f252: 'bool' =None, f253: 'bool' =None, f254: 'bool' =None, f255: 'bool' =None, f258: 'bool' =None, f259: 'bool' =None, f260: 'bool' =None, f261: 'bool' =None, f262: 'bool' =None, f263: 'bool' =None, f265: 'bool' =None, f266: 'bool' =None, f267: 'bool' =None, f268: 'bool' =None, f269: 'bool' =None, f270: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: '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, 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, 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, vultr: 'bool' =None, weka: 'bool' =None, writable_s3_connections: 'bool' =None):  # noqa: E501
         """V1UserFeatures - a model defined in Swagger"""  # noqa: E501
         self._affiliate_links = None
         self._agents_v2 = None
@@ -248,7 +246,6 @@ class V1UserFeatures(object):
         self._deployment_persistent_disk = None
         self._drive_v2 = None
         self._enterprise_compute_admin = None
-        self._f227 = None
         self._f234 = None
         self._f236 = None
         self._f239 = None
@@ -257,25 +254,25 @@ class V1UserFeatures(object):
         self._f243 = None
         self._f245 = None
         self._f247 = None
-        self._f248 = None
         self._f250 = None
         self._f252 = None
         self._f253 = None
         self._f254 = None
         self._f255 = None
-        self._f257 = None
         self._f258 = None
         self._f259 = None
         self._f260 = None
         self._f261 = None
         self._f262 = None
         self._f263 = None
-        self._f264 = None
         self._f265 = None
         self._f266 = None
+        self._f267 = None
+        self._f268 = None
+        self._f269 = None
+        self._f270 = None
         self._fair_share = None
         self._featured_studios_admin = None
-        self._gcs_connections_optimized = None
         self._job_artifacts_v2 = None
         self._kubernetes_cluster_ui = None
         self._kubernetes_clusters = None
@@ -360,8 +357,6 @@ class V1UserFeatures(object):
             self.drive_v2 = drive_v2
         if enterprise_compute_admin is not None:
             self.enterprise_compute_admin = enterprise_compute_admin
-        if f227 is not None:
-            self.f227 = f227
         if f234 is not None:
             self.f234 = f234
         if f236 is not None:
@@ -378,8 +373,6 @@ class V1UserFeatures(object):
             self.f245 = f245
         if f247 is not None:
             self.f247 = f247
-        if f248 is not None:
-            self.f248 = f248
         if f250 is not None:
             self.f250 = f250
         if f252 is not None:
@@ -390,8 +383,6 @@ class V1UserFeatures(object):
             self.f254 = f254
         if f255 is not None:
             self.f255 = f255
-        if f257 is not None:
-            self.f257 = f257
         if f258 is not None:
             self.f258 = f258
         if f259 is not None:
@@ -404,18 +395,22 @@ class V1UserFeatures(object):
             self.f262 = f262
         if f263 is not None:
             self.f263 = f263
-        if f264 is not None:
-            self.f264 = f264
         if f265 is not None:
             self.f265 = f265
         if f266 is not None:
             self.f266 = f266
+        if f267 is not None:
+            self.f267 = f267
+        if f268 is not None:
+            self.f268 = f268
+        if f269 is not None:
+            self.f269 = f269
+        if f270 is not None:
+            self.f270 = f270
         if fair_share is not None:
             self.fair_share = fair_share
         if featured_studios_admin is not None:
             self.featured_studios_admin = featured_studios_admin
-        if gcs_connections_optimized is not None:
-            self.gcs_connections_optimized = gcs_connections_optimized
         if job_artifacts_v2 is not None:
             self.job_artifacts_v2 = job_artifacts_v2
         if kubernetes_cluster_ui is not None:
@@ -957,27 +952,6 @@ class V1UserFeatures(object):
 
         self._enterprise_compute_admin = enterprise_compute_admin
 
-    @property
-    def f227(self) -> 'bool':
-        """Gets the f227 of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The f227 of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._f227
-
-    @f227.setter
-    def f227(self, f227: 'bool'):
-        """Sets the f227 of this V1UserFeatures.
-
-
-        :param f227: The f227 of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._f227 = f227
-
     @property
     def f234(self) -> 'bool':
         """Gets the f234 of this V1UserFeatures.  # noqa: E501
@@ -1146,27 +1120,6 @@ class V1UserFeatures(object):
 
         self._f247 = f247
 
-    @property
-    def f248(self) -> 'bool':
-        """Gets the f248 of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The f248 of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._f248
-
-    @f248.setter
-    def f248(self, f248: 'bool'):
-        """Sets the f248 of this V1UserFeatures.
-
-
-        :param f248: The f248 of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._f248 = f248
-
     @property
     def f250(self) -> 'bool':
         """Gets the f250 of this V1UserFeatures.  # noqa: E501
@@ -1272,27 +1225,6 @@ class V1UserFeatures(object):
 
         self._f255 = f255
 
-    @property
-    def f257(self) -> 'bool':
-        """Gets the f257 of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The f257 of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._f257
-
-    @f257.setter
-    def f257(self, f257: 'bool'):
-        """Sets the f257 of this V1UserFeatures.
-
-
-        :param f257: The f257 of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._f257 = f257
-
     @property
     def f258(self) -> 'bool':
         """Gets the f258 of this V1UserFeatures.  # noqa: E501
@@ -1419,27 +1351,6 @@ class V1UserFeatures(object):
 
         self._f263 = f263
 
-    @property
-    def f264(self) -> 'bool':
-        """Gets the f264 of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The f264 of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._f264
-
-    @f264.setter
-    def f264(self, f264: 'bool'):
-        """Sets the f264 of this V1UserFeatures.
-
-
-        :param f264: The f264 of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._f264 = f264
-
     @property
     def f265(self) -> 'bool':
         """Gets the f265 of this V1UserFeatures.  # noqa: E501
@@ -1482,6 +1393,90 @@ class V1UserFeatures(object):
 
         self._f266 = f266
 
+    @property
+    def f267(self) -> 'bool':
+        """Gets the f267 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f267 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f267
+
+    @f267.setter
+    def f267(self, f267: 'bool'):
+        """Sets the f267 of this V1UserFeatures.
+
+
+        :param f267: The f267 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f267 = f267
+
+    @property
+    def f268(self) -> 'bool':
+        """Gets the f268 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f268 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f268
+
+    @f268.setter
+    def f268(self, f268: 'bool'):
+        """Sets the f268 of this V1UserFeatures.
+
+
+        :param f268: The f268 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f268 = f268
+
+    @property
+    def f269(self) -> 'bool':
+        """Gets the f269 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f269 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f269
+
+    @f269.setter
+    def f269(self, f269: 'bool'):
+        """Sets the f269 of this V1UserFeatures.
+
+
+        :param f269: The f269 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f269 = f269
+
+    @property
+    def f270(self) -> 'bool':
+        """Gets the f270 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f270 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f270
+
+    @f270.setter
+    def f270(self, f270: 'bool'):
+        """Sets the f270 of this V1UserFeatures.
+
+
+        :param f270: The f270 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f270 = f270
+
     @property
     def fair_share(self) -> 'bool':
         """Gets the fair_share of this V1UserFeatures.  # noqa: E501
@@ -1524,27 +1519,6 @@ class V1UserFeatures(object):
 
         self._featured_studios_admin = featured_studios_admin
 
-    @property
-    def gcs_connections_optimized(self) -> 'bool':
-        """Gets the gcs_connections_optimized of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The gcs_connections_optimized of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._gcs_connections_optimized
-
-    @gcs_connections_optimized.setter
-    def gcs_connections_optimized(self, gcs_connections_optimized: 'bool'):
-        """Sets the gcs_connections_optimized of this V1UserFeatures.
-
-
-        :param gcs_connections_optimized: The gcs_connections_optimized of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._gcs_connections_optimized = gcs_connections_optimized
-
     @property
     def job_artifacts_v2(self) -> 'bool':
         """Gets the job_artifacts_v2 of this V1UserFeatures.  # noqa: E501

lightning_sdk/machine.py

@@ -218,3 +218,6 @@ Machine.H200 = Machine(name="H200", slug="lit-h200x-1", family="H200", accelerat
 Machine.H200_X_8 = Machine(name="H200_X_8", slug="lit-h200x-8", family="H200", accelerator_count=8)
 # available B200 machines
 Machine.B200_X_8 = Machine(name="B200_X_8", slug="lit-b200x-8", family="B200", accelerator_count=8)
+
+
+DEFAULT_MACHINE = Machine.CPU.name

lightning_sdk/mmt/base.py

@@ -75,6 +75,7 @@ class _BaseMMT(_BaseJob):
         artifacts_local: Optional[str] = None,  # deprecated in favor of path_mappings
         artifacts_remote: Optional[str] = None,  # deprecated in favor of path_mappings
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
+        reuse_snapshot: bool = True,
     ) -> "_BaseMMT":
         """Run async workloads using a docker image across multiple machines.
 
@@ -119,6 +120,8 @@ class _BaseMMT(_BaseJob):
                 Irrelevant for most machines, required for some of the top-end machines on GCP.
                 If in doubt, set it. Won't have an effect on machines not requiring it.
                 Defaults to 3h
+            reuse_snapshot: Whether the job should reuse a Studio snapshot when multiple jobs for the same Studio are
+                submitted. Turning this off may result in longer job startup times. Defaults to True.
         """
         from lightning_sdk.lightning_cloud.openapi.rest import ApiException
         from lightning_sdk.studio import Studio
@@ -214,6 +217,7 @@ class _BaseMMT(_BaseJob):
             artifacts_local=artifacts_local,
             artifacts_remote=artifacts_remote,
             max_runtime=max_runtime,
+            reuse_snapshot=reuse_snapshot,
         )
         return inst
 
@@ -236,6 +240,7 @@ class _BaseMMT(_BaseJob):
         artifacts_local: Optional[str] = None,  # deprecated in favor of path_mappings
         artifacts_remote: Optional[str] = None,  # deprecated in favor of path_mappings
         max_runtime: Optional[int] = None,
+        reuse_snapshot: bool = True,
     ) -> None:
         """Submit a new multi-machine job to the Lightning AI platform.
 
@@ -271,6 +276,8 @@ class _BaseMMT(_BaseJob):
                 Irrelevant for most machines, required for some of the top-end machines on GCP.
                 If in doubt, set it. Won't have an effect on machines not requiring it.
                 Defaults to 3h
+            reuse_snapshot: Whether the job should reuse a Studio snapshot when multiple jobs for the same Studio are
+                submitted. Turning this off may result in longer job startup times. Defaults to True.
         """
 
     @property

lightning_sdk/mmt/mmt.py

@@ -102,6 +102,7 @@ class MMT(_BaseMMT):
         artifacts_local: Optional[str] = None,
         artifacts_remote: Optional[str] = None,
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
+        reuse_snapshot: bool = True,
     ) -> "MMT":
         """Run async workloads using a docker image across multiple machines.
 
@@ -141,6 +142,8 @@ class MMT(_BaseMMT):
                     }
                 If the path inside the connection is omitted it's assumed to be the root path of that connection.
                 Only applicable when submitting docker jobs.
+            reuse_snapshot: Whether the job should reuse a Studio snapshot when multiple jobs for the same Studio are
+                submitted. Turning this off may result in longer job startup times. Defaults to True.
         """
         ret_val = super().run(
             name=name,
@@ -164,6 +167,7 @@ class MMT(_BaseMMT):
             artifacts_remote=artifacts_remote,
             cluster=cluster,  # deprecated in favor of cloud_account
             max_runtime=max_runtime,
+            reuse_snapshot=reuse_snapshot,
         )
         # required for typing with "MMT"
         assert isinstance(ret_val, cls)
@@ -191,6 +195,7 @@ class MMT(_BaseMMT):
         max_runtime: Optional[int] = None,
         artifacts_local: Optional[str] = None,  # deprecated in favor of path_mappings
         artifacts_remote: Optional[str] = None,  # deprecated in favor of path_mappings
+        reuse_snapshot: bool = True,
     ) -> "MMT":
         """Submit a new multi-machine job to the Lightning AI platform.
 
@@ -231,6 +236,8 @@ class MMT(_BaseMMT):
                 Irrelevant for most machines, required for some of the top-end machines on GCP.
                 If in doubt, set it. Won't have an effect on machines not requiring it.
                 Defaults to 3h
+            reuse_snapshot: Whether the job should reuse a Studio snapshot when multiple jobs for the same Studio are
+                submitted. Turning this off may result in longer job startup times. Defaults to True.
         """
         self._job = self._internal_mmt._submit(
             num_machines=num_machines,
@@ -249,6 +256,7 @@ class MMT(_BaseMMT):
             artifacts_local=artifacts_local,
             artifacts_remote=artifacts_remote,
             max_runtime=max_runtime,
+            reuse_snapshot=reuse_snapshot,
         )
         return self
 

lightning_sdk/mmt/v1.py

@@ -58,6 +58,7 @@ class _MMTV1(_BaseMMT):
         max_runtime: Optional[int] = None,
         artifacts_local: Optional[str] = None,
         artifacts_remote: Optional[str] = None,
+        reuse_snapshot: bool = True,
     ) -> "_MMTV1":
         """Submit a new multi-machine job to the Lightning AI platform.
 
@@ -96,7 +97,8 @@ class _MMTV1(_BaseMMT):
                 Irrelevant for most machines, required for some of the top-end machines on GCP.
                 If in doubt, set it. Won't have an effect on machines not requiring it.
                 Defaults to 3h
-
+            reuse_snapshot: Whether the job should reuse a Studio snapshot when multiple jobs for the same Studio are
+                submitted. Turning this off may result in longer job startup times. Defaults to True.
         """
         raise NotImplementedError("Cannot submit new mmts with MMTV1!")
 

lightning_sdk/mmt/v2.py

@@ -58,6 +58,7 @@ class _MMTV2(_BaseMMT):
         max_runtime: Optional[int] = None,
         artifacts_local: Optional[str] = None,  # deprecated in favor of path_mappings
         artifacts_remote: Optional[str] = None,  # deprecated in favor of path_mappings
+        reuse_snapshot: bool = True,
     ) -> "_MMTV2":
         """Submit a new multi-machine job to the Lightning AI platform.
 
@@ -98,6 +99,8 @@ class _MMTV2(_BaseMMT):
                 Irrelevant for most machines, required for some of the top-end machines on GCP.
                 If in doubt, set it. Won't have an effect on machines not requiring it.
                 Defaults to 3h
+            reuse_snapshot: Whether the job should reuse a Studio snapshot when multiple jobs for the same Studio are
+                submitted. Turning this off may result in longer job startup times. Defaults to True.
         """
         # Command is required if Studio is provided to know what to run
         # Image is mutually exclusive with Studio
@@ -141,6 +144,7 @@ class _MMTV2(_BaseMMT):
             artifacts_local=artifacts_local,
             artifacts_remote=artifacts_remote,
             max_runtime=max_runtime,
+            reuse_snapshot=reuse_snapshot,
         )
         self._job = submitted
         self._name = submitted.name