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

lightning_sdk/__init__.py

@@ -35,7 +35,7 @@ __all__ = [
     "VM",
 ]
 
-__version__ = "2025.10.23"
+__version__ = "2025.10.27"
 
 _version_checker = VersionChecker()
 _version_checker.check_and_prompt_upgrade(__version__)

lightning_sdk/api/mmt_api.py

@@ -88,6 +88,7 @@ class MMTApiV2:
         artifacts_local: Optional[str],  # deprecated in favor of path_mappings
         artifacts_remote: Optional[str],  # deprecated in favor of path_mappings
         max_runtime: Optional[int],
+        reuse_snapshot: bool,
     ) -> V1MultiMachineJob:
         body = self._create_mmt_body(
             name=name,
@@ -106,6 +107,7 @@ class MMTApiV2:
             artifacts_local=artifacts_local,  # deprecated in favor of path_mappings
             artifacts_remote=artifacts_remote,  # deprecated in favor of path_mappings
             max_runtime=max_runtime,
+            reuse_snapshot=reuse_snapshot,
         )
 
         job: V1MultiMachineJob = self._client.jobs_service_create_multi_machine_job(project_id=teamspace_id, body=body)
@@ -128,6 +130,7 @@ class MMTApiV2:
         path_mappings: Optional[Dict[str, str]],
         artifacts_local: Optional[str],  # deprecated in favor of path_mappings
         artifacts_remote: Optional[str],  # deprecated in favor of path_mappings
+        reuse_snapshot: bool,
         max_runtime: Optional[int] = None,
         machine_image_version: Optional[str] = None,
     ) -> ProjectIdMultimachinejobsBody:
@@ -138,7 +141,7 @@ class MMTApiV2:
 
         instance_name = _machine_to_compute_name(machine)
 
-        run_id = __GLOBAL_LIGHTNING_UNIQUE_IDS_STORE__[studio_id] if studio_id is not None else ""
+        run_id = __GLOBAL_LIGHTNING_UNIQUE_IDS_STORE__[studio_id] if (studio_id is not None and reuse_snapshot) else ""
 
         path_mappings_list = resolve_path_mappings(
             mappings=path_mappings or {},

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

lightning_sdk/pipeline/steps.py

@@ -175,6 +175,7 @@ class JobStep:
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         wait_for: Union[str, List[str], None] = DEFAULT,
+        reuse_snapshot: bool = True,
     ) -> None:
         self.name = name
         self.machine = machine or Machine.CPU
@@ -201,6 +202,7 @@ class JobStep:
         self.path_mappings = path_mappings
         self.max_runtime = max_runtime
         self.wait_for = wait_for
+        self.reuse_snapshot = reuse_snapshot
 
     def to_proto(
         self, teamspace: "Teamspace", cloud_account: str, shared_filesystem: Union[bool, V1SharedFilesystem]
@@ -239,7 +241,7 @@ class JobStep:
             artifacts_remote=None,
             max_runtime=self.max_runtime,
             machine_image_version=machine_image_version,
-            reuse_snapshot=True,
+            reuse_snapshot=self.reuse_snapshot,
         )
 
         return V1PipelineStep(
@@ -273,6 +275,7 @@ class MMTStep:
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         wait_for: Optional[Union[str, List[str]]] = DEFAULT,
+        reuse_snapshot: bool = True,
     ) -> None:
         self.machine = machine or Machine.CPU
         self.num_machines = num_machines
@@ -298,6 +301,7 @@ class MMTStep:
         self.path_mappings = path_mappings
         self.max_runtime = max_runtime
         self.wait_for = wait_for
+        self.reuse_snapshot = reuse_snapshot
 
     def to_proto(
         self, teamspace: "Teamspace", cloud_account: str, shared_filesystem: Union[bool, V1SharedFilesystem]
@@ -333,6 +337,7 @@ class MMTStep:
             artifacts_remote=None,  # deprecated in favor of path_mappings
             max_runtime=self.max_runtime,
             machine_image_version=machine_image_version,
+            reuse_snapshot=self.reuse_snapshot,
         )
 
         return V1PipelineStep(

lightning_sdk/plugin.py

@@ -128,6 +128,7 @@ class JobsPlugin(_Plugin):
         machine: Machine = Machine.CPU,
         cloud_compute: Optional[Machine] = None,
         interruptible: bool = False,
+        reuse_snapshot: bool = True,
     ) -> Job:
         """Launches an asynchronous job.
 
@@ -137,6 +138,8 @@ class JobsPlugin(_Plugin):
             machine: The machine to run the job on.
             interruptible: Whether to run the job on an interruptible machine.
                 These are cheaper but can be preempted at any time.
+            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.
         """
         if not name:
             name = _run_name("job")
@@ -151,6 +154,7 @@ class JobsPlugin(_Plugin):
             teamspace=self._studio.teamspace,
             cloud_account=self._studio.cloud_account,
             interruptible=interruptible,
+            reuse_snapshot=reuse_snapshot,
         )
 
 

lightning_sdk/studio.py

@@ -627,6 +627,7 @@ class Studio:
         command: str,
         env: Optional[Dict[str, str]] = None,
         interruptible: bool = False,
+        reuse_snapshot: bool = True,
     ) -> "Job":
         """Run async workloads using the compute environment from your studio.
 
@@ -636,6 +637,8 @@ class Studio:
             command: The command to run inside your job.
             env: Environment variables to set inside the job.
             interruptible: Whether the job should run on interruptible instances. They are cheaper but can be preempted.
+            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.job import Job
 
@@ -649,6 +652,7 @@ class Studio:
             cloud_account=self.cloud_account,
             env=env,
             interruptible=interruptible,
+            reuse_snapshot=reuse_snapshot,
         )
 
     def run_mmt(

{lightning_sdk-2025.10.23.dist-info → lightning_sdk-2025.10.27.dist-info}/METADATA

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

{lightning_sdk-2025.10.23.dist-info → lightning_sdk-2025.10.27.dist-info}/RECORD

@@ -1,5 +1,5 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=1Fy5JKAINN88i-aGaLpdZ0CcDOVMctjDeVtiyxy0IOM,1262
+lightning_sdk/__init__.py,sha256=ynn4c3czFAQaLzR5sDdZx7dFJ9S1CMTNEyW-ksxckzc,1262
 lightning_sdk/agents.py,sha256=ly6Ma1j0ZgGPFyvPvMN28JWiB9dATIstFa5XM8pMi6I,1577
 lightning_sdk/ai_hub.py,sha256=iI1vNhgcz_Ff1c3rN1ogN7dK-r-HXRj6NMtS2cA14UA,6925
 lightning_sdk/base_studio.py,sha256=yMvZ1dDq_avW3iu98CLEo9W9UL54mx4C6FC43IKLWAE,4691
@@ -10,11 +10,11 @@ lightning_sdk/machine.py,sha256=GBx4w8qbx_V7ji4-f-uiZ04Z202muORW83Ku0nIbLfA,9614
 lightning_sdk/models.py,sha256=ur1My3oivAe0p2tAYtOAGM_1tPBht3Oh-xLun-oUK08,7861
 lightning_sdk/organization.py,sha256=jizjG4oW_oxkP2Qz4vL5cle4Cg12xB_vf4eT-bnvLP0,1352
 lightning_sdk/owner.py,sha256=t5svD2it4C9pbSpVuG9WJL46CYi37JXNziwnXxhiU5U,1361
-lightning_sdk/plugin.py,sha256=f3P5-xZY6x-MX0Fs2z_Q2erSxPSiHZARO0BVkCezHw4,15192
+lightning_sdk/plugin.py,sha256=_XIjgxAmqFtbYjqJc8ItjhBsjE-XWuVUgsPj4UD_d_8,15492
 lightning_sdk/sandbox.py,sha256=_NvnWotEXW2rBiVFZZ4krKXxVjuAqfNh04qELSM0-Pg,5786
 lightning_sdk/serve.py,sha256=uW7zLhQ3X90ifetpxzTb8FNxifv5vIs7qZlgfEjVKzk,11794
 lightning_sdk/status.py,sha256=lLGAuSvXBoXQFEEsEYwdCi0RcSNatUn5OPjJVjDtoM0,386
-lightning_sdk/studio.py,sha256=7Z-zhIQdXwANfyJrHUFDx0_2pvVpf35dmJzcxXV-Oo4,36399
+lightning_sdk/studio.py,sha256=fvHZGF3jg7pXWHeBxTzKh3W1rpu-ZE7ITlMTxjp97Zc,36699
 lightning_sdk/teamspace.py,sha256=iYaEaZH8N8Yj4XXj6M4zIH0tsjPVrAf_BzAAv6Jt9VI,26556
 lightning_sdk/user.py,sha256=TSYh38rxoi7qKOfrK2JYh_Nknya2Kbz2ngDIY85fFOY,1778
 lightning_sdk/api/__init__.py,sha256=xrp_RNECJGQtL5rZHF69WOzEuEIbWSLtjWAJAz4R5K4,500
@@ -27,7 +27,7 @@ lightning_sdk/api/job_api.py,sha256=_wB666oyRd6P3Gm8Q0MERQx3Av9QuUpi0XG96x2GhP4,
 lightning_sdk/api/license_api.py,sha256=nr72JMjGxJktvZmzF1yTz-8kwbeLSBjq1IqHpxV820w,1344
 lightning_sdk/api/lit_container_api.py,sha256=jCJVwd-3MNjejL9FyvH89pzt-SeG3G8qCJD16iTMJAQ,11454
 lightning_sdk/api/llm_api.py,sha256=8ouowlX-Ld0hRvueIS6fQqaEJLQMYskvhKV5-Z_A6gE,11886
-lightning_sdk/api/mmt_api.py,sha256=adVGNFpinnsE3lkY4QZhpWxq4ZQjhyC1XTmOhpBWXvk,10691
+lightning_sdk/api/mmt_api.py,sha256=yI_iFtxGd1g0pW82xQWY80jSYVmF_xU-eT1-LgUA-KI,10815
 lightning_sdk/api/org_api.py,sha256=Ze3z_ATVrukobujV5YdC42DKj45Vuwl7X52q_Vr-o3U,803
 lightning_sdk/api/pipeline_api.py,sha256=rJYp_FN7uUjC5xbc6K67l2eRSmVuOkijd5i8Nm5BF7I,4621
 lightning_sdk/api/studio_api.py,sha256=itQqpC9Be2O69CX7eful3Q2t66NrtMPAizhxUDImpKA,39927
@@ -1227,15 +1227,15 @@ lightning_sdk/llm/__init__.py,sha256=ErZva0HqN2iPtK_6hI6GN7A_HPGNrHo3wYh7vyFdO3Q
 lightning_sdk/llm/llm.py,sha256=pjBs8fbGZjs0tQRKHx-WdfTmeECKAwI_T1ShVBgdyO4,20836
 lightning_sdk/llm/public_assistants.py,sha256=k0yc41AyrKImnRa8Fv-ow9javnlrRQeP63507p2VybA,1579
 lightning_sdk/mmt/__init__.py,sha256=ExMu90-96bGBnyp5h0CErQszUGB1-PcjC4-R8_NYbeY,117
-lightning_sdk/mmt/base.py,sha256=iqVudKDxazomuezj6l7pa-m9I1EQnM82OKs4ZQbo4Ck,16434
-lightning_sdk/mmt/mmt.py,sha256=CZK0KopnW8j-oez464bYMxc0UbqzsYazsNTq1vBtdlg,14982
-lightning_sdk/mmt/v1.py,sha256=tIMp7Oy5243LEExbLEAm8frKqffdiNA5WjvTdr4NOio,8858
-lightning_sdk/mmt/v2.py,sha256=8DDXYpr5U1Nh8IlMRl3ePBartqh1At_-FcTQ-gsvQEw,10869
+lightning_sdk/mmt/base.py,sha256=fP5bcZewLgUCLTH82CTauKSH6J4994pS8ZuFnzzN9BI,16991
+lightning_sdk/mmt/mmt.py,sha256=UqRWtXUAF8XYbatEAwJOW3sOiHcxQ0CxNLldcybW54s,15582
+lightning_sdk/mmt/v1.py,sha256=PRyqgoHUlzZWsjUv2q8vHT00FxyCSqqVt5p5R9tu36s,9114
+lightning_sdk/mmt/v2.py,sha256=iQOXb631M1zk63nJzomG-7fh0BLScZz98ndKjn4NMYo,11169
 lightning_sdk/pipeline/__init__.py,sha256=Sja_0NJ8vgh-2ThSVP3WDI9a9qghrWd21LkaQp4Zsp8,378
 lightning_sdk/pipeline/pipeline.py,sha256=A11Q_bRhiQm89n-VcdaXBJsvK8Yb_EY_IijOGnBmPoI,6163
 lightning_sdk/pipeline/printer.py,sha256=fsewFE_nnI-x5KscviYvQbKgNcgMyn-uT0UWlArGn-8,4828
 lightning_sdk/pipeline/schedule.py,sha256=J6lfjeuUMqGhX6xo6-qIliyi7KmB4e8BimTKJh7cJXY,20059
-lightning_sdk/pipeline/steps.py,sha256=70jRyZnIQICIwSQjcDGHlCbTWCuuALt193jC1PMl8j4,14197
+lightning_sdk/pipeline/steps.py,sha256=vo7pC4gmcx_xMfWwAIctBk39QCxzmbjvDVwWSnTswI4,14424
 lightning_sdk/pipeline/utils.py,sha256=3L7UgY78w2VAVSQmelQNnz_X9zJdFo1NJBtez221ge0,3923
 lightning_sdk/services/__init__.py,sha256=wi1yv0SCnfJub5tOq8y9SblK3-CEseHJuvH-HmtAy7U,229
 lightning_sdk/services/file_endpoint.py,sha256=F_ivy1cPvyvdE5C4aJY7PBSVGn6Dj88QqObbwSvKfYQ,7673
@@ -1249,9 +1249,9 @@ lightning_sdk/utils/license.py,sha256=r68pBJPcjCQ_I-U-0dXxguk9vCskxw8EEhA-Ug0z8c
 lightning_sdk/utils/names.py,sha256=1EuXbIh7wldkDp1FG10oz9vIOyWrpGWeFFVy-DQBgzA,18162
 lightning_sdk/utils/progress.py,sha256=bLWw39fzq29PMWoFXaPIVfoS3Ug245950oWOFJ2ZaiU,12596
 lightning_sdk/utils/resolve.py,sha256=ukC-Zn35gNZV-fQ-ZyUgZkRPFb8nwsFh_aN7YcJ0sl8,10613
-lightning_sdk-2025.10.23.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-2025.10.23.dist-info/METADATA,sha256=5DrRQImxzMNSt5ZNWxoJLXBhFP0YQS1xE6H7TvY30Js,4131
-lightning_sdk-2025.10.23.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-lightning_sdk-2025.10.23.dist-info/entry_points.txt,sha256=OoZa4Fc8NMs6GSN0cdA1J8e6couzAcL82CbM1yo4f_M,122
-lightning_sdk-2025.10.23.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-2025.10.23.dist-info/RECORD,,
+lightning_sdk-2025.10.27.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-2025.10.27.dist-info/METADATA,sha256=YHS7qReBt5GlmYyMjd6lnBlLQskh9x3Fr0uaQbP9IPo,4131
+lightning_sdk-2025.10.27.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+lightning_sdk-2025.10.27.dist-info/entry_points.txt,sha256=OoZa4Fc8NMs6GSN0cdA1J8e6couzAcL82CbM1yo4f_M,122
+lightning_sdk-2025.10.27.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-2025.10.27.dist-info/RECORD,,