lightning-sdk 2025.8.6rc2__py3-none-any.whl → 2025.11.5__py3-none-any.whl

lightning_sdk/cli/vm/delete.py

@@ -0,0 +1,25 @@
+from typing import Optional
+
+import click
+
+from lightning_sdk.cli.studio.delete import delete_impl
+
+
+@click.command("delete")
+@click.option(
+    "--name",
+    help=(
+        "The name of the VM to delete. "
+        "If not provided, will try to infer from environment, "
+        "use the default value from the config or prompt for interactive selection."
+    ),
+)
+@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)")
+def delete_vm(name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
+    """Delete a VM.
+
+    Example:
+      lightning vm delete --name my-vm
+
+    """
+    return delete_impl(name=name, teamspace=teamspace, vm=True)

lightning_sdk/cli/vm/list.py

@@ -0,0 +1,30 @@
+from typing import Optional
+
+import click
+
+from lightning_sdk.cli.studio.list import list_impl
+
+
+@click.command("list")
+@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)")
+@click.option(
+    "--all",
+    is_flag=True,
+    flag_value=True,
+    default=False,
+    help="List all VMs, not just the ones belonging to the authed user",
+)
+@click.option(
+    "--sort-by",
+    default=None,
+    type=click.Choice(["name", "teamspace", "status", "machine", "cloud-account"], case_sensitive=False),
+    help="the attribute to sort the VMs by.",
+)
+def list_vms(teamspace: Optional[str] = None, all: bool = False, sort_by: Optional[str] = None) -> None:  # noqa: A002
+    """List VMs in a teamspace.
+
+    Example:
+        lightning vm list --teamspace owner/teamspace
+
+    """
+    return list_impl(teamspace=teamspace, all=all, sort_by=sort_by, vm=True)

lightning_sdk/cli/vm/ssh.py

@@ -0,0 +1,31 @@
+from typing import List, Optional
+
+import click
+
+from lightning_sdk.cli.studio.ssh import ssh_impl
+
+
+@click.command("ssh")
+@click.option(
+    "--name",
+    help=(
+        "The name of the VM to ssh into. "
+        "If not provided, will try to infer from environment, "
+        "use the default value from the config or prompt for interactive selection."
+    ),
+)
+@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)", type=click.STRING)
+@click.option(
+    "--option",
+    "-o",
+    help="Additional options to pass to the SSH command. Can be specified multiple times.",
+    multiple=True,
+    type=click.STRING,
+)
+def ssh_vm(name: Optional[str] = None, teamspace: Optional[str] = None, option: Optional[List[str]] = None) -> None:
+    """SSH into a VM.
+
+    Example:
+        lightning vm ssh --name my-vm
+    """
+    return ssh_impl(name=name, teamspace=teamspace, option=option, vm=False)

lightning_sdk/cli/vm/start.py

@@ -0,0 +1,60 @@
+from typing import Optional
+
+import click
+
+from lightning_sdk.cli.studio.start import start_impl
+from lightning_sdk.machine import CloudProvider, Machine
+
+
+@click.command("start")
+@click.option(
+    "--name",
+    help=(
+        "The name of the VM to start. "
+        "If not provided, will try to infer from environment, "
+        "use the default value from the config or prompt for interactive selection."
+    ),
+)
+@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)")
+@click.option("--create", is_flag=True, help="Create the VM if it doesn't exist")
+@click.option(
+    "--machine",
+    help="The machine type to start the VM on. Defaults to CPU-4",
+    type=click.Choice(m.name for m in Machine.__dict__.values() if isinstance(m, Machine) and m._include_in_cli),
+)
+@click.option("--interruptible", is_flag=True, help="Start the VM on an interruptible instance.")
+@click.option(
+    "--cloud-provider",
+    help=("The cloud provider to start the VM on. Defaults to teamspace default. Only used if --create is specified."),
+    type=click.Choice(m.name for m in list(CloudProvider)),
+)
+@click.option(
+    "--cloud-account",
+    help="The cloud account to start the VM on. Defaults to teamspace default. Only used if --create is specified.",
+    type=click.STRING,
+)
+def start_vm(
+    name: Optional[str] = None,
+    teamspace: Optional[str] = None,
+    create: bool = False,
+    machine: str = "CPU",
+    interruptible: bool = False,
+    cloud_provider: Optional[str] = None,
+    cloud_account: Optional[str] = None,
+) -> None:
+    """Start a VM.
+
+    Example:
+        lightning vm start --name my-vm
+
+    """
+    return start_impl(
+        name=name,
+        teamspace=teamspace,
+        create=create,
+        machine=machine,
+        interruptible=interruptible,
+        cloud_provider=cloud_provider,
+        cloud_account=cloud_account,
+        vm=True,
+    )

lightning_sdk/cli/vm/stop.py

@@ -0,0 +1,25 @@
+from typing import Optional
+
+import click
+
+from lightning_sdk.cli.studio.stop import stop_impl
+
+
+@click.command("stop")
+@click.option(
+    "--name",
+    help=(
+        "The name of the VM to stop. "
+        "If not provided, will try to infer from environment, "
+        "use the default value from the config or prompt for interactive selection."
+    ),
+)
+@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)")
+def stop_vm(name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
+    """Stop a VM.
+
+    Example:
+        lightning vm stop --name my-vm
+
+    """
+    return stop_impl(name=name, teamspace=teamspace, vm=True)

lightning_sdk/cli/vm/switch.py

@@ -0,0 +1,38 @@
+from typing import Optional
+
+import click
+
+from lightning_sdk.cli.studio.switch import switch_impl
+from lightning_sdk.machine import Machine
+
+
+@click.command("switch")
+@click.option(
+    "--name",
+    help=(
+        "The name of the VM to switch to a different machine. "
+        "If not provided, will try to infer from environment, "
+        "use the default value from the config or prompt for interactive selection."
+    ),
+)
+@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)")
+@click.option(
+    "--machine",
+    help="The machine type to switch the studio to.",
+    type=click.Choice(m.name for m in Machine.__dict__.values() if isinstance(m, Machine)),
+)
+@click.option("--interruptible", is_flag=True, help="Switch the studio to an interruptible instance.")
+def switch_vm(
+    name: Optional[str] = None,
+    teamspace: Optional[str] = None,
+    machine: Optional[str] = None,
+    interruptible: bool = False,
+) -> None:
+    """Switch a VM to a different machine type."""
+    return switch_impl(
+        name=name,
+        teamspace=teamspace,
+        machine=machine,
+        interruptible=interruptible,
+        vm=True,
+    )

lightning_sdk/constants.py

@@ -29,3 +29,4 @@ class Store:
 
 
 __GLOBAL_LIGHTNING_UNIQUE_IDS_STORE__ = Store()
+_LIGHTNING_DISABLE_VERSION_CHECK = int(os.getenv("LIGHTNING_DISABLE_VERSION_CHECK", "0"))

lightning_sdk/deployment/__init__.py

@@ -1,4 +1,5 @@
 from lightning_sdk.api.deployment_api import (
+    ApiKeyAuth,
     AutoScaleConfig,
     AutoScalingMetric,
     BasicAuth,
@@ -24,4 +25,5 @@ __all__ = [
     "Secret",
     "TokenAuth",
     "Deployment",
+    "ApiKeyAuth",
 ]

lightning_sdk/deployment/deployment.py

@@ -6,6 +6,7 @@ import requests
 
 from lightning_sdk.api import UserApi
 from lightning_sdk.api.deployment_api import (
+    ApiKeyAuth,
     Auth,
     AutoScaleConfig,
     BasicAuth,
@@ -35,10 +36,11 @@ from lightning_sdk.services.utilities import _get_cluster
 from lightning_sdk.studio import Studio
 from lightning_sdk.teamspace import Teamspace
 from lightning_sdk.user import User
+from lightning_sdk.utils.logging import TrackCallsMeta
 from lightning_sdk.utils.resolve import _resolve_deprecated_cluster, _resolve_org, _resolve_teamspace, _resolve_user
 
 
-class Deployment:
+class Deployment(metaclass=TrackCallsMeta):
     """The Lightning AI Deployment.
 
     Allows to fully control a deployment, including retrieving the status, making new release
@@ -121,7 +123,7 @@ class Deployment:
         spot: Optional[bool] = None,
         replicas: Optional[int] = None,
         health_check: Optional[Union[HttpHealthCheck, ExecHealthCheck]] = None,
-        auth: Optional[Union[BasicAuth, TokenAuth]] = None,
+        auth: Optional[Union[BasicAuth, TokenAuth, ApiKeyAuth]] = None,
         cloud_account: Optional[str] = None,
         custom_domain: Optional[str] = None,
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
@@ -131,6 +133,7 @@ class Deployment:
         from_onboarding: Optional[bool] = None,
         from_litserve: Optional[bool] = None,
         max_runtime: Optional[int] = None,
+        path_mappings: Optional[Dict[str, str]] = None,
     ) -> None:
         """The Lightning AI Deployment.
 
@@ -162,6 +165,15 @@ class Deployment:
                 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
+            path_mappings: Dictionary of path mappings. The keys are the path inside the container whereas the value
+                represents the data-connection name and the path inside that connection.
+                Should be of form
+                    {
+                        "<CONTAINER_PATH_1>": "<CONNECTION_NAME_1>:<PATH_WITHIN_CONNECTION_1>",
+                        "<CONTAINER_PATH_2>": "<CONNECTION_NAME_2>"
+                    }
+                If the path inside the connection is omitted it's assumed to be the root path of that connection.
+                Only applicable when deploying docker containers.
 
         Note:
             Since a teamspace can either be owned by an org or by a user directly,
@@ -238,6 +250,7 @@ class Deployment:
                     cloudspace_id=cloudspace_id,
                     max_runtime=max_runtime,
                     machine_image_version=machine_image_version,
+                    path_mappings=path_mappings,
                 ),
                 strategy=to_strategy(release_strategy),
             ),
@@ -274,6 +287,7 @@ class Deployment:
         quantity: Optional[int] = None,
         include_credentials: Optional[bool] = None,
         max_runtime: Optional[int] = None,
+        path_mappings: Optional[Dict[str, str]] = None,
     ) -> None:
         cloud_account = _resolve_deprecated_cluster(cloud_account, cluster)
 
@@ -301,6 +315,7 @@ class Deployment:
             quantity=quantity,
             include_credentials=include_credentials,
             max_runtime=max_runtime,
+            path_mappings=path_mappings,
         )
 
     def stop(self) -> None:

lightning_sdk/helpers.py

@@ -1,4 +1,3 @@
-import functools
 import importlib
 import os
 import sys
@@ -10,51 +9,71 @@ import tqdm
 import tqdm.std
 from packaging import version as packaging_version
 
-__package_name__ = "lightning-sdk"
+from lightning_sdk.constants import _LIGHTNING_DISABLE_VERSION_CHECK
 
 
-@functools.lru_cache(maxsize=1)
-def _get_newer_version(curr_version: str) -> Optional[str]:
-    """Check PyPI for newer versions of ``lightning-sdk``.
-
-    Returning the newest version if different from the current or ``None`` otherwise.
+class VersionChecker:
+    """Handles version checking and upgrade prompts for lightning-sdk.
 
+    This class ensures that version check warnings are only shown once per session,
+    preventing duplicate warnings in multithreaded scenarios.
     """
-    if packaging_version.parse(curr_version).is_prerelease:
-        return None
-    try:
-        response = requests.get(f"https://pypi.org/pypi/{__package_name__}/json")
-        response_json = response.json()
-        releases = response_json["releases"]
-        if curr_version not in releases:
-            # Always return None if not installed from PyPI (e.g. dev versions)
+
+    def __init__(self, package_name: str = "lightning-sdk") -> None:
+        self.package_name = package_name
+        self._warning_shown = False
+        self._cached_version: Optional[str] = None
+
+    def _get_newer_version(self, curr_version: str) -> Optional[str]:
+        """Check PyPI for newer versions of ``lightning-sdk``.
+
+        Returning the newest version if different from the current or ``None`` otherwise.
+        """
+        if self._cached_version is not None:
+            return self._cached_version
+
+        if _LIGHTNING_DISABLE_VERSION_CHECK == 1 or packaging_version.parse(curr_version).is_prerelease:
+            self._cached_version = None
             return None
-        latest_version = response_json["info"]["version"]
-        parsed_version = packaging_version.parse(latest_version)
-        is_invalid = response_json["info"]["yanked"] or parsed_version.is_devrelease or parsed_version.is_prerelease
-        return None if curr_version == latest_version or is_invalid else latest_version
-    except requests.exceptions.RequestException:
-        return None
 
+        try:
+            response = requests.get(f"https://pypi.org/pypi/{self.package_name}/json")
+            response_json = response.json()
+            releases = response_json["releases"]
+            if curr_version not in releases:
+                # Always return None if not installed from PyPI (e.g. dev versions)
+                self._cached_version = None
+                return None
+            latest_version = response_json["info"]["version"]
+            parsed_version = packaging_version.parse(latest_version)
+            is_invalid = response_json["info"]["yanked"] or parsed_version.is_devrelease or parsed_version.is_prerelease
+            self._cached_version = None if curr_version == latest_version or is_invalid else latest_version
+            return self._cached_version
+        except requests.exceptions.RequestException:
+            self._cached_version = None
+            return None
 
-def _check_version_and_prompt_upgrade(curr_version: str) -> None:
-    """Checks that the current version of ``lightning-sdk`` is the latest on PyPI.
+    def check_and_prompt_upgrade(self, curr_version: str) -> None:
+        """Checks that the current version of ``lightning-sdk`` is the latest on PyPI.
 
-    If not, warn the user to upgrade ``lightning-sdk``.
+        If not, warn the user to upgrade ``lightning-sdk``.
+        Tracks if the warning has already been shown in this session to avoid duplicate warnings.
+        """
+        if self._warning_shown:
+            return
 
-    """
-    new_version = _get_newer_version(curr_version)
-    if new_version:
-        warnings.warn(
-            f"A newer version of {__package_name__} is available ({new_version}). "
-            f"Please consider upgrading with `pip install -U {__package_name__}`. "
-            "Not all platform functionality can be guaranteed to work with the current version.",
-            UserWarning,
-        )
-    return
-
-
-def _set_tqdm_envvars_noninteractive() -> None:
+        new_version = self._get_newer_version(curr_version)
+        if new_version:
+            warnings.warn(
+                f"A newer version of {self.package_name} is available ({new_version}). "
+                f"Please consider upgrading with `pip install -U {self.package_name}`. "
+                "Not all platform functionality can be guaranteed to work with the current version.",
+                UserWarning,
+            )
+            self._warning_shown = True
+
+
+def set_tqdm_envvars_noninteractive() -> None:
     # note: stderr is the default stream tqdm writes progressbars to
     # so we check that one.
     if os.isatty(sys.stderr.fileno()):

lightning_sdk/job/base.py

@@ -4,7 +4,8 @@ from typing import TYPE_CHECKING, Any, Dict, Optional, TypedDict, Union
 
 from lightning_sdk.api.cloud_account_api import CloudAccountApi
 from lightning_sdk.api.utils import _get_cloud_url
-from lightning_sdk.utils.resolve import _resolve_deprecated_cluster, _resolve_teamspace, in_studio
+from lightning_sdk.utils.logging import TrackCallsABCMeta
+from lightning_sdk.utils.resolve import _resolve_deprecated_cluster, _resolve_teamspace, in_studio, skip_studio_setup
 
 if TYPE_CHECKING:
     from lightning_sdk.machine import CloudProvider, Machine
@@ -29,7 +30,7 @@ class JobDict(MachineDict):
     total_cost: float
 
 
-class _BaseJob(ABC):
+class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
     """Base interface to all job types."""
 
     def __init__(
@@ -90,6 +91,7 @@ class _BaseJob(ABC):
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
+        reuse_snapshot: bool = True,
     ) -> "_BaseJob":
         """Run async workloads using a docker image or a compute environment from your studio.
 
@@ -134,6 +136,8 @@ class _BaseJob(ABC):
                 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
@@ -145,9 +149,15 @@ class _BaseJob(ABC):
 
         if image is None:
             if not isinstance(studio, Studio):
-                studio = Studio(
-                    name=studio, teamspace=teamspace, org=org, user=user, cloud_account=cloud_account, create_ok=False
-                )
+                with skip_studio_setup():
+                    studio = Studio(
+                        name=studio,
+                        teamspace=teamspace,
+                        org=org,
+                        user=user,
+                        cloud_account=cloud_account,
+                        create_ok=False,
+                    )
 
             # studio is a Studio instance at this point
             if teamspace is None:
@@ -192,7 +202,8 @@ class _BaseJob(ABC):
                 )
             if cloud_account is None and in_studio():
                 try:
-                    resolve_studio = Studio(teamspace=teamspace, user=user, org=org)
+                    with skip_studio_setup():
+                        resolve_studio = Studio(teamspace=teamspace, user=user, org=org)
                     cloud_account = resolve_studio.cloud_account
                 except (ValueError, ApiException):
                     warnings.warn("Could not infer cloud account from studio. Using teamspace default.")
@@ -224,6 +235,7 @@ class _BaseJob(ABC):
             entrypoint=entrypoint,
             path_mappings=path_mappings,
             max_runtime=max_runtime,
+            reuse_snapshot=reuse_snapshot,
         )
 
     @abstractmethod
@@ -244,6 +256,7 @@ class _BaseJob(ABC):
         entrypoint: str = "sh -c",
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
+        reuse_snapshot: bool = True,
     ) -> "_BaseJob":
         """Submit a new job to the Lightning AI platform.
 
@@ -283,6 +296,8 @@ class _BaseJob(ABC):
                 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.
         """
 
     @abstractmethod

lightning_sdk/job/job.py

@@ -97,6 +97,7 @@ class Job(_BaseJob):
         artifacts_local: Optional[str] = None,  # deprecated in terms of path_mappings
         artifacts_remote: Optional[str] = None,  # deprecated in terms of path_mappings
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
+        reuse_snapshot: bool = True,
     ) -> "Job":
         """Run async workloads using a docker image or a compute environment from your studio.
 
@@ -140,6 +141,8 @@ class Job(_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.
         """
         ret_val = super().run(
             name=name,
@@ -162,6 +165,7 @@ class Job(_BaseJob):
             path_mappings=path_mappings,
             max_runtime=max_runtime,
             cluster=cluster,
+            reuse_snapshot=reuse_snapshot,
         )
         # required for typing with "Job"
         assert isinstance(ret_val, cls)
@@ -186,6 +190,7 @@ class Job(_BaseJob):
         artifacts_local: Optional[str] = None,  # deprecated in terms of path_mappings
         artifacts_remote: Optional[str] = None,  # deprecated in terms of path_mappings
         max_runtime: Optional[int] = None,
+        reuse_snapshot: bool = True,
     ) -> "Job":
         """Submit a new job to the Lightning AI platform.
 
@@ -224,6 +229,8 @@ class Job(_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.
         """
         self._job = self._internal_job._submit(
             machine=machine,
@@ -241,6 +248,7 @@ class Job(_BaseJob):
             artifacts_local=artifacts_local,
             artifacts_remote=artifacts_remote,
             max_runtime=max_runtime,
+            reuse_snapshot=reuse_snapshot,
         )
         return self
 
@@ -268,6 +276,11 @@ class Job(_BaseJob):
         """The machine type the job is running on."""
         return self._internal_job.machine
 
+    @property
+    def public_ip(self) -> Optional[str]:
+        """The public IP address of the machine the job is running on."""
+        return self._internal_job.public_ip
+
     @property
     def artifact_path(self) -> Optional[str]:
         """Path to the artifacts created by the job within the distributed teamspace filesystem."""

lightning_sdk/job/v1.py

@@ -54,6 +54,7 @@ class _JobV1(_BaseJob):
         cloud_provider: Optional[str] = None,
         interruptible: bool = False,
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
+        reuse_snapshot: bool = True,
     ) -> "_BaseJob":
         """Start a new async workload from your studio.
 
@@ -89,6 +90,7 @@ class _JobV1(_BaseJob):
             cluster=cluster,
             path_mappings=None,
             max_runtime=None,
+            reuse_snapshot=reuse_snapshot,
         )
 
     def _submit(
@@ -108,6 +110,7 @@ class _JobV1(_BaseJob):
         entrypoint: str = "sh -c",
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
+        reuse_snapshot: bool = True,
     ) -> "_JobV1":
         """Submit a job to run on a machine.
 
@@ -181,6 +184,14 @@ class _JobV1(_BaseJob):
         """Get the machine the job is running on."""
         return self.work.machine
 
+    @property
+    def public_ip(self) -> Optional[str]:
+        """Get the public IP of the machine the job is running on."""
+        try:
+            return self._job.status.ip_address
+        except AttributeError:
+            return None
+
     @property
     def name(self) -> str:
         """The name of the job."""

lightning_sdk/job/v2.py

@@ -53,6 +53,7 @@ class _JobV2(_BaseJob):
         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,
     ) -> "_JobV2":
         """Submit a new job to the Lightning AI platform.
 
@@ -87,6 +88,8 @@ class _JobV2(_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.
         """
         # Command is required if Studio is provided to know what to run
         # Image is mutually exclusive with Studio
@@ -129,6 +132,7 @@ class _JobV2(_BaseJob):
             entrypoint=entrypoint,
             path_mappings=path_mappings,
             max_runtime=max_runtime,
+            reuse_snapshot=reuse_snapshot,
         )
         self._job = submitted
         self._name = submitted.name
@@ -173,6 +177,14 @@ class _JobV2(_BaseJob):
             _get_org_id(self.teamspace),
         )
 
+    @property
+    def public_ip(self) -> Optional[str]:
+        """Get the public IP of the machine the job is running on."""
+        try:
+            return self._job.public_ip_address
+        except AttributeError:
+            return None
+
     @property
     def artifact_path(self) -> Optional[str]:
         """The path to the artifacts of the job within the distributed teamspace filesystem."""