lightning-sdk 2025.8.1__py3-none-any.whl → 2025.8.6rc0__py3-none-any.whl

lightning_sdk/__init__.py

@@ -32,6 +32,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "2025.08.01"
+__version__ = "2025.08.06rc0"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/api/deployment_api.py

@@ -103,7 +103,7 @@ class ExecHealthCheck(HealthCheck):
         self,
         command: str,
         initial_delay_seconds: int = 0,
-        failure_threshold: int = 600,
+        failure_threshold: int = 3600,
         interval_seconds: int = 1,
         timeout_seconds: int = 30,
     ) -> None:
@@ -132,7 +132,7 @@ class HttpHealthCheck(HealthCheck):
         path: str,
         port: float,
         initial_delay_seconds: int = 0,
-        failure_threshold: int = 600,
+        failure_threshold: int = 3600,
         interval_seconds: int = 1,
         timeout_seconds: int = 30,
     ) -> None:
@@ -538,10 +538,10 @@ def to_health_check(
     # Use Default health check if none is provided
     if not health_check:
         return V1JobHealthCheckConfig(
-            failure_threshold=600,
+            failure_threshold=3600,
             initial_delay_seconds=0,
             interval_seconds=1,
-            timeout_seconds=600,
+            timeout_seconds=60,
         )
 
     health_check_config = V1JobHealthCheckConfig(
@@ -574,6 +574,7 @@ def to_spec(
     include_credentials: Optional[bool] = None,
     cloudspace_id: Optional[None] = None,
     max_runtime: Optional[int] = None,
+    machine_image_version: Optional[str] = None,
 ) -> V1JobSpec:
     if cloud_account is None:
         raise ValueError("The cloud account should be defined.")
@@ -607,6 +608,7 @@ def to_spec(
         quantity=quantity,
         include_credentials=include_credentials,
         cloudspace_id=cloudspace_id,
+        machine_image_version=machine_image_version,
         **optional_spec_kwargs,
     )
 

lightning_sdk/api/job_api.py

@@ -288,6 +288,7 @@ class JobApiV2:
         artifacts_local: Optional[str],  # deprecated in favor of path_mappings
         artifacts_remote: Optional[str],  # deprecated in favor of path_mappings)
         max_runtime: Optional[int] = None,
+        machine_image_version: Optional[str] = None,
     ) -> ProjectIdJobsBody:
         env_vars = []
         if env is not None:
@@ -322,6 +323,7 @@ class JobApiV2:
             image_cluster_credentials=cloud_account_auth,
             image_secret_ref=image_credentials or "",
             path_mappings=path_mappings_list,
+            machine_image_version=machine_image_version,
             **optional_spec_kwargs,
         )
         return ProjectIdJobsBody(name=name, spec=spec)

lightning_sdk/api/mmt_api.py

@@ -129,6 +129,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] = None,
+        machine_image_version: Optional[str] = None,
     ) -> ProjectIdMultimachinejobsBody:
         env_vars = []
         if env is not None:
@@ -163,6 +164,7 @@ class MMTApiV2:
             image_cluster_credentials=cloud_account_auth,
             image_secret_ref=image_credentials or "",
             path_mappings=path_mappings_list,
+            machine_image_version=machine_image_version,
             **optional_spec_kwargs,
         )
         return ProjectIdMultimachinejobsBody(

lightning_sdk/deployment/deployment.py

@@ -171,14 +171,18 @@ class Deployment:
         if self._is_created:
             raise RuntimeError("This deployment has already been started.")
 
+        machine_image_version = None
+
         if isinstance(studio, Studio):
             cloudspace_id = studio._studio.id
             cloud_account = studio._studio.cluster_id
+            machine_image_version = studio._studio.machine_image_version
 
         if isinstance(studio, str):
             studio = Studio(studio)
             cloudspace_id = studio._studio.id
             cloud_account = studio._studio.cluster_id
+            machine_image_version = studio._studio.machine_image_version
 
         if cloud_account is None:
             cloud_account = _resolve_deprecated_cluster(cloud_account, cluster)
@@ -233,6 +237,7 @@ class Deployment:
                     include_credentials=include_credentials if include_credentials is not None else True,
                     cloudspace_id=cloudspace_id,
                     max_runtime=max_runtime,
+                    machine_image_version=machine_image_version,
                 ),
                 strategy=to_strategy(release_strategy),
             ),

lightning_sdk/lightning_cloud/login.py

@@ -43,14 +43,7 @@ class Auth:
         for key in Keys:
             setattr(self, key.suffix, os.environ.get(key.value, None))
 
-        # used by authenticate method
-        self._with_env_var = bool(self.user_id and self.api_key)
-        if self.api_key and not self.user_id:
-            raise ValueError(
-                f"{Keys.USER_ID.value} is missing from env variables. "
-                "To use env vars for authentication both "
-                f"{Keys.USER_ID.value} and {Keys.API_KEY.value} should be set."
-            )
+        self._with_env_var = bool(self.api_key)
 
     def load(self) -> bool:
         """Load credentials from disk and update properties with credentials.

lightning_sdk/pipeline/steps.py

@@ -59,6 +59,7 @@ class DeploymentStep:
         include_credentials: Optional[bool] = None,
         max_runtime: Optional[int] = None,
         wait_for: Optional[Union[str, List[str]]] = DEFAULT,
+        cloud_provider: Optional[Union["CloudProvider", str]] = None,
     ) -> None:
         self.name = name
         self.studio = _get_studio(studio)
@@ -99,11 +100,28 @@ class DeploymentStep:
         self.include_credentials = include_credentials or True
         self.max_runtime = max_runtime
         self.wait_for = wait_for
+        self.cloud_provider = cloud_provider
 
     def to_proto(
         self, teamspace: "Teamspace", cloud_account: str, shared_filesystem: Union[bool, V1SharedFilesystem]
     ) -> V1PipelineStep:
-        _validate_cloud_account(cloud_account, self.cloud_account, shared_filesystem)
+        machine_image_version = None
+
+        studio = _get_studio(self.studio)
+        if isinstance(studio, Studio):
+            machine_image_version = studio._studio.machine_image_version
+
+            if self.cloud_account is None:
+                self.cloud_account = studio.cloud_account
+            elif studio.cloud_account != self.cloud_account:
+                raise ValueError("The provided cloud account doesn't match the studio")
+
+        resolved_cloud_account = CloudAccountApi().resolve_cloud_account(
+            teamspace.id, self.cloud_account, self.cloud_provider, teamspace.default_cloud_account
+        )
+
+        _validate_cloud_account(cloud_account, resolved_cloud_account, shared_filesystem)
+
         return V1PipelineStep(
             name=self.name,
             type=V1PipelineStepType.DEPLOYMENT,
@@ -127,6 +145,7 @@ class DeploymentStep:
                     cloudspace_id=self.studio._studio.id if self.studio else None,
                     include_credentials=self.include_credentials,
                     max_runtime=self.max_runtime,
+                    machine_image_version=machine_image_version,
                 ),
                 strategy=to_strategy(self.release_strategy),
             ),
@@ -186,8 +205,12 @@ class JobStep:
     def to_proto(
         self, teamspace: "Teamspace", cloud_account: str, shared_filesystem: Union[bool, V1SharedFilesystem]
     ) -> V1PipelineStep:
+        machine_image_version = None
+
         studio = _get_studio(self.studio)
         if isinstance(studio, Studio):
+            machine_image_version = studio._studio.machine_image_version
+
             if self.cloud_account is None:
                 self.cloud_account = studio.cloud_account
             elif studio.cloud_account != self.cloud_account:
@@ -215,6 +238,7 @@ class JobStep:
             artifacts_local=None,
             artifacts_remote=None,
             max_runtime=self.max_runtime,
+            machine_image_version=machine_image_version,
         )
 
         return V1PipelineStep(
@@ -277,8 +301,12 @@ class MMTStep:
     def to_proto(
         self, teamspace: "Teamspace", cloud_account: str, shared_filesystem: Union[bool, V1SharedFilesystem]
     ) -> V1PipelineStep:
+        machine_image_version = None
+
         studio = _get_studio(self.studio)
         if isinstance(studio, Studio):
+            machine_image_version = studio._studio.machine_image_version
+
             if self.cloud_account is None:
                 self.cloud_account = studio.cloud_account
             elif studio.cloud_account != self.cloud_account:
@@ -303,6 +331,7 @@ class MMTStep:
             artifacts_local=None,  # deprecated in favor of path_mappings
             artifacts_remote=None,  # deprecated in favor of path_mappings
             max_runtime=self.max_runtime,
+            machine_image_version=machine_image_version,
         )
 
         return V1PipelineStep(

{lightning_sdk-2025.8.1.dist-info → lightning_sdk-2025.8.6rc0.dist-info}/METADATA

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

{lightning_sdk-2025.8.1.dist-info → lightning_sdk-2025.8.6rc0.dist-info}/RECORD

@@ -1,5 +1,5 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=KbrZDMh0tCQGvM0R2Fb_0uMmNCyyP1baFbPah_DVoyI,1145
+lightning_sdk/__init__.py,sha256=I-NEEYsT0l2oHbX8ai0KlhtH5pqVw2A_Hh4oJP29_6o,1148
 lightning_sdk/agents.py,sha256=ly6Ma1j0ZgGPFyvPvMN28JWiB9dATIstFa5XM8pMi6I,1577
 lightning_sdk/ai_hub.py,sha256=iI1vNhgcz_Ff1c3rN1ogN7dK-r-HXRj6NMtS2cA14UA,6925
 lightning_sdk/base_studio.py,sha256=_Pwwl37R9GRd7t-f2kO5aQXiLNrP4sUtUNht2ZkP8LE,3678
@@ -22,12 +22,12 @@ lightning_sdk/api/agents_api.py,sha256=G47TbFo9kYqnBMqdw2RW-lfS1VAUBSXDmzs6fpIEM
 lightning_sdk/api/ai_hub_api.py,sha256=azqDZ-PzasVAcoQHno7k7OO_xFOHQ4NDozxF8jEh83Y,7864
 lightning_sdk/api/base_studio_api.py,sha256=3R8tucZX2e9yKHBcY2rWFRP4dxqLrC6H75vdBDkH0ck,3617
 lightning_sdk/api/cloud_account_api.py,sha256=Rxet7QuXjzkeAok-_azvB0r_koO7K-NioAldmFVsibs,8223
-lightning_sdk/api/deployment_api.py,sha256=ys9HspgKMTehF_h8IIxXO51UNXRjqgPPVctSp0m1iAE,24748
-lightning_sdk/api/job_api.py,sha256=I1ks0g1WrLZa7aN0fZz_LSa_lJmma1XYs7Cz-gaw5CY,16300
+lightning_sdk/api/deployment_api.py,sha256=v2AfoTDkQ-1CBh75FOjFkRpf6yc3U_edDy43uYSn19I,24852
+lightning_sdk/api/job_api.py,sha256=7spFPyotlSQ0Krx6WymslnpSQaaeOmFkydvhtg6p0Ic,16410
 lightning_sdk/api/license_api.py,sha256=XV3RhefyPQDYjwY9AaBZe4rByZTEAnsvLDxcdm9q0Wo,2438
 lightning_sdk/api/lit_container_api.py,sha256=jCJVwd-3MNjejL9FyvH89pzt-SeG3G8qCJD16iTMJAQ,11454
 lightning_sdk/api/llm_api.py,sha256=oc3OB1l0Jca5Nc6MAkfI-Nq-42hXJkgmkSqn22J4TeA,9261
-lightning_sdk/api/mmt_api.py,sha256=FQSEbe2Pui5amqnNTyo3FJbfNGaAr6bsYSF5fUGs8FI,10589
+lightning_sdk/api/mmt_api.py,sha256=hIBsGiJ2qn5UjcHDxP5WUyKGT_AIFfpSHrQVwg0afBw,10699
 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=LbZQ2HtOHNcLXLtrOtG1jhkqNOhqfelBxIQy6lSbirQ,32306
@@ -65,7 +65,7 @@ lightning_sdk/cli/deploy/_auth.py,sha256=ccIzl0W9hJtFmhdEAn1xcBGIHQfO4d_k8grTbq4
 lightning_sdk/cli/deploy/devbox.py,sha256=HOQMWehwaveawVTyBDaZhNZs_NL5g4RCWs_lf0hIW5U,6800
 lightning_sdk/cli/deploy/serve.py,sha256=K2e7gz1gg_UTr0j5FuI1_EKRa0t6hRaoHGpwFAXRSso,15040
 lightning_sdk/deployment/__init__.py,sha256=dXsa4psDzFYFklsq3JC-2V_L4FQjGZnQAf-ZiVlqG9c,545
-lightning_sdk/deployment/deployment.py,sha256=qFmuFw96kcEp6dubvhDeaDc2n9ylpNCwLL1_ImtQ3O0,21526
+lightning_sdk/deployment/deployment.py,sha256=7Sk4ORKmgMDk7Owpk_q-89ws8-Y9orX0jRZF99tNv1E,21775
 lightning_sdk/job/__init__.py,sha256=1MxjQ6rHkyUHCypSW9RuXuVMVH11WiqhIXcU2LCFMwE,64
 lightning_sdk/job/base.py,sha256=GAE7wvTH9e_F9bU8sogT88EMCP4D0d1deZMeVmBtWTE,20856
 lightning_sdk/job/job.py,sha256=jPlW3u34lJ6tpszm-Q6EHzWrfMIpQfM7-z8UCFFVyaE,14780
@@ -75,7 +75,7 @@ lightning_sdk/job/work.py,sha256=aRknNDja-96qQaYw0fNboEGtOZCmvztYEzUPxLNAt8g,245
 lightning_sdk/lightning_cloud/__init__.py,sha256=o91SMAlwr4Ke5ESe8fHjqXcj31_h7rT-MlFoXA-n2EI,173
 lightning_sdk/lightning_cloud/__version__.py,sha256=lOfmWHtjmiuSG28TbKQqd2B3nwmSGOlKVFwhaj_cRJk,23
 lightning_sdk/lightning_cloud/env.py,sha256=XZXpF4sD9jlB8DY0herTy_8XiUJuDVjxy5APjRD2_aU,1379
-lightning_sdk/lightning_cloud/login.py,sha256=k4iE5qLUMB8_VeQa1sNz7jPCWLOV5S7j9Q776J7ZXhY,7631
+lightning_sdk/lightning_cloud/login.py,sha256=HGfixrD_j5pifen3KvnYGmTE3ZeR-z8AXN3aSN0x73U,7275
 lightning_sdk/lightning_cloud/rest_client.py,sha256=K2J_QXihpbp5tVzs9PzffgMCQY1NeH5tRpuJBYj27vU,7125
 lightning_sdk/lightning_cloud/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lightning_sdk/lightning_cloud/cli/__main__.py,sha256=aHrigVV0qIp74MuYTdN6uO8a9Xp6E1aOwdR8w1fG-7c,688
@@ -1100,7 +1100,7 @@ lightning_sdk/pipeline/__init__.py,sha256=Sja_0NJ8vgh-2ThSVP3WDI9a9qghrWd21LkaQp
 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=H2emV6ZURRAlK7TeHvos9KXvbphYOi7U6PyBoqigZoU,13013
+lightning_sdk/pipeline/steps.py,sha256=wCpZr8MOXCEWJ3_WMwtuxMpqAG6ZT8ep2vnA9Jhl8hs,14164
 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=we5HC_o74J4Y6fSP_31jIizi_I_1FO_Rb2qblspD9eE,7855
@@ -1111,9 +1111,9 @@ lightning_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 lightning_sdk/utils/dynamic.py,sha256=glUTO1JC9APtQ6Gr9SO02a3zr56-sPAXM5C3NrTpgyQ,1959
 lightning_sdk/utils/enum.py,sha256=h2JRzqoBcSlUdanFHmkj_j5DleBHAu1esQYUsdNI-hU,4106
 lightning_sdk/utils/resolve.py,sha256=JjHwWNQq375d1JBcVkJrZWkcTehvuTxF2h6awvSUDoo,8094
-lightning_sdk-2025.8.1.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-2025.8.1.dist-info/METADATA,sha256=18-mzeoeFtsfSJ2Cyb7c8-qMr7_rbFoucRZGFLpw61M,4107
-lightning_sdk-2025.8.1.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-lightning_sdk-2025.8.1.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
-lightning_sdk-2025.8.1.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-2025.8.1.dist-info/RECORD,,
+lightning_sdk-2025.8.6rc0.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-2025.8.6rc0.dist-info/METADATA,sha256=e_zqAI7G1vHn8qXPosKEI2zyIKCuX_9zW1ux7RjYCK4,4110
+lightning_sdk-2025.8.6rc0.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+lightning_sdk-2025.8.6rc0.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-2025.8.6rc0.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-2025.8.6rc0.dist-info/RECORD,,