modal 1.4.4.dev14__tar.gz → 1.4.4.dev15__tar.gz

{modal-1.4.4.dev14 → modal-1.4.4.dev15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: modal
-Version: 1.4.4.dev14
+Version: 1.4.4.dev15
 Summary: Python client library for Modal
 Author-email: Modal Labs <support@modal.com>
 License-Expression: Apache-2.0

{modal-1.4.4.dev14 → modal-1.4.4.dev15}/modal/_functions.py

@@ -19,6 +19,7 @@ from modal_proto import api_pb2
 from modal_proto.modal_api_grpc import ModalClientModal
 
 from ._function_variants import _FunctionOptions, _make_function_variant
+from ._image import _Image
 from ._load_context import LoadContext
 from ._object import _Object, live_method, live_method_gen
 from ._output.pty import get_pty_info
@@ -65,7 +66,6 @@ from .exception import (
     NotFoundError,
     OutputExpiredError,
 )
-from .image import _Image
 from .mount import _get_client_mount, _Mount
 from .network_file_system import _NetworkFileSystem, network_file_system_mount_protos
 from .output import OutputManager

modal-1.4.4.dev14/modal/image.py → modal-1.4.4.dev15/modal/_image.py

@@ -8,7 +8,7 @@ import sys
 import typing
 import warnings
 from collections.abc import Callable, Collection, Sequence
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from inspect import isfunction
 from pathlib import Path, PurePosixPath
 from typing import (
@@ -32,7 +32,7 @@ from ._load_context import LoadContext
 from ._object import _Object, live_method_gen
 from ._resolver import Resolver
 from ._serialization import get_preferred_payload_format, serialize
-from ._utils.async_utils import TaskContext, deprecate_aio_usage, synchronize_api, synchronizer
+from ._utils.async_utils import TaskContext, deprecate_aio_usage, synchronizer
 from ._utils.blob_utils import MAX_OBJECT_SIZE_BYTES
 from ._utils.docker_utils import (
     extract_copy_command_patterns,
@@ -362,9 +362,8 @@ class _ImageRegistryConfig:
 
 @dataclass
 class DockerfileSpec:
-    # Ideally we would use field() with default_factory=, but doesn't work with synchronicity type-stub gen
-    commands: list[str]
-    context_files: dict[str, str]
+    commands: list[str] = field(default_factory=list)
+    context_files: dict[str, str] = field(default_factory=dict)
 
 
 async def _image_await_build_result(image_id: str, client: _Client) -> api_pb2.ImageJoinStreamingResponse:
@@ -538,7 +537,7 @@ class _Image(_Object, type_prefix="im"):
             raise InvalidError("Cannot run a build function with multiple base images!")
 
         def _deps() -> Sequence[_Object]:
-            deps = tuple(base_images.values()) + tuple(secrets)
+            deps: tuple[_Object, ...] = tuple(base_images.values()) + tuple(secrets)
             if build_function:
                 deps += (build_function,)
             if image_registry_config and image_registry_config.secret:
@@ -670,10 +669,10 @@ class _Image(_Object, type_prefix="im"):
             else:
                 # not built or in the process of building - wait for build
                 logger.debug("Waiting for image %s" % image_id)
-                resp = await _image_await_build_result(image_id, load_context.client)
-                result = resp.result
-                if resp.HasField("metadata"):
-                    metadata = resp.metadata
+                build_resp = await _image_await_build_result(image_id, load_context.client)
+                result = build_resp.result
+                if build_resp.HasField("metadata"):
+                    metadata = build_resp.metadata
 
             if result.status == api_pb2.GenericResult.GENERIC_STATUS_FAILURE:
                 if result.exception:
@@ -701,7 +700,7 @@ class _Image(_Object, type_prefix="im"):
                 raise RemoteError("Unknown status %s!" % result.status)
 
             self._hydrate(image_id, load_context.client, metadata)
-            local_mounts = set()
+            local_mounts: set[_Mount] = set()
             for base in base_images.values():
                 local_mounts |= base._serve_mounts
             if context_mount and context_mount.is_local():
@@ -1865,8 +1864,8 @@ class _Image(_Object, type_prefix="im"):
         """
         if not isinstance(entrypoint_commands, list) or not all(isinstance(x, str) for x in entrypoint_commands):
             raise InvalidError("entrypoint_commands must be a list of strings.")
-        args_str = _flatten_str_args("entrypoint", "entrypoint_commands", entrypoint_commands)
-        args_str = '"' + '", "'.join(args_str) + '"' if args_str else ""
+        args = _flatten_str_args("entrypoint", "entrypoint_commands", entrypoint_commands)
+        args_str = '"' + '", "'.join(args) + '"' if args else ""
         dockerfile_cmd = f"ENTRYPOINT [{args_str}]"
 
         return self.dockerfile_commands(dockerfile_cmd)
@@ -1885,8 +1884,8 @@ class _Image(_Object, type_prefix="im"):
         """
         if not isinstance(shell_commands, list) or not all(isinstance(x, str) for x in shell_commands):
             raise InvalidError("shell_commands must be a list of strings.")
-        args_str = _flatten_str_args("shell", "shell_commands", shell_commands)
-        args_str = '"' + '", "'.join(args_str) + '"' if args_str else ""
+        args = _flatten_str_args("shell", "shell_commands", shell_commands)
+        args_str = '"' + '", "'.join(args) + '"' if args else ""
         dockerfile_cmd = f"SHELL [{args_str}]"
 
         return self.dockerfile_commands(dockerfile_cmd)
@@ -2639,7 +2638,7 @@ class _Image(_Object, type_prefix="im"):
         function = _Function.from_local(
             info,
             app=None,
-            image=self,  # type: ignore[reportArgumentType]  # TODO: probably conflict with type stub?
+            image=self,
             secrets=secrets,
             gpu=gpu,
             volumes=volumes,
@@ -2756,8 +2755,8 @@ class _Image(_Object, type_prefix="im"):
         if not isinstance(cmd, list) or not all(isinstance(x, str) for x in cmd):
             raise InvalidError("Image CMD must be a list of strings.")
 
-        cmd_str = _flatten_str_args("cmd", "cmd", cmd)
-        cmd_str = '"' + '", "'.join(cmd_str) + '"' if cmd_str else ""
+        cmd_args = _flatten_str_args("cmd", "cmd", cmd)
+        cmd_str = '"' + '", "'.join(cmd_args) + '"' if cmd_args else ""
         dockerfile_cmd = f"CMD [{cmd_str}]"
 
         return self.dockerfile_commands(dockerfile_cmd)
@@ -2849,6 +2848,3 @@ class _Image(_Object, type_prefix="im"):
                 "Images cannot currently be hydrated on demand; you can build an Image by running an App that uses it."
             )
         return self
-
-
-Image = synchronize_api(_Image)

{modal-1.4.4.dev14 → modal-1.4.4.dev15}/modal/_runtime/execution_context.py

@@ -7,6 +7,7 @@ from modal._utils.async_utils import synchronize_api
 from modal.exception import InvalidError
 
 from .container_io_manager import _ContainerIOManager
+from .task_lifecycle_manager import _TaskLifecycleManager
 
 
 def is_local() -> bool:
@@ -18,7 +19,7 @@ def is_local() -> bool:
     even though those processes are running on Modal hardware.
 
     """
-    return not _ContainerIOManager._singleton
+    return _TaskLifecycleManager._singleton is None
 
 
 async def _interact() -> None:
@@ -29,7 +30,7 @@ async def _interact() -> None:
     """
     container_io_manager = _ContainerIOManager._singleton
     if not container_io_manager:
-        raise InvalidError("Interactivity only works inside a Modal container.")
+        raise InvalidError("Interactivity only works inside a Modal Function's container.")
     else:
         await container_io_manager.interact()
 

{modal-1.4.4.dev14 → modal-1.4.4.dev15}/modal/_runtime/task_lifecycle_manager.py

@@ -7,6 +7,10 @@ import sys
 import traceback
 from collections.abc import AsyncGenerator
 from pathlib import Path
+from typing import (
+    ClassVar,
+    Optional,
+)
 
 from synchronicity.async_wrap import asynccontextmanager
 
@@ -30,6 +34,8 @@ class UserException(Exception):
 
 
 class _TaskLifecycleManager:
+    _singleton: ClassVar[Optional["_TaskLifecycleManager"]] = None
+
     task_id: str
     function_id: str
     function_def: api_pb2.Function
@@ -37,7 +43,7 @@ class _TaskLifecycleManager:
     _client: _Client
     _cuda_checkpoint_session: gpu_memory_snapshot.CudaCheckpointSession | None
 
-    def __init__(
+    def _init(
         self,
         task_id: str,
         function_id: str,
@@ -52,6 +58,23 @@ class _TaskLifecycleManager:
         self._client = client
         self._cuda_checkpoint_session = None
 
+    def __new__(
+        cls,
+        task_id: str,
+        function_id: str,
+        function_def: api_pb2.Function,
+        checkpoint_id: str | None,
+        client: _Client,
+    ) -> "_TaskLifecycleManager":
+        cls._singleton = super().__new__(cls)
+        cls._singleton._init(task_id, function_id, function_def, checkpoint_id, client)
+        return cls._singleton
+
+    @classmethod
+    def _reset_singleton(cls) -> None:
+        """Only used for tests."""
+        cls._singleton = None
+
     @asynccontextmanager
     async def handle_task_lifecycle_exception(
         self,

{modal-1.4.4.dev14 → modal-1.4.4.dev15}/modal/_runtime/task_lifecycle_manager.pyi

@@ -11,6 +11,7 @@ class UserException(Exception):
     ...
 
 class _TaskLifecycleManager:
+    _singleton: typing.ClassVar[typing.Optional[_TaskLifecycleManager]]
     task_id: str
     function_id: str
     function_def: modal_proto.api_pb2.Function
@@ -18,15 +19,29 @@ class _TaskLifecycleManager:
     _client: modal.client._Client
     _cuda_checkpoint_session: typing.Optional[modal._runtime.gpu_memory_snapshot.CudaCheckpointSession]
 
-    def __init__(
+    def _init(
         self,
         task_id: str,
         function_id: str,
         function_def: modal_proto.api_pb2.Function,
         checkpoint_id: typing.Optional[str],
         client: modal.client._Client,
-    ) -> None:
-        """Initialize self.  See help(type(self)) for accurate signature."""
+    ) -> None: ...
+    @staticmethod
+    def __new__(
+        cls,
+        task_id: str,
+        function_id: str,
+        function_def: modal_proto.api_pb2.Function,
+        checkpoint_id: typing.Optional[str],
+        client: modal.client._Client,
+    ) -> _TaskLifecycleManager:
+        """Create and return a new object.  See help(type) for accurate signature."""
+        ...
+
+    @classmethod
+    def _reset_singleton(cls) -> None:
+        """Only used for tests."""
         ...
 
     def handle_task_lifecycle_exception(self) -> typing.AsyncContextManager[None]:
@@ -45,6 +60,7 @@ class _TaskLifecycleManager:
         ...
 
 class TaskLifecycleManager:
+    _singleton: typing.ClassVar[typing.Optional[TaskLifecycleManager]]
     task_id: str
     function_id: str
     function_def: modal_proto.api_pb2.Function
@@ -52,7 +68,11 @@ class TaskLifecycleManager:
     _client: modal.client.Client
     _cuda_checkpoint_session: typing.Optional[modal._runtime.gpu_memory_snapshot.CudaCheckpointSession]
 
-    def __init__(
+    def __init__(self, /, *args, **kwargs):
+        """Initialize self.  See help(type(self)) for accurate signature."""
+        ...
+
+    def _init(
         self,
         task_id: str,
         function_id: str,
@@ -60,6 +80,10 @@ class TaskLifecycleManager:
         checkpoint_id: typing.Optional[str],
         client: modal.client.Client,
     ) -> None: ...
+    @classmethod
+    def _reset_singleton(cls) -> None:
+        """Only used for tests."""
+        ...
 
     class __handle_task_lifecycle_exception_spec(typing_extensions.Protocol):
         def __call__(self, /) -> synchronicity.combined_types.AsyncAndBlockingContextManager[None]:

{modal-1.4.4.dev14 → modal-1.4.4.dev15}/modal/app.py

@@ -18,6 +18,7 @@ from synchronicity.async_wrap import asynccontextmanager
 from modal_proto import api_pb2
 
 from ._functions import _Function
+from ._image import _Image
 from ._ipython import is_interactive_ipython
 from ._load_context import LoadContext
 from ._object import _get_environment_name, _Object
@@ -47,7 +48,6 @@ from .cls import _Cls, parameter
 from .config import logger
 from .exception import ExecutionError, InvalidError
 from .functions import Function
-from .image import _Image
 from .network_file_system import _NetworkFileSystem
 from .partial_function import PartialFunction
 from .proxy import _Proxy

{modal-1.4.4.dev14 → modal-1.4.4.dev15}/modal/app.pyi

@@ -1,5 +1,6 @@
 import collections.abc
 import modal._functions
+import modal._image
 import modal._load_context
 import modal._partial_function
 import modal._server
@@ -80,7 +81,7 @@ class _LocalAppState:
 
     functions: dict[str, modal._functions._Function]
     classes: dict[str, modal.cls._Cls]
-    image_default: typing.Optional[modal.image._Image]
+    image_default: typing.Optional[modal._image._Image]
     web_endpoints: list[str]
     local_entrypoints: dict[str, _LocalEntrypoint]
     tags: dict[str, str]
@@ -92,7 +93,7 @@ class _LocalAppState:
         self,
         functions: dict[str, modal._functions._Function],
         classes: dict[str, modal.cls._Cls],
-        image_default: typing.Optional[modal.image._Image],
+        image_default: typing.Optional[modal._image._Image],
         web_endpoints: list[str],
         local_entrypoints: dict[str, _LocalEntrypoint],
         tags: dict[str, str],
@@ -163,7 +164,7 @@ class _App:
         name: typing.Optional[str] = None,
         *,
         tags: typing.Optional[dict[str, str]] = None,
-        image: typing.Optional[modal.image._Image] = None,
+        image: typing.Optional[modal._image._Image] = None,
         secrets: collections.abc.Sequence[modal.secret._Secret] = [],
         volumes: dict[typing.Union[str, pathlib.PurePosixPath], modal.volume._Volume] = {},
         include_source: bool = True,
@@ -282,7 +283,7 @@ class _App:
 
     def _validate_blueprint_value(self, key: str, value: typing.Any): ...
     @property
-    def image(self) -> modal.image._Image:
+    def image(self) -> modal._image._Image:
         """mdmd:hidden
         Retrieve the Image that will be used as the default for any Functions registered to the App.
 
@@ -546,7 +547,7 @@ class _App:
         self,
         _warn_parentheses_missing=None,
         *,
-        image: typing.Optional[modal.image._Image] = None,
+        image: typing.Optional[modal._image._Image] = None,
         schedule: typing.Optional[modal.schedule.Schedule] = None,
         env: typing.Optional[dict[str, typing.Optional[str]]] = None,
         secrets: typing.Optional[collections.abc.Collection[modal.secret._Secret]] = None,
@@ -645,7 +646,7 @@ class _App:
         self,
         _warn_parentheses_missing=None,
         *,
-        image: typing.Optional[modal.image._Image] = None,
+        image: typing.Optional[modal._image._Image] = None,
         env: typing.Optional[dict[str, typing.Optional[str]]] = None,
         secrets: typing.Optional[collections.abc.Collection[modal.secret._Secret]] = None,
         gpu: typing.Union[str, list[str], None] = None,
@@ -731,7 +732,7 @@ class _App:
         self,
         _warn_parentheses_missing=None,
         *,
-        image: typing.Optional[modal.image._Image] = None,
+        image: typing.Optional[modal._image._Image] = None,
         env: typing.Optional[dict[str, typing.Optional[str]]] = None,
         secrets: typing.Optional[collections.abc.Collection[modal.secret._Secret]] = None,
         gpu: typing.Union[str, list[str], None] = None,
@@ -1948,4 +1949,4 @@ class App:
         """Only used for tests."""
         ...
 
-_default_image: modal.image._Image
+_default_image: modal._image._Image

{modal-1.4.4.dev14 → modal-1.4.4.dev15}/modal/client.pyi

@@ -35,7 +35,7 @@ class _Client:
         server_url: str,
         client_type: int,
         credentials: typing.Optional[tuple[str, str]],
-        version: str = "1.4.4.dev14",
+        version: str = "1.4.4.dev15",
     ):
         """mdmd:hidden
         The Modal client object is not intended to be instantiated directly by users.
@@ -205,7 +205,7 @@ class Client:
         server_url: str,
         client_type: int,
         credentials: typing.Optional[tuple[str, str]],
-        version: str = "1.4.4.dev14",
+        version: str = "1.4.4.dev15",
     ):
         """mdmd:hidden
         The Modal client object is not intended to be instantiated directly by users.

{modal-1.4.4.dev14 → modal-1.4.4.dev15}/modal/experimental/__init__.py

@@ -6,6 +6,12 @@ from modal_proto import api_pb2
 
 from .._clustered_functions import ClusterInfo, get_cluster_info as _get_cluster_info
 from .._functions import _Function
+from .._image import (
+    DockerfileSpec as DockerfileSpec,
+    ImageBuilderVersion as ImageBuilderVersion,
+    _Image as _Image,
+    _ImageRegistryConfig as _ImageRegistryConfig,
+)
 from .._object import _get_environment_name
 from .._partial_function import _clustered
 from .._runtime.container_io_manager import _ContainerIOManager
@@ -14,12 +20,6 @@ from ..app import _App
 from ..client import _Client
 from ..cls import _Cls
 from ..exception import InvalidError as InvalidError
-from ..image import (
-    DockerfileSpec as DockerfileSpec,
-    ImageBuilderVersion as ImageBuilderVersion,
-    _Image as _Image,
-    _ImageRegistryConfig as _ImageRegistryConfig,
-)
 from ..secret import _Secret as _Secret
 from .flash import (  # noqa: F401
     flash_forward,

modal-1.4.4.dev15/modal/image.py

@@ -0,0 +1,5 @@
+# Copyright Modal Labs 2026
+from ._image import _Image
+from ._utils.async_utils import synchronize_api
+
+Image = synchronize_api(_Image, target_module=__name__)