modal 0.72.38__py3-none-any.whl → 0.72.40__py3-none-any.whl

modal/mount.pyi

@@ -36,7 +36,7 @@ class _MountFile(_MountEntry):
 class _MountDir(_MountEntry):
     local_dir: pathlib.Path
     remote_path: pathlib.PurePosixPath
-    ignore: typing.Callable[[pathlib.Path], bool]
+    ignore: collections.abc.Callable[[pathlib.Path], bool]
     recursive: bool
 
     def description(self): ...
@@ -47,7 +47,7 @@ class _MountDir(_MountEntry):
         self,
         local_dir: pathlib.Path,
         remote_path: pathlib.PurePosixPath,
-        ignore: typing.Callable[[pathlib.Path], bool],
+        ignore: collections.abc.Callable[[pathlib.Path], bool],
         recursive: bool,
     ) -> None: ...
     def __repr__(self): ...
@@ -59,7 +59,7 @@ def module_mount_ignore_condition(module_base: pathlib.Path): ...
 class _MountedPythonModule(_MountEntry):
     module_name: str
     remote_dir: typing.Union[pathlib.PurePosixPath, str]
-    ignore: typing.Optional[typing.Callable[[pathlib.Path], bool]]
+    ignore: typing.Optional[collections.abc.Callable[[pathlib.Path], bool]]
 
     def description(self) -> str: ...
     def _proxy_entries(self) -> list[_MountEntry]: ...
@@ -70,7 +70,7 @@ class _MountedPythonModule(_MountEntry):
         self,
         module_name: str,
         remote_dir: typing.Union[pathlib.PurePosixPath, str] = "/root",
-        ignore: typing.Optional[typing.Callable[[pathlib.Path], bool]] = None,
+        ignore: typing.Optional[collections.abc.Callable[[pathlib.Path], bool]] = None,
     ) -> None: ...
     def __repr__(self): ...
     def __eq__(self, other): ...
@@ -96,14 +96,14 @@ class _Mount(modal._object._Object):
     def _add_local_dir(
         local_path: pathlib.Path,
         remote_path: pathlib.PurePosixPath,
-        ignore: typing.Callable[[pathlib.Path], bool] = modal.file_pattern_matcher._NOTHING,
+        ignore: collections.abc.Callable[[pathlib.Path], bool] = modal.file_pattern_matcher._NOTHING,
     ): ...
     def add_local_dir(
         self,
         local_path: typing.Union[str, pathlib.Path],
         *,
         remote_path: typing.Union[str, pathlib.PurePosixPath, None] = None,
-        condition: typing.Optional[typing.Callable[[str], bool]] = None,
+        condition: typing.Optional[collections.abc.Callable[[str], bool]] = None,
         recursive: bool = True,
     ) -> _Mount: ...
     @staticmethod
@@ -111,7 +111,7 @@ class _Mount(modal._object._Object):
         local_path: typing.Union[str, pathlib.Path],
         *,
         remote_path: typing.Union[str, pathlib.PurePosixPath, None] = None,
-        condition: typing.Optional[typing.Callable[[str], bool]] = None,
+        condition: typing.Optional[collections.abc.Callable[[str], bool]] = None,
         recursive: bool = True,
     ) -> _Mount: ...
     @staticmethod
@@ -119,7 +119,7 @@ class _Mount(modal._object._Object):
         local_path: typing.Union[str, pathlib.Path],
         *,
         remote_path: typing.Union[str, pathlib.PurePosixPath, None] = None,
-        condition: typing.Optional[typing.Callable[[str], bool]] = None,
+        condition: typing.Optional[collections.abc.Callable[[str], bool]] = None,
         recursive: bool = True,
     ) -> _Mount: ...
     def add_local_file(
@@ -148,15 +148,15 @@ class _Mount(modal._object._Object):
     def from_local_python_packages(
         *module_names: str,
         remote_dir: typing.Union[str, pathlib.PurePosixPath] = "/root",
-        condition: typing.Optional[typing.Callable[[str], bool]] = None,
-        ignore: typing.Union[typing.Sequence[str], typing.Callable[[pathlib.Path], bool], None] = None,
+        condition: typing.Optional[collections.abc.Callable[[str], bool]] = None,
+        ignore: typing.Union[typing.Sequence[str], collections.abc.Callable[[pathlib.Path], bool], None] = None,
     ) -> _Mount: ...
     @staticmethod
     def _from_local_python_packages(
         *module_names: str,
         remote_dir: typing.Union[str, pathlib.PurePosixPath] = "/root",
-        condition: typing.Optional[typing.Callable[[str], bool]] = None,
-        ignore: typing.Union[typing.Sequence[str], typing.Callable[[pathlib.Path], bool], None] = None,
+        condition: typing.Optional[collections.abc.Callable[[str], bool]] = None,
+        ignore: typing.Union[typing.Sequence[str], collections.abc.Callable[[pathlib.Path], bool], None] = None,
     ) -> _Mount: ...
     @staticmethod
     def from_name(name: str, namespace=1, environment_name: typing.Optional[str] = None) -> _Mount: ...
@@ -177,6 +177,8 @@ class _Mount(modal._object._Object):
     ) -> None: ...
     def _get_metadata(self) -> modal_proto.api_pb2.MountHandleMetadata: ...
 
+SUPERSELF = typing.TypeVar("SUPERSELF", covariant=True)
+
 class Mount(modal.object.Object):
     _entries: typing.Optional[list[_MountEntry]]
     _deployment_name: typing.Optional[str]
@@ -197,14 +199,14 @@ class Mount(modal.object.Object):
     def _add_local_dir(
         local_path: pathlib.Path,
         remote_path: pathlib.PurePosixPath,
-        ignore: typing.Callable[[pathlib.Path], bool] = modal.file_pattern_matcher._NOTHING,
+        ignore: collections.abc.Callable[[pathlib.Path], bool] = modal.file_pattern_matcher._NOTHING,
     ): ...
     def add_local_dir(
         self,
         local_path: typing.Union[str, pathlib.Path],
         *,
         remote_path: typing.Union[str, pathlib.PurePosixPath, None] = None,
-        condition: typing.Optional[typing.Callable[[str], bool]] = None,
+        condition: typing.Optional[collections.abc.Callable[[str], bool]] = None,
         recursive: bool = True,
     ) -> Mount: ...
     @staticmethod
@@ -212,7 +214,7 @@ class Mount(modal.object.Object):
         local_path: typing.Union[str, pathlib.Path],
         *,
         remote_path: typing.Union[str, pathlib.PurePosixPath, None] = None,
-        condition: typing.Optional[typing.Callable[[str], bool]] = None,
+        condition: typing.Optional[collections.abc.Callable[[str], bool]] = None,
         recursive: bool = True,
     ) -> Mount: ...
     @staticmethod
@@ -220,7 +222,7 @@ class Mount(modal.object.Object):
         local_path: typing.Union[str, pathlib.Path],
         *,
         remote_path: typing.Union[str, pathlib.PurePosixPath, None] = None,
-        condition: typing.Optional[typing.Callable[[str], bool]] = None,
+        condition: typing.Optional[collections.abc.Callable[[str], bool]] = None,
         recursive: bool = True,
     ) -> Mount: ...
     def add_local_file(
@@ -249,25 +251,25 @@ class Mount(modal.object.Object):
 
     _get_files: ___get_files_spec
 
-    class ___load_mount_spec(typing_extensions.Protocol):
+    class ___load_mount_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self, resolver: modal._resolver.Resolver, existing_object_id: typing.Optional[str]): ...
         async def aio(self, resolver: modal._resolver.Resolver, existing_object_id: typing.Optional[str]): ...
 
-    _load_mount: ___load_mount_spec
+    _load_mount: ___load_mount_spec[typing_extensions.Self]
 
     @staticmethod
     def from_local_python_packages(
         *module_names: str,
         remote_dir: typing.Union[str, pathlib.PurePosixPath] = "/root",
-        condition: typing.Optional[typing.Callable[[str], bool]] = None,
-        ignore: typing.Union[typing.Sequence[str], typing.Callable[[pathlib.Path], bool], None] = None,
+        condition: typing.Optional[collections.abc.Callable[[str], bool]] = None,
+        ignore: typing.Union[typing.Sequence[str], collections.abc.Callable[[pathlib.Path], bool], None] = None,
     ) -> Mount: ...
     @staticmethod
     def _from_local_python_packages(
         *module_names: str,
         remote_dir: typing.Union[str, pathlib.PurePosixPath] = "/root",
-        condition: typing.Optional[typing.Callable[[str], bool]] = None,
-        ignore: typing.Union[typing.Sequence[str], typing.Callable[[pathlib.Path], bool], None] = None,
+        condition: typing.Optional[collections.abc.Callable[[str], bool]] = None,
+        ignore: typing.Union[typing.Sequence[str], collections.abc.Callable[[pathlib.Path], bool], None] = None,
     ) -> Mount: ...
     @staticmethod
     def from_name(name: str, namespace=1, environment_name: typing.Optional[str] = None) -> Mount: ...
@@ -280,7 +282,7 @@ class Mount(modal.object.Object):
         environment_name: typing.Optional[str] = None,
     ) -> Mount: ...
 
-    class ___deploy_spec(typing_extensions.Protocol):
+    class ___deploy_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(
             self,
             deployment_name: typing.Optional[str] = None,
@@ -296,7 +298,7 @@ class Mount(modal.object.Object):
             client: typing.Optional[modal.client.Client] = None,
         ) -> None: ...
 
-    _deploy: ___deploy_spec
+    _deploy: ___deploy_spec[typing_extensions.Self]
 
     def _get_metadata(self) -> modal_proto.api_pb2.MountHandleMetadata: ...
 

modal/network_file_system.pyi

@@ -48,7 +48,7 @@ class _NetworkFileSystem(modal._object._Object):
         self,
         remote_path: str,
         fp: typing.BinaryIO,
-        progress_cb: typing.Optional[typing.Callable[..., typing.Any]] = None,
+        progress_cb: typing.Optional[collections.abc.Callable[..., typing.Any]] = None,
     ) -> int: ...
     def read_file(self, path: str) -> collections.abc.AsyncIterator[bytes]: ...
     def iterdir(self, path: str) -> collections.abc.AsyncIterator[modal.volume.FileEntry]: ...
@@ -56,17 +56,19 @@ class _NetworkFileSystem(modal._object._Object):
         self,
         local_path: typing.Union[pathlib.Path, str],
         remote_path: typing.Union[str, pathlib.PurePosixPath, None] = None,
-        progress_cb: typing.Optional[typing.Callable[..., typing.Any]] = None,
+        progress_cb: typing.Optional[collections.abc.Callable[..., typing.Any]] = None,
     ): ...
     async def add_local_dir(
         self,
         local_path: typing.Union[pathlib.Path, str],
         remote_path: typing.Union[str, pathlib.PurePosixPath, None] = None,
-        progress_cb: typing.Optional[typing.Callable[..., typing.Any]] = None,
+        progress_cb: typing.Optional[collections.abc.Callable[..., typing.Any]] = None,
     ): ...
     async def listdir(self, path: str) -> list[modal.volume.FileEntry]: ...
     async def remove_file(self, path: str, recursive=False): ...
 
+SUPERSELF = typing.TypeVar("SUPERSELF", covariant=True)
+
 class NetworkFileSystem(modal.object.Object):
     def __init__(self, *args, **kwargs): ...
     @staticmethod
@@ -135,74 +137,74 @@ class NetworkFileSystem(modal.object.Object):
 
     delete: __delete_spec
 
-    class __write_file_spec(typing_extensions.Protocol):
+    class __write_file_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(
             self,
             remote_path: str,
             fp: typing.BinaryIO,
-            progress_cb: typing.Optional[typing.Callable[..., typing.Any]] = None,
+            progress_cb: typing.Optional[collections.abc.Callable[..., typing.Any]] = None,
         ) -> int: ...
         async def aio(
             self,
             remote_path: str,
             fp: typing.BinaryIO,
-            progress_cb: typing.Optional[typing.Callable[..., typing.Any]] = None,
+            progress_cb: typing.Optional[collections.abc.Callable[..., typing.Any]] = None,
         ) -> int: ...
 
-    write_file: __write_file_spec
+    write_file: __write_file_spec[typing_extensions.Self]
 
-    class __read_file_spec(typing_extensions.Protocol):
+    class __read_file_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self, path: str) -> typing.Iterator[bytes]: ...
         def aio(self, path: str) -> collections.abc.AsyncIterator[bytes]: ...
 
-    read_file: __read_file_spec
+    read_file: __read_file_spec[typing_extensions.Self]
 
-    class __iterdir_spec(typing_extensions.Protocol):
+    class __iterdir_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self, path: str) -> typing.Iterator[modal.volume.FileEntry]: ...
         def aio(self, path: str) -> collections.abc.AsyncIterator[modal.volume.FileEntry]: ...
 
-    iterdir: __iterdir_spec
+    iterdir: __iterdir_spec[typing_extensions.Self]
 
-    class __add_local_file_spec(typing_extensions.Protocol):
+    class __add_local_file_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(
             self,
             local_path: typing.Union[pathlib.Path, str],
             remote_path: typing.Union[str, pathlib.PurePosixPath, None] = None,
-            progress_cb: typing.Optional[typing.Callable[..., typing.Any]] = None,
+            progress_cb: typing.Optional[collections.abc.Callable[..., typing.Any]] = None,
         ): ...
         async def aio(
             self,
             local_path: typing.Union[pathlib.Path, str],
             remote_path: typing.Union[str, pathlib.PurePosixPath, None] = None,
-            progress_cb: typing.Optional[typing.Callable[..., typing.Any]] = None,
+            progress_cb: typing.Optional[collections.abc.Callable[..., typing.Any]] = None,
         ): ...
 
-    add_local_file: __add_local_file_spec
+    add_local_file: __add_local_file_spec[typing_extensions.Self]
 
-    class __add_local_dir_spec(typing_extensions.Protocol):
+    class __add_local_dir_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(
             self,
             local_path: typing.Union[pathlib.Path, str],
             remote_path: typing.Union[str, pathlib.PurePosixPath, None] = None,
-            progress_cb: typing.Optional[typing.Callable[..., typing.Any]] = None,
+            progress_cb: typing.Optional[collections.abc.Callable[..., typing.Any]] = None,
         ): ...
         async def aio(
             self,
             local_path: typing.Union[pathlib.Path, str],
             remote_path: typing.Union[str, pathlib.PurePosixPath, None] = None,
-            progress_cb: typing.Optional[typing.Callable[..., typing.Any]] = None,
+            progress_cb: typing.Optional[collections.abc.Callable[..., typing.Any]] = None,
         ): ...
 
-    add_local_dir: __add_local_dir_spec
+    add_local_dir: __add_local_dir_spec[typing_extensions.Self]
 
-    class __listdir_spec(typing_extensions.Protocol):
+    class __listdir_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self, path: str) -> list[modal.volume.FileEntry]: ...
         async def aio(self, path: str) -> list[modal.volume.FileEntry]: ...
 
-    listdir: __listdir_spec
+    listdir: __listdir_spec[typing_extensions.Self]
 
-    class __remove_file_spec(typing_extensions.Protocol):
+    class __remove_file_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self, path: str, recursive=False): ...
         async def aio(self, path: str, recursive=False): ...
 
-    remove_file: __remove_file_spec
+    remove_file: __remove_file_spec[typing_extensions.Self]

modal/object.pyi

@@ -5,24 +5,28 @@ import modal.client
 import typing
 import typing_extensions
 
+SUPERSELF = typing.TypeVar("SUPERSELF", covariant=True)
+
 class Object:
     _type_prefix: typing.ClassVar[typing.Optional[str]]
     _prefix_to_type: typing.ClassVar[dict[str, type]]
     _load: typing.Optional[
-        typing.Callable[
+        collections.abc.Callable[
             [typing_extensions.Self, modal._resolver.Resolver, typing.Optional[str]], collections.abc.Awaitable[None]
         ]
     ]
     _preload: typing.Optional[
-        typing.Callable[
+        collections.abc.Callable[
             [typing_extensions.Self, modal._resolver.Resolver, typing.Optional[str]], collections.abc.Awaitable[None]
         ]
     ]
     _rep: str
     _is_another_app: bool
     _hydrate_lazily: bool
-    _deps: typing.Optional[typing.Callable[..., collections.abc.Sequence[Object]]]
-    _deduplication_key: typing.Optional[typing.Callable[[], collections.abc.Awaitable[collections.abc.Hashable]]]
+    _deps: typing.Optional[collections.abc.Callable[..., collections.abc.Sequence[Object]]]
+    _deduplication_key: typing.Optional[
+        collections.abc.Callable[[], collections.abc.Awaitable[collections.abc.Hashable]]
+    ]
     _object_id: typing.Optional[str]
     _client: typing.Optional[modal.client.Client]
     _is_hydrated: bool
@@ -32,45 +36,43 @@ class Object:
     @classmethod
     def __init_subclass__(cls, type_prefix: typing.Optional[str] = None): ...
 
-    class ___init_spec(typing_extensions.Protocol):
+    class ___init_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(
             self,
             rep: str,
             load: typing.Optional[
-                typing.Callable[[typing_extensions.Self, modal._resolver.Resolver, typing.Optional[str]], None]
+                collections.abc.Callable[[SUPERSELF, modal._resolver.Resolver, typing.Optional[str]], None]
             ] = None,
             is_another_app: bool = False,
             preload: typing.Optional[
-                typing.Callable[[typing_extensions.Self, modal._resolver.Resolver, typing.Optional[str]], None]
+                collections.abc.Callable[[SUPERSELF, modal._resolver.Resolver, typing.Optional[str]], None]
             ] = None,
             hydrate_lazily: bool = False,
-            deps: typing.Optional[typing.Callable[..., collections.abc.Sequence[Object]]] = None,
-            deduplication_key: typing.Optional[typing.Callable[[], collections.abc.Hashable]] = None,
+            deps: typing.Optional[collections.abc.Callable[..., collections.abc.Sequence[Object]]] = None,
+            deduplication_key: typing.Optional[collections.abc.Callable[[], collections.abc.Hashable]] = None,
         ): ...
         def aio(
             self,
             rep: str,
             load: typing.Optional[
-                typing.Callable[
-                    [typing_extensions.Self, modal._resolver.Resolver, typing.Optional[str]],
-                    collections.abc.Awaitable[None],
+                collections.abc.Callable[
+                    [SUPERSELF, modal._resolver.Resolver, typing.Optional[str]], collections.abc.Awaitable[None]
                 ]
             ] = None,
             is_another_app: bool = False,
             preload: typing.Optional[
-                typing.Callable[
-                    [typing_extensions.Self, modal._resolver.Resolver, typing.Optional[str]],
-                    collections.abc.Awaitable[None],
+                collections.abc.Callable[
+                    [SUPERSELF, modal._resolver.Resolver, typing.Optional[str]], collections.abc.Awaitable[None]
                 ]
             ] = None,
             hydrate_lazily: bool = False,
-            deps: typing.Optional[typing.Callable[..., collections.abc.Sequence[Object]]] = None,
+            deps: typing.Optional[collections.abc.Callable[..., collections.abc.Sequence[Object]]] = None,
             deduplication_key: typing.Optional[
-                typing.Callable[[], collections.abc.Awaitable[collections.abc.Hashable]]
+                collections.abc.Callable[[], collections.abc.Awaitable[collections.abc.Hashable]]
             ] = None,
         ): ...
 
-    _init: ___init_spec
+    _init: ___init_spec[typing_extensions.Self]
 
     def _unhydrate(self): ...
     def _initialize_from_empty(self): ...
@@ -85,15 +87,15 @@ class Object:
     @classmethod
     def _from_loader(
         cls,
-        load: typing.Callable[[typing_extensions.Self, modal._resolver.Resolver, typing.Optional[str]], None],
+        load: collections.abc.Callable[[typing_extensions.Self, modal._resolver.Resolver, typing.Optional[str]], None],
         rep: str,
         is_another_app: bool = False,
         preload: typing.Optional[
-            typing.Callable[[typing_extensions.Self, modal._resolver.Resolver, typing.Optional[str]], None]
+            collections.abc.Callable[[typing_extensions.Self, modal._resolver.Resolver, typing.Optional[str]], None]
         ] = None,
         hydrate_lazily: bool = False,
-        deps: typing.Optional[typing.Callable[..., collections.abc.Sequence[Object]]] = None,
-        deduplication_key: typing.Optional[typing.Callable[[], collections.abc.Hashable]] = None,
+        deps: typing.Optional[collections.abc.Callable[..., collections.abc.Sequence[Object]]] = None,
+        deduplication_key: typing.Optional[collections.abc.Callable[[], collections.abc.Hashable]] = None,
     ): ...
     @staticmethod
     def _get_type_from_id(object_id: str) -> type[Object]: ...
@@ -118,10 +120,16 @@ class Object:
     @property
     def is_hydrated(self) -> bool: ...
     @property
-    def deps(self) -> typing.Callable[..., collections.abc.Sequence[Object]]: ...
+    def deps(self) -> collections.abc.Callable[..., collections.abc.Sequence[Object]]: ...
 
-    class __resolve_spec(typing_extensions.Protocol):
+    class __resolve_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self, client: typing.Optional[modal.client.Client] = None): ...
         async def aio(self, client: typing.Optional[modal.client.Client] = None): ...
 
-    resolve: __resolve_spec
+    resolve: __resolve_spec[typing_extensions.Self]
+
+    class __hydrate_spec(typing_extensions.Protocol[SUPERSELF]):
+        def __call__(self, client: typing.Optional[modal.client.Client] = None) -> SUPERSELF: ...
+        async def aio(self, client: typing.Optional[modal.client.Client] = None) -> SUPERSELF: ...
+
+    hydrate: __hydrate_spec[typing_extensions.Self]

modal/parallel_map.pyi

@@ -1,3 +1,4 @@
+import collections.abc
 import modal._utils.async_utils
 import modal.client
 import modal.functions
@@ -9,26 +10,28 @@ class _SynchronizedQueue:
     async def put(self, item): ...
     async def get(self): ...
 
+SUPERSELF = typing.TypeVar("SUPERSELF", covariant=True)
+
 class SynchronizedQueue:
     def __init__(self, /, *args, **kwargs): ...
 
-    class __init_spec(typing_extensions.Protocol):
+    class __init_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self): ...
         async def aio(self): ...
 
-    init: __init_spec
+    init: __init_spec[typing_extensions.Self]
 
-    class __put_spec(typing_extensions.Protocol):
+    class __put_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self, item): ...
         async def aio(self, item): ...
 
-    put: __put_spec
+    put: __put_spec[typing_extensions.Self]
 
-    class __get_spec(typing_extensions.Protocol):
+    class __get_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self): ...
         async def aio(self): ...
 
-    get: __get_spec
+    get: __get_spec[typing_extensions.Self]
 
 class _OutputValue:
     value: typing.Any
@@ -43,7 +46,7 @@ def _map_invocation(
     client: modal.client._Client,
     order_outputs: bool,
     return_exceptions: bool,
-    count_update_callback: typing.Optional[typing.Callable[[int, int], None]],
+    count_update_callback: typing.Optional[collections.abc.Callable[[int, int], None]],
 ): ...
 def _map_sync(
     self, *input_iterators, kwargs={}, order_outputs: bool = True, return_exceptions: bool = False