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

lightning_sdk/api/mmt_api.py

@@ -0,0 +1,147 @@
+import time
+from typing import TYPE_CHECKING, Dict, Optional
+
+from lightning_sdk.api.utils import (
+    _COMPUTE_NAME_TO_MACHINE,
+    _MACHINE_TO_COMPUTE_NAME,
+)
+from lightning_sdk.api.utils import (
+    _get_cloud_url as _cloud_url,
+)
+from lightning_sdk.constants import __GLOBAL_LIGHTNING_UNIQUE_IDS_STORE__
+from lightning_sdk.lightning_cloud.openapi import (
+    MultimachinejobsIdBody,
+    ProjectIdMultimachinejobsBody,
+    V1EnvVar,
+    V1JobSpec,
+    V1MultiMachineJob,
+    V1MultiMachineJobState,
+)
+from lightning_sdk.lightning_cloud.rest_client import LightningClient
+from lightning_sdk.machine import Machine
+
+if TYPE_CHECKING:
+    from lightning_sdk.status import Status
+
+
+class MMTApi:
+    mmt_state_unspecified = "MultiMachineJob_STATE_UNSPECIFIED"
+    mmt_state_running = "MultiMachineJob_STATE_RUNNING"
+    mmt_state_stopped = "MultiMachineJob_STATE_STOPPED"
+    mmt_state_deleted = "MultiMachineJob_STATE_DELETED"
+    mmt_state_failed = "MultiMachineJob_STATE_FAILED"
+    mmt_state_completed = "MultiMachineJob_STATE_COMPLETED"
+
+    def __init__(self) -> None:
+        self._cloud_url = _cloud_url()
+        self._client = LightningClient(max_tries=7)
+
+    def submit_job(
+        self,
+        name: str,
+        num_machines: int,
+        command: Optional[str],
+        cluster_id: Optional[str],
+        teamspace_id: str,
+        studio_id: Optional[str],
+        image: Optional[str],
+        machine: Machine,
+        interruptible: bool,
+        env: Optional[Dict[str, str]],
+        image_credentials: Optional[str],
+        cluster_auth: bool,
+        artifacts_local: Optional[str],
+        artifacts_remote: Optional[str],
+    ) -> V1MultiMachineJob:
+        env_vars = []
+        if env is not None:
+            for k, v in env.items():
+                env_vars.append(V1EnvVar(name=k, value=v))
+
+        instance_name = _MACHINE_TO_COMPUTE_NAME[machine]
+
+        run_id = __GLOBAL_LIGHTNING_UNIQUE_IDS_STORE__[studio_id] if studio_id is not None else ""
+
+        spec = V1JobSpec(
+            cloudspace_id=studio_id or "",
+            cluster_id=cluster_id or "",
+            command=command or "",
+            env=env_vars,
+            image=image or "",
+            instance_name=instance_name,
+            run_id=run_id,
+            spot=interruptible,
+            image_cluster_credentials=cluster_auth,
+            image_secret_ref=image_credentials or "",
+            artifacts_source=artifacts_local or "",
+            artifacts_destination=artifacts_remote or "",
+        )
+        body = ProjectIdMultimachinejobsBody(name=name, spec=spec, cluster_id=cluster_id or "", machines=num_machines)
+
+        job: V1MultiMachineJob = self._client.jobs_service_create_multi_machine_job(project_id=teamspace_id, body=body)
+        return job
+
+    def get_job_by_name(self, name: str, teamspace_id: str) -> V1MultiMachineJob:
+        job: V1MultiMachineJob = self._client.jobs_service_get_multi_machine_job_by_name(
+            project_id=teamspace_id, name=name
+        )
+        return job
+
+    def get_job(self, job_id: str, teamspace_id: str) -> V1MultiMachineJob:
+        job: V1MultiMachineJob = self._client.jobs_service_get_multi_machine_job(project_id=teamspace_id, id=job_id)
+        return job
+
+    def stop_job(self, job_id: str, teamspace_id: str) -> None:
+        from lightning_sdk.status import Status
+
+        current_job = self.get_job(job_id=job_id, teamspace_id=teamspace_id)
+
+        current_state = self._job_state_to_external(current_job.desired_state)
+
+        if current_state in (
+            Status.Stopped,
+            Status.Completed,
+            Status.Failed,
+        ):
+            return
+
+        if current_state != Status.Stopped:
+            update_body = MultimachinejobsIdBody(desired_state=self.mmt_state_stopped)
+            self._client.jobs_service_update_multi_machine_job(body=update_body, project_id=teamspace_id, id=job_id)
+
+        while True:
+            current_job = self.get_job(job_id=job_id, teamspace_id=teamspace_id)
+            if self._job_state_to_external(current_job.desired_state) in (
+                Status.Stopped,
+                Status.Completed,
+                Status.Stopped,
+                Status.Failed,
+            ):
+                break
+            time.sleep(1)
+
+    def delete_job(self, job_id: str, teamspace_id: str) -> None:
+        self._client.jobs_service_delete_multi_machine_job(project_id=teamspace_id, id=job_id)
+
+    def _job_state_to_external(self, state: V1MultiMachineJobState) -> "Status":
+        from lightning_sdk.status import Status
+
+        if str(state) == self.mmt_state_unspecified:
+            return Status.Pending
+        if str(state) == self.mmt_state_running:
+            return Status.Running
+        if str(state) == self.mmt_state_stopped:
+            return Status.Stopped
+        if str(state) == self.mmt_state_completed:
+            return Status.Completed
+        if str(state) == self.mmt_state_failed:
+            return Status.Failed
+        return Status.Pending
+
+    def _get_job_machine_from_spec(self, spec: V1JobSpec) -> "Machine":
+        instance_name = spec.instance_name
+        instance_type = spec.instance_type
+
+        return _COMPUTE_NAME_TO_MACHINE.get(
+            instance_type, _COMPUTE_NAME_TO_MACHINE.get(instance_name, instance_type or instance_name)
+        )

lightning_sdk/api/teamspace_api.py

@@ -1,5 +1,4 @@
 import os
-from dataclasses import dataclass
 from pathlib import Path
 from typing import Dict, List, Optional
 
@@ -260,22 +259,16 @@ class TeamspaceApi:
         name: str,
         version: str,
         download_dir: Path,
-        teamspace_id: str,
+        teamspace_name: str,
+        teamspace_owner_name: str,
         progress_bar: bool = True,
     ) -> List[str]:
         return _download_model_files(
             client=self._client,
-            teamspace_id=teamspace_id,
+            teamspace_name=teamspace_name,
+            teamspace_owner_name=teamspace_owner_name,
             name=name,
             version=version,
             download_dir=download_dir,
             progress_bar=progress_bar,
         )
-
-
-@dataclass
-class UploadedModelInfo:
-    name: str
-    version: str
-    teamspace: str
-    cluster: str

lightning_sdk/api/utils.py

@@ -513,7 +513,8 @@ def _get_model_version(client: LightningClient, teamspace_id: str, name: str, ve
 
 def _download_model_files(
     client: LightningClient,
-    teamspace_id: str,
+    teamspace_name: str,
+    teamspace_owner_name: str,
     name: str,
     version: str,
     download_dir: Path,
@@ -521,7 +522,9 @@ def _download_model_files(
     num_workers: int = 20,
 ) -> List[str]:
     api = ModelsStoreApi(client.api_client)
-    response = api.models_store_get_model_files(project_id=teamspace_id, name=name, version=version)
+    response = api.models_store_get_model_files(
+        project_name=teamspace_name, project_owner_name=teamspace_owner_name, name=name, version=version
+    )
 
     pbar = None
     if progress_bar:
@@ -541,7 +544,7 @@ def _download_model_files(
             client=client,
             model_id=response.model_id,
             version=response.version,
-            teamspace_id=teamspace_id,
+            teamspace_id=response.project_id,
             remote_path=filepath,
             file_path=str(local_file),
             num_workers=num_workers,

lightning_sdk/cli/download.py

@@ -4,8 +4,8 @@ from pathlib import Path
 from typing import Optional
 
 from lightning_sdk.cli.exceptions import StudioCliError
-from lightning_sdk.cli.models import _get_teamspace, _parse_model_name
 from lightning_sdk.cli.studios_menu import _StudiosMenu
+from lightning_sdk.models import download_model
 from lightning_sdk.studio import Studio
 from lightning_sdk.utils.resolve import _get_authed_user, skip_studio_init
 
@@ -21,10 +21,8 @@ class _Downloads(_StudiosMenu):
             This should have the format <ORGANIZATION-NAME>/<TEAMSPACE-NAME>/<MODEL-NAME>.
           download_dir: The directory where the Model should be downloaded.
         """
-        org_name, teamspace_name, model_name = _parse_model_name(name)
-        teamspace = _get_teamspace(name=teamspace_name, organization=org_name)
-        teamspace.download_model(
-            name=model_name,
+        download_model(
+            name=name,
             download_dir=download_dir,
             progress_bar=True,
         )

lightning_sdk/cli/mmt.py

@@ -0,0 +1,137 @@
+from typing import Dict, Optional
+
+from fire import Fire
+
+from lightning_sdk._mmt import MMT
+from lightning_sdk.api.studio_api import _cloud_url
+from lightning_sdk.lightning_cloud.login import Auth
+from lightning_sdk.machine import Machine
+from lightning_sdk.teamspace import Teamspace
+
+_MACHINE_VALUES = tuple([machine.value for machine in Machine])
+
+
+class MMTCLI:
+    """Command line interface (CLI) to interact with/manage Lightning AI MMT."""
+
+    def __init__(self) -> None:
+        # Need to set the docstring here for f-strings to work.
+        # Sadly this is the only way to really show options as f-strings are not allowed as docstrings directly
+        # and fire does not show values for literals, just that it is a literal.
+        docstr = f"""Run async workloads on multiple machines using a docker image.
+
+        Args:
+            name: The name of the job. Needs to be unique within the teamspace.
+            num_machines: The number of Machines to run on. Defaults to 2 Machines
+            machine: The machine type to run the job on. One of {", ".join(_MACHINE_VALUES)}. Defaults to CPU
+            command: The command to run inside your job. Required if using a studio. Optional if using an image.
+                If not provided for images, will run the container entrypoint and default command.
+            studio: The studio env to run the job with. Mutually exclusive with image.
+            image: The docker image to run the job with. Mutually exclusive with studio.
+            teamspace: The teamspace the job should be associated with. Defaults to the current teamspace.
+            org: The organization owning the teamspace (if any). Defaults to the current organization.
+            user: The user owning the teamspace (if any). Defaults to the current user.
+            cluster: The cluster to run the job on. Defaults to the studio cluster if running with studio compute env.
+                If not provided will fall back to the teamspaces default cluster.
+            env: Environment variables to set inside the job.
+            interruptible: Whether the job should run on interruptible instances. They are cheaper but can be preempted.
+            image_credentials: The credentials used to pull the image. Required if the image is private.
+                This should be the name of the respective credentials secret created on the Lightning AI platform.
+            cluster_auth: Whether to authenticate with the cluster to pull the image.
+                Required if the registry is part of a cluster provider (e.g. ECR).
+            artifacts_local: The path of inside the docker container, you want to persist images from.
+                CAUTION: When setting this to "/", it will effectively erase your container.
+                Only supported for jobs with a docker image compute environment.
+            artifacts_remote: The remote storage to persist your artifacts to.
+                Should be of format <CONNECTION_TYPE>:<CONNECTION_NAME>:<PATH_WITHIN_CONNECTION>.
+                PATH_WITHIN_CONNECTION hereby is a path relative to the connection's root.
+                E.g. efs:data:some-path would result in an EFS connection named `data` and to the path `some-path`
+                within it.
+                Note that the connection needs to be added to the teamspace already in order for it to be found.
+                Only supported for jobs with a docker image compute environment.
+        """
+        # TODO: the docstrings from artifacts_local and artifacts_remote don't show up completely,
+        # might need to switch to explicit cli definition
+        self.run.__func__.__doc__ = docstr
+
+    def login(self) -> None:
+        """Login to Lightning AI Studios."""
+        auth = Auth()
+        auth.clear()
+
+        try:
+            auth.authenticate()
+        except ConnectionError:
+            raise RuntimeError(f"Unable to connect to {_cloud_url()}. Please check your internet connection.") from None
+
+    def logout(self) -> None:
+        """Logout from Lightning AI Studios."""
+        auth = Auth()
+        auth.clear()
+
+    # TODO: sadly, fire displays both Optional[type] and Union[type, None] as Optional[Optional]
+    # see https://github.com/google/python-fire/pull/513
+    # might need to move to different cli library
+    def run(
+        self,
+        name: Optional[str] = None,
+        num_machines: int = 2,
+        machine: Optional[str] = None,
+        command: Optional[str] = None,
+        studio: Optional[str] = None,
+        image: Optional[str] = None,
+        teamspace: Optional[str] = None,
+        org: Optional[str] = None,
+        user: Optional[str] = None,
+        cluster: Optional[str] = None,
+        env: Optional[Dict[str, str]] = None,
+        interruptible: bool = False,
+        image_credentials: Optional[str] = None,
+        cluster_auth: bool = False,
+        artifacts_local: Optional[str] = None,
+        artifacts_remote: Optional[str] = None,
+    ) -> None:
+        if name is None:
+            from datetime import datetime
+
+            timestr = datetime.now().strftime("%Y-%m-%d-%H-%M-%S")
+            name = f"mmt-{timestr}"
+
+        if machine is None:
+            # TODO: infer from studio
+            machine = "CPU"
+        machine_enum = Machine(machine.upper())
+
+        teamspace = Teamspace(name=teamspace, org=org, user=user)
+        if cluster is None:
+            cluster = teamspace.default_cluster
+
+        if image is None:
+            raise RuntimeError("Currently only docker images are specified")
+        MMT.run(
+            name=name,
+            num_machines=num_machines,
+            machine=machine_enum,
+            command=command,
+            studio=studio,
+            image=image,
+            teamspace=teamspace,
+            org=org,
+            user=user,
+            cluster=cluster,
+            env=env,
+            interruptible=interruptible,
+            image_credentials=image_credentials,
+            cluster_auth=cluster_auth,
+            artifacts_local=artifacts_local,
+            artifacts_remote=artifacts_remote,
+        )
+
+
+def main_cli() -> None:
+    """CLI entrypoint."""
+    Fire(MMTCLI(), name="_mmt")
+
+
+if __name__ == "__main__":
+    main_cli()

lightning_sdk/cli/run.py

@@ -40,7 +40,19 @@ class _Run:
                 This should be the name of the respective credentials secret created on the Lightning AI platform.
             cluster_auth: Whether to authenticate with the cluster to pull the image.
                 Required if the registry is part of a cluster provider (e.g. ECR).
+            artifacts_local: The path of inside the docker container, you want to persist images from.
+                CAUTION: When setting this to "/", it will effectively erase your container.
+                Only supported for jobs with a docker image compute environment.
+            artifacts_remote: The remote storage to persist your artifacts to.
+                Should be of format <CONNECTION_TYPE>:<CONNECTION_NAME>:<PATH_WITHIN_CONNECTION>.
+                PATH_WITHIN_CONNECTION hereby is a path relative to the connection's root.
+                E.g. efs:data:some-path would result in an EFS connection named `data` and to the path `some-path`
+                within it.
+                Note that the connection needs to be added to the teamspace already in order for it to be found.
+                Only supported for jobs with a docker image compute environment.
         """
+        # TODO: the docstrings from artifacts_local and artifacts_remote don't show up completely,
+        # might need to switch to explicit cli definition
         self.job.__func__.__doc__ = docstr
 
     # TODO: sadly, fire displays both Optional[type] and Union[type, None] as Optional[Optional]
@@ -61,6 +73,8 @@ class _Run:
         interruptible: bool = False,
         image_credentials: Optional[str] = None,
         cluster_auth: bool = False,
+        artifacts_local: Optional[str] = None,
+        artifacts_remote: Optional[str] = None,
     ) -> None:
         machine_enum = Machine(machine.upper())
         Job.run(
@@ -77,4 +91,6 @@ class _Run:
             interruptible=interruptible,
             image_credentials=image_credentials,
             cluster_auth=cluster_auth,
+            artifacts_local=artifacts_local,
+            artifacts_remote=artifacts_remote,
         )

lightning_sdk/cli/upload.py

@@ -9,8 +9,8 @@ from tqdm import tqdm
 
 from lightning_sdk.api.utils import _get_cloud_url
 from lightning_sdk.cli.exceptions import StudioCliError
-from lightning_sdk.cli.models import _get_teamspace, _parse_model_name
 from lightning_sdk.cli.studios_menu import _StudiosMenu
+from lightning_sdk.models import upload_model
 from lightning_sdk.studio import Studio
 from lightning_sdk.utils.resolve import _get_authed_user, skip_studio_init
 
@@ -20,7 +20,7 @@ class _Uploads(_StudiosMenu):
 
     _studio_upload_status_path = "~/.lightning/studios/uploads"
 
-    def model(self, name: str, path: Optional[str] = None, cloud_account: Optional[str] = None) -> None:
+    def model(self, name: str, path: str = ".", cloud_account: Optional[str] = None) -> None:
         """Upload a Model.
 
         Args:
@@ -29,14 +29,7 @@ class _Uploads(_StudiosMenu):
           path: The path to the file or directory you want to upload. Defaults to the current directory.
           cloud_account: The name of the cloud account to store the Model in.
         """
-        org_name, teamspace_name, model_name = _parse_model_name(name)
-        teamspace = _get_teamspace(name=teamspace_name, organization=org_name)
-        teamspace.upload_model(
-            path=path or ".",
-            name=model_name,
-            progress_bar=True,
-            cluster_id=cloud_account,
-        )
+        upload_model(name, path, cloud_account=cloud_account)
 
     def _resolve_studio(self, studio: Optional[str]) -> Studio:
         user = _get_authed_user()

lightning_sdk/job/base.py

@@ -52,6 +52,8 @@ class _BaseJob(ABC):
         interruptible: bool = False,
         image_credentials: Optional[str] = None,
         cluster_auth: bool = False,
+        artifacts_local: Optional[str] = None,
+        artifacts_remote: Optional[str] = None,
     ) -> "_BaseJob":
         from lightning_sdk.studio import Studio
 
@@ -89,14 +91,30 @@ class _BaseJob(ABC):
             if cluster_auth:
                 raise ValueError("cluster_auth is only supported when using a custom image")
 
+            if artifacts_local is not None or artifacts_remote is not None:
+                raise ValueError(
+                    "Specifying artifacts persistence is supported for docker images only. "
+                    "Other jobs will automatically persist artifacts to the teamspace distributed filesystem."
+                )
+
         else:
             if studio is not None:
                 raise RuntimeError(
                     "image and studio are mutually exclusive as both define the environment to run the job in"
                 )
 
+            # they either need to specified both or none of them
+            if bool(artifacts_local) != bool(artifacts_remote):
+                raise ValueError("Artifact persistence requires both artifacts_local and artifacts_remote to be set")
+
+            if artifacts_remote and len(artifacts_remote.split(":")) != 3:
+                raise ValueError(
+                    "Artifact persistence requires exactly three arguments separated by colon of kind "
+                    f"<CONNECTION_TYPE>:<CONNECTION_NAME>:<PATH_WITHIN_CONNECTION>, got {artifacts_local}"
+                )
+
         inst = cls(name=name, teamspace=teamspace, org=org, user=user, _fetch_job=False)
-        inst._submit(
+        return inst._submit(
             machine=machine,
             cluster=cluster,
             command=command,
@@ -106,8 +124,9 @@ class _BaseJob(ABC):
             interruptible=interruptible,
             image_credentials=image_credentials,
             cluster_auth=cluster_auth,
+            artifacts_local=artifacts_local,
+            artifacts_remote=artifacts_remote,
         )
-        return inst
 
     @abstractmethod
     def _submit(
@@ -121,7 +140,9 @@ class _BaseJob(ABC):
         cluster: Optional[str] = None,
         image_credentials: Optional[str] = None,
         cluster_auth: bool = False,
-    ) -> None:
+        artifacts_local: Optional[str] = None,
+        artifacts_remote: Optional[str] = None,
+    ) -> "_BaseJob":
         """Submits a job and updates the internal _job attribute as well as the _name attribute."""
 
     @abstractmethod

lightning_sdk/job/job.py

@@ -60,6 +60,8 @@ class Job(_BaseJob):
         interruptible: bool = False,
         image_credentials: Optional[str] = None,
         cluster_auth: bool = False,
+        artifacts_local: Optional[str] = None,
+        artifacts_remote: Optional[str] = None,
     ) -> "Job":
         ret_val = super().run(
             name=name,
@@ -75,6 +77,8 @@ class Job(_BaseJob):
             interruptible=interruptible,
             image_credentials=image_credentials,
             cluster_auth=cluster_auth,
+            artifacts_local=artifacts_local,
+            artifacts_remote=artifacts_remote,
         )
         # required for typing with "Job"
         assert isinstance(ret_val, cls)
@@ -91,8 +95,10 @@ class Job(_BaseJob):
         cluster: Optional[str] = None,
         image_credentials: Optional[str] = None,
         cluster_auth: bool = False,
+        artifacts_local: Optional[str] = None,
+        artifacts_remote: Optional[str] = None,
     ) -> None:
-        return self._internal_job._submit(
+        self._job = self._internal_job._submit(
             machine=machine,
             cluster=cluster,
             command=command,
@@ -102,7 +108,10 @@ class Job(_BaseJob):
             interruptible=interruptible,
             image_credentials=image_credentials,
             cluster_auth=cluster_auth,
+            artifacts_local=artifacts_local,
+            artifacts_remote=artifacts_remote,
         )
+        return self
 
     def stop(self) -> None:
         return self._internal_job.stop()

lightning_sdk/job/v1.py

@@ -69,13 +69,18 @@ class _JobV1(_BaseJob):
         cluster: Optional[str] = None,
         image_credentials: Optional[str] = None,
         cluster_auth: bool = False,
-    ) -> None:
+        artifacts_local: Optional[str] = None,
+        artifacts_remote: Optional[str] = None,
+    ) -> "_JobV1":
         if studio is None:
             raise ValueError("Studio is required for submitting jobs")
 
         if image is not None or image_credentials is not None or cluster_auth:
             raise ValueError("Image is not supported for submitting jobs")
 
+        if artifacts_local is not None or artifacts_remote is not None:
+            raise ValueError("Specifying how to persist artifacts is not yet supported with jobs")
+
         if env is not None:
             raise ValueError("Environment variables are not supported for submitting jobs")
 
@@ -95,6 +100,7 @@ class _JobV1(_BaseJob):
         )
         self._name = _submitted.name
         self._job = _submitted
+        return self
 
     def _update_internal_job(self) -> None:
         try:

lightning_sdk/job/v2.py

@@ -36,7 +36,9 @@ class _JobV2(_BaseJob):
         cluster: Optional[str] = None,
         image_credentials: Optional[str] = None,
         cluster_auth: bool = False,
-    ) -> None:
+        artifacts_local: Optional[str] = None,
+        artifacts_remote: Optional[str] = None,
+    ) -> "_JobV2":
         # Command is required if Studio is provided to know what to run
         # Image is mutually exclusive with Studio
         # Command is optional for Image
@@ -66,22 +68,21 @@ class _JobV2(_BaseJob):
             env=env,
             image_credentials=image_credentials,
             cluster_auth=cluster_auth,
+            artifacts_local=artifacts_local,
+            artifacts_remote=artifacts_remote,
         )
         self._job = submitted
         self._name = submitted.name
+        return self
 
     def stop(self) -> None:
-        if self._job is None:
-            self._update_internal_job()
-
-        self._job_api.stop_job(job_id=self._job.id, teamspace_id=self._teamspace.id)
+        self._job_api.stop_job(job_id=self._guaranteed_job.id, teamspace_id=self._teamspace.id)
 
     def delete(self) -> None:
-        if self._job is None:
-            self._update_internal_job()
-
         self._job_api.delete_job(
-            job_id=self._job.id, teamspace_id=self._teamspace.id, cloudspace_id=self._job.spec.cloudspace_id
+            job_id=self._guaranteed_job.id,
+            teamspace_id=self._teamspace.id,
+            cloudspace_id=self._guaranteed_job.spec.cloudspace_id,
         )
 
     @property
@@ -112,10 +113,18 @@ class _JobV2(_BaseJob):
 
     @property
     def artifact_path(self) -> Optional[str]:
+        if self._guaranteed_job.spec.image != "":
+            if self._guaranteed_job.spec.artifacts_destination != "":
+                splits = self._guaranteed_job.spec.artifacts_destination.split(":")
+                return f"/teamspace/{splits[0]}_connections/{splits[1]}/{splits[2]}"
+            return None
+
         return f"/teamspace/jobs/{self._guaranteed_job.name}/artifacts"
 
     @property
     def snapshot_path(self) -> Optional[str]:
+        if self._guaranteed_job.spec.image != "":
+            return None
         return f"/teamspace/jobs/{self._guaranteed_job.name}/snapshot"
 
     @property