modal 0.67.1__py3-none-any.whl → 0.67.33__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- modal/_clustered_functions.py +2 -2
- modal/_clustered_functions.pyi +2 -2
- modal/_container_entrypoint.py +8 -5
- modal/_output.py +29 -28
- modal/_pty.py +2 -2
- modal/_resolver.py +6 -5
- modal/_resources.py +3 -3
- modal/_runtime/asgi.py +46 -6
- modal/_runtime/container_io_manager.py +22 -26
- modal/_runtime/execution_context.py +2 -2
- modal/_runtime/telemetry.py +1 -2
- modal/_runtime/user_code_imports.py +12 -14
- modal/_serialization.py +3 -7
- modal/_traceback.py +5 -5
- modal/_tunnel.py +5 -4
- modal/_tunnel.pyi +2 -2
- modal/_utils/async_utils.py +53 -17
- modal/_utils/blob_utils.py +22 -7
- modal/_utils/function_utils.py +20 -10
- modal/_utils/grpc_testing.py +7 -6
- modal/_utils/grpc_utils.py +2 -3
- modal/_utils/hash_utils.py +2 -2
- modal/_utils/mount_utils.py +5 -4
- modal/_utils/package_utils.py +2 -3
- modal/_utils/pattern_matcher.py +6 -6
- modal/_utils/rand_pb_testing.py +3 -3
- modal/_utils/shell_utils.py +2 -1
- modal/_vendor/a2wsgi_wsgi.py +62 -72
- modal/_vendor/cloudpickle.py +1 -1
- modal/_watcher.py +8 -7
- modal/app.py +68 -62
- modal/app.pyi +104 -99
- modal/call_graph.py +6 -6
- modal/cli/_download.py +3 -2
- modal/cli/_traceback.py +4 -4
- modal/cli/app.py +4 -4
- modal/cli/container.py +4 -4
- modal/cli/dict.py +1 -1
- modal/cli/environment.py +2 -3
- modal/cli/import_refs.py +1 -1
- modal/cli/launch.py +2 -2
- modal/cli/network_file_system.py +1 -1
- modal/cli/profile.py +1 -1
- modal/cli/programs/run_jupyter.py +2 -2
- modal/cli/programs/vscode.py +3 -3
- modal/cli/queues.py +1 -1
- modal/cli/run.py +6 -6
- modal/cli/secret.py +3 -3
- modal/cli/utils.py +2 -1
- modal/cli/volume.py +3 -3
- modal/client.py +6 -11
- modal/client.pyi +18 -27
- modal/cloud_bucket_mount.py +3 -3
- modal/cloud_bucket_mount.pyi +2 -2
- modal/cls.py +100 -47
- modal/cls.pyi +40 -40
- modal/config.py +3 -2
- modal/container_process.py +6 -2
- modal/dict.py +6 -3
- modal/dict.pyi +10 -9
- modal/environments.py +3 -3
- modal/environments.pyi +3 -3
- modal/exception.py +2 -3
- modal/functions.py +112 -104
- modal/functions.pyi +77 -58
- modal/image.py +59 -57
- modal/image.pyi +104 -103
- modal/io_streams.py +20 -12
- modal/io_streams.pyi +24 -14
- modal/mount.py +24 -24
- modal/mount.pyi +28 -29
- modal/network_file_system.py +14 -11
- modal/network_file_system.pyi +12 -11
- modal/object.py +9 -8
- modal/object.pyi +47 -34
- modal/output.py +2 -1
- modal/parallel_map.py +4 -4
- modal/partial_function.py +10 -14
- modal/partial_function.pyi +17 -18
- modal/queue.py +11 -8
- modal/queue.pyi +23 -22
- modal/retries.py +38 -0
- modal/runner.py +8 -7
- modal/runner.pyi +8 -14
- modal/running_app.py +3 -3
- modal/sandbox.py +20 -13
- modal/sandbox.pyi +73 -72
- modal/scheduler_placement.py +2 -1
- modal/secret.py +7 -7
- modal/secret.pyi +12 -12
- modal/serving.py +4 -3
- modal/serving.pyi +5 -4
- modal/token_flow.py +3 -2
- modal/token_flow.pyi +3 -3
- modal/volume.py +16 -23
- modal/volume.pyi +17 -16
- {modal-0.67.1.dist-info → modal-0.67.33.dist-info}/METADATA +2 -2
- modal-0.67.33.dist-info/RECORD +168 -0
- modal_docs/mdmd/signatures.py +1 -2
- modal_global_objects/mounts/python_standalone.py +1 -1
- modal_proto/api.proto +15 -0
- modal_proto/api_grpc.py +32 -0
- modal_proto/api_pb2.py +674 -654
- modal_proto/api_pb2.pyi +45 -1
- modal_proto/api_pb2_grpc.py +66 -0
- modal_proto/api_pb2_grpc.pyi +20 -0
- modal_proto/modal_api_grpc.py +2 -0
- modal_version/_version_generated.py +1 -1
- modal-0.67.1.dist-info/RECORD +0 -168
- {modal-0.67.1.dist-info → modal-0.67.33.dist-info}/LICENSE +0 -0
- {modal-0.67.1.dist-info → modal-0.67.33.dist-info}/WHEEL +0 -0
- {modal-0.67.1.dist-info → modal-0.67.33.dist-info}/entry_points.txt +0 -0
- {modal-0.67.1.dist-info → modal-0.67.33.dist-info}/top_level.txt +0 -0
modal/cli/launch.py
CHANGED
@@ -4,7 +4,7 @@ import inspect
|
|
4
4
|
import json
|
5
5
|
import os
|
6
6
|
from pathlib import Path
|
7
|
-
from typing import Any,
|
7
|
+
from typing import Any, Optional
|
8
8
|
|
9
9
|
from typer import Typer
|
10
10
|
|
@@ -25,7 +25,7 @@ launch_cli = Typer(
|
|
25
25
|
)
|
26
26
|
|
27
27
|
|
28
|
-
def _launch_program(name: str, filename: str, detach: bool, args:
|
28
|
+
def _launch_program(name: str, filename: str, detach: bool, args: dict[str, Any]) -> None:
|
29
29
|
os.environ["MODAL_LAUNCH_ARGS"] = json.dumps(args)
|
30
30
|
|
31
31
|
program_path = str(Path(__file__).parent / "programs" / filename)
|
modal/cli/network_file_system.py
CHANGED
@@ -29,7 +29,7 @@ nfs_cli = Typer(name="nfs", help="Read and edit `modal.NetworkFileSystem` file s
|
|
29
29
|
|
30
30
|
@nfs_cli.command(name="list", help="List the names of all network file systems.", rich_help_panel="Management")
|
31
31
|
@synchronizer.create_blocking
|
32
|
-
async def
|
32
|
+
async def list_(env: Optional[str] = ENV_OPTION, json: Optional[bool] = False):
|
33
33
|
env = ensure_env(env)
|
34
34
|
|
35
35
|
client = await _Client.from_env()
|
modal/cli/profile.py
CHANGED
@@ -28,7 +28,7 @@ def current():
|
|
28
28
|
|
29
29
|
@profile_cli.command(name="list", help="Show all Modal profiles and highlight the active one.")
|
30
30
|
@synchronizer.create_blocking
|
31
|
-
async def
|
31
|
+
async def list_(json: Optional[bool] = False):
|
32
32
|
config = Config()
|
33
33
|
profiles = config_profiles()
|
34
34
|
lookup_coros = [
|
@@ -8,12 +8,12 @@ import subprocess
|
|
8
8
|
import threading
|
9
9
|
import time
|
10
10
|
import webbrowser
|
11
|
-
from typing import Any
|
11
|
+
from typing import Any
|
12
12
|
|
13
13
|
from modal import App, Image, Mount, Queue, Secret, Volume, forward
|
14
14
|
|
15
15
|
# Passed by `modal launch` locally via CLI, plumbed to remote runner through secrets.
|
16
|
-
args:
|
16
|
+
args: dict[str, Any] = json.loads(os.environ.get("MODAL_LAUNCH_ARGS", "{}"))
|
17
17
|
|
18
18
|
|
19
19
|
app = App()
|
modal/cli/programs/vscode.py
CHANGED
@@ -8,12 +8,12 @@ import subprocess
|
|
8
8
|
import threading
|
9
9
|
import time
|
10
10
|
import webbrowser
|
11
|
-
from typing import Any
|
11
|
+
from typing import Any
|
12
12
|
|
13
13
|
from modal import App, Image, Mount, Queue, Secret, Volume, forward
|
14
14
|
|
15
15
|
# Passed by `modal launch` locally via CLI, plumbed to remote runner through secrets.
|
16
|
-
args:
|
16
|
+
args: dict[str, Any] = json.loads(os.environ.get("MODAL_LAUNCH_ARGS", "{}"))
|
17
17
|
|
18
18
|
|
19
19
|
app = App()
|
@@ -41,7 +41,7 @@ volume = (
|
|
41
41
|
volumes = {"/home/coder/volume": volume} if volume else {}
|
42
42
|
|
43
43
|
|
44
|
-
def wait_for_port(data:
|
44
|
+
def wait_for_port(data: tuple[str, str], q: Queue):
|
45
45
|
start_time = time.monotonic()
|
46
46
|
while True:
|
47
47
|
try:
|
modal/cli/queues.py
CHANGED
@@ -58,7 +58,7 @@ async def delete(name: str, *, yes: bool = YES_OPTION, env: Optional[str] = ENV_
|
|
58
58
|
|
59
59
|
@queue_cli.command(name="list", rich_help_panel="Management")
|
60
60
|
@synchronizer.create_blocking
|
61
|
-
async def
|
61
|
+
async def list_(*, json: bool = False, env: Optional[str] = ENV_OPTION):
|
62
62
|
"""List all named Queues."""
|
63
63
|
env = ensure_env(env)
|
64
64
|
|
modal/cli/run.py
CHANGED
@@ -9,7 +9,7 @@ import sys
|
|
9
9
|
import time
|
10
10
|
import typing
|
11
11
|
from functools import partial
|
12
|
-
from typing import Any, Callable,
|
12
|
+
from typing import Any, Callable, Optional, get_type_hints
|
13
13
|
|
14
14
|
import click
|
15
15
|
import typer
|
@@ -60,7 +60,7 @@ class NoParserAvailable(InvalidError):
|
|
60
60
|
pass
|
61
61
|
|
62
62
|
|
63
|
-
def _get_signature(f: Callable[..., Any], is_method: bool = False) ->
|
63
|
+
def _get_signature(f: Callable[..., Any], is_method: bool = False) -> dict[str, ParameterMetadata]:
|
64
64
|
try:
|
65
65
|
type_hints = get_type_hints(f)
|
66
66
|
except Exception as exc:
|
@@ -71,7 +71,7 @@ def _get_signature(f: Callable[..., Any], is_method: bool = False) -> Dict[str,
|
|
71
71
|
if is_method:
|
72
72
|
self = None # Dummy, doesn't matter
|
73
73
|
f = functools.partial(f, self)
|
74
|
-
signature:
|
74
|
+
signature: dict[str, ParameterMetadata] = {}
|
75
75
|
for param in inspect.signature(f).parameters.values():
|
76
76
|
signature[param.name] = {
|
77
77
|
"name": param.name,
|
@@ -98,7 +98,7 @@ def _get_param_type_as_str(annot: Any) -> str:
|
|
98
98
|
return annot_str
|
99
99
|
|
100
100
|
|
101
|
-
def _add_click_options(func, signature:
|
101
|
+
def _add_click_options(func, signature: dict[str, ParameterMetadata]):
|
102
102
|
"""Adds @click.option based on function signature
|
103
103
|
|
104
104
|
Kind of like typer, but using options instead of positional arguments
|
@@ -148,7 +148,7 @@ def _get_click_command_for_function(app: App, function_tag):
|
|
148
148
|
if function.is_generator:
|
149
149
|
raise InvalidError("`modal run` is not supported for generator functions")
|
150
150
|
|
151
|
-
signature:
|
151
|
+
signature: dict[str, ParameterMetadata]
|
152
152
|
cls: Optional[Cls] = None
|
153
153
|
if function.info.user_cls is not None:
|
154
154
|
cls = typing.cast(Cls, app.registered_classes[class_name])
|
@@ -364,7 +364,7 @@ def shell(
|
|
364
364
|
default=None, help="Container image tag for inside the shell (if not using REF)."
|
365
365
|
),
|
366
366
|
add_python: Optional[str] = typer.Option(default=None, help="Add Python to the image (if not using REF)."),
|
367
|
-
volume: Optional[
|
367
|
+
volume: Optional[list[str]] = typer.Option(
|
368
368
|
default=None,
|
369
369
|
help=(
|
370
370
|
"Name of a `modal.Volume` to mount inside the shell at `/mnt/{name}` (if not using REF)."
|
modal/cli/secret.py
CHANGED
@@ -3,7 +3,7 @@ import os
|
|
3
3
|
import platform
|
4
4
|
import subprocess
|
5
5
|
from tempfile import NamedTemporaryFile
|
6
|
-
from typing import
|
6
|
+
from typing import Optional
|
7
7
|
|
8
8
|
import click
|
9
9
|
import typer
|
@@ -23,7 +23,7 @@ secret_cli = typer.Typer(name="secret", help="Manage secrets.", no_args_is_help=
|
|
23
23
|
|
24
24
|
@secret_cli.command("list", help="List your published secrets.")
|
25
25
|
@synchronizer.create_blocking
|
26
|
-
async def
|
26
|
+
async def list_(env: Optional[str] = ENV_OPTION, json: Optional[bool] = False):
|
27
27
|
env = ensure_env(env)
|
28
28
|
client = await _Client.from_env()
|
29
29
|
response = await retry_transient_errors(client.stub.SecretList, api_pb2.SecretListRequest(environment_name=env))
|
@@ -47,7 +47,7 @@ async def list(env: Optional[str] = ENV_OPTION, json: Optional[bool] = False):
|
|
47
47
|
@synchronizer.create_blocking
|
48
48
|
async def create(
|
49
49
|
secret_name,
|
50
|
-
keyvalues:
|
50
|
+
keyvalues: list[str] = typer.Argument(..., help="Space-separated KEY=VALUE items"),
|
51
51
|
env: Optional[str] = ENV_OPTION,
|
52
52
|
force: bool = typer.Option(False, "--force", help="Overwrite the secret if it already exists."),
|
53
53
|
):
|
modal/cli/utils.py
CHANGED
@@ -1,8 +1,9 @@
|
|
1
1
|
# Copyright Modal Labs 2022
|
2
2
|
import asyncio
|
3
|
+
from collections.abc import Sequence
|
3
4
|
from datetime import datetime
|
4
5
|
from json import dumps
|
5
|
-
from typing import Optional,
|
6
|
+
from typing import Optional, Union
|
6
7
|
|
7
8
|
import typer
|
8
9
|
from click import UsageError
|
modal/cli/volume.py
CHANGED
@@ -2,7 +2,7 @@
|
|
2
2
|
import os
|
3
3
|
import sys
|
4
4
|
from pathlib import Path
|
5
|
-
from typing import
|
5
|
+
from typing import Optional
|
6
6
|
|
7
7
|
import typer
|
8
8
|
from click import UsageError
|
@@ -108,7 +108,7 @@ async def get(
|
|
108
108
|
rich_help_panel="Management",
|
109
109
|
)
|
110
110
|
@synchronizer.create_blocking
|
111
|
-
async def
|
111
|
+
async def list_(env: Optional[str] = ENV_OPTION, json: Optional[bool] = False):
|
112
112
|
env = ensure_env(env)
|
113
113
|
client = await _Client.from_env()
|
114
114
|
response = await retry_transient_errors(client.stub.VolumeList, api_pb2.VolumeListRequest(environment_name=env))
|
@@ -257,7 +257,7 @@ async def rm(
|
|
257
257
|
@synchronizer.create_blocking
|
258
258
|
async def cp(
|
259
259
|
volume_name: str,
|
260
|
-
paths:
|
260
|
+
paths: list[str], # accepts multiple paths, last path is treated as destination path
|
261
261
|
env: Optional[str] = ENV_OPTION,
|
262
262
|
):
|
263
263
|
ensure_env(env)
|
modal/client.py
CHANGED
@@ -3,17 +3,12 @@ import asyncio
|
|
3
3
|
import os
|
4
4
|
import platform
|
5
5
|
import warnings
|
6
|
+
from collections.abc import AsyncGenerator, AsyncIterator, Collection, Mapping
|
6
7
|
from typing import (
|
7
8
|
Any,
|
8
|
-
AsyncGenerator,
|
9
|
-
AsyncIterator,
|
10
9
|
ClassVar,
|
11
|
-
Collection,
|
12
|
-
Dict,
|
13
10
|
Generic,
|
14
|
-
Mapping,
|
15
11
|
Optional,
|
16
|
-
Tuple,
|
17
12
|
TypeVar,
|
18
13
|
Union,
|
19
14
|
)
|
@@ -40,7 +35,7 @@ CLIENT_CREATE_ATTEMPT_TIMEOUT: float = 4.0
|
|
40
35
|
CLIENT_CREATE_TOTAL_TIMEOUT: float = 15.0
|
41
36
|
|
42
37
|
|
43
|
-
def _get_metadata(client_type: int, credentials: Optional[
|
38
|
+
def _get_metadata(client_type: int, credentials: Optional[tuple[str, str]], version: str) -> dict[str, str]:
|
44
39
|
# This implements a simplified version of platform.platform() that's still machine-readable
|
45
40
|
uname: platform.uname_result = platform.uname()
|
46
41
|
if uname.system == "Darwin":
|
@@ -69,7 +64,7 @@ def _get_metadata(client_type: int, credentials: Optional[Tuple[str, str]], vers
|
|
69
64
|
|
70
65
|
ReturnType = TypeVar("ReturnType")
|
71
66
|
_Value = Union[str, bytes]
|
72
|
-
_MetadataLike = Union[Mapping[str, _Value], Collection[
|
67
|
+
_MetadataLike = Union[Mapping[str, _Value], Collection[tuple[str, _Value]]]
|
73
68
|
RequestType = TypeVar("RequestType", bound=Message)
|
74
69
|
ResponseType = TypeVar("ResponseType", bound=Message)
|
75
70
|
|
@@ -85,7 +80,7 @@ class _Client:
|
|
85
80
|
self,
|
86
81
|
server_url: str,
|
87
82
|
client_type: int,
|
88
|
-
credentials: Optional[
|
83
|
+
credentials: Optional[tuple[str, str]],
|
89
84
|
version: str = __version__,
|
90
85
|
):
|
91
86
|
"""mdmd:hidden
|
@@ -201,7 +196,7 @@ class _Client:
|
|
201
196
|
else:
|
202
197
|
c = config
|
203
198
|
|
204
|
-
credentials: Optional[
|
199
|
+
credentials: Optional[tuple[str, str]]
|
205
200
|
|
206
201
|
if cls._client_from_env_lock is None:
|
207
202
|
cls._client_from_env_lock = asyncio.Lock()
|
@@ -266,7 +261,7 @@ class _Client:
|
|
266
261
|
return client
|
267
262
|
|
268
263
|
@classmethod
|
269
|
-
async def verify(cls, server_url: str, credentials:
|
264
|
+
async def verify(cls, server_url: str, credentials: tuple[str, str]) -> None:
|
270
265
|
"""mdmd:hidden
|
271
266
|
Check whether can the client can connect to this server with these credentials; raise if not.
|
272
267
|
"""
|
modal/client.pyi
CHANGED
@@ -1,5 +1,6 @@
|
|
1
1
|
import asyncio.events
|
2
2
|
import asyncio.locks
|
3
|
+
import collections.abc
|
3
4
|
import google.protobuf.message
|
4
5
|
import grpclib.client
|
5
6
|
import modal._utils.async_utils
|
@@ -9,9 +10,7 @@ import synchronicity.combined_types
|
|
9
10
|
import typing
|
10
11
|
import typing_extensions
|
11
12
|
|
12
|
-
def _get_metadata(
|
13
|
-
client_type: int, credentials: typing.Optional[typing.Tuple[str, str]], version: str
|
14
|
-
) -> typing.Dict[str, str]: ...
|
13
|
+
def _get_metadata(client_type: int, credentials: typing.Optional[tuple[str, str]], version: str) -> dict[str, str]: ...
|
15
14
|
|
16
15
|
ReturnType = typing.TypeVar("ReturnType")
|
17
16
|
|
@@ -27,11 +26,7 @@ class _Client:
|
|
27
26
|
_stub: typing.Optional[modal_proto.api_grpc.ModalClientStub]
|
28
27
|
|
29
28
|
def __init__(
|
30
|
-
self,
|
31
|
-
server_url: str,
|
32
|
-
client_type: int,
|
33
|
-
credentials: typing.Optional[typing.Tuple[str, str]],
|
34
|
-
version: str = "0.67.1",
|
29
|
+
self, server_url: str, client_type: int, credentials: typing.Optional[tuple[str, str]], version: str = "0.67.33"
|
35
30
|
): ...
|
36
31
|
def is_closed(self) -> bool: ...
|
37
32
|
@property
|
@@ -48,7 +43,7 @@ class _Client:
|
|
48
43
|
@classmethod
|
49
44
|
async def from_credentials(cls, token_id: str, token_secret: str) -> _Client: ...
|
50
45
|
@classmethod
|
51
|
-
async def verify(cls, server_url: str, credentials:
|
46
|
+
async def verify(cls, server_url: str, credentials: tuple[str, str]) -> None: ...
|
52
47
|
@classmethod
|
53
48
|
def set_env_client(cls, client: typing.Optional[_Client]): ...
|
54
49
|
async def _call_safely(self, coro, readable_method: str): ...
|
@@ -61,8 +56,8 @@ class _Client:
|
|
61
56
|
*,
|
62
57
|
timeout: typing.Optional[float] = None,
|
63
58
|
metadata: typing.Union[
|
64
|
-
|
65
|
-
|
59
|
+
collections.abc.Mapping[str, typing.Union[str, bytes]],
|
60
|
+
collections.abc.Collection[tuple[str, typing.Union[str, bytes]]],
|
66
61
|
None,
|
67
62
|
] = None,
|
68
63
|
) -> typing.Any: ...
|
@@ -72,11 +67,11 @@ class _Client:
|
|
72
67
|
request: typing.Any,
|
73
68
|
*,
|
74
69
|
metadata: typing.Union[
|
75
|
-
|
76
|
-
|
70
|
+
collections.abc.Mapping[str, typing.Union[str, bytes]],
|
71
|
+
collections.abc.Collection[tuple[str, typing.Union[str, bytes]]],
|
77
72
|
None,
|
78
73
|
],
|
79
|
-
) ->
|
74
|
+
) -> collections.abc.AsyncGenerator[typing.Any, None]: ...
|
80
75
|
|
81
76
|
class Client:
|
82
77
|
_client_from_env: typing.ClassVar[typing.Optional[Client]]
|
@@ -86,11 +81,7 @@ class Client:
|
|
86
81
|
_stub: typing.Optional[modal_proto.api_grpc.ModalClientStub]
|
87
82
|
|
88
83
|
def __init__(
|
89
|
-
self,
|
90
|
-
server_url: str,
|
91
|
-
client_type: int,
|
92
|
-
credentials: typing.Optional[typing.Tuple[str, str]],
|
93
|
-
version: str = "0.67.1",
|
84
|
+
self, server_url: str, client_type: int, credentials: typing.Optional[tuple[str, str]], version: str = "0.67.33"
|
94
85
|
): ...
|
95
86
|
def is_closed(self) -> bool: ...
|
96
87
|
@property
|
@@ -125,7 +116,7 @@ class Client:
|
|
125
116
|
@classmethod
|
126
117
|
def from_credentials(cls, token_id: str, token_secret: str) -> Client: ...
|
127
118
|
@classmethod
|
128
|
-
def verify(cls, server_url: str, credentials:
|
119
|
+
def verify(cls, server_url: str, credentials: tuple[str, str]) -> None: ...
|
129
120
|
@classmethod
|
130
121
|
def set_env_client(cls, client: typing.Optional[Client]): ...
|
131
122
|
|
@@ -154,8 +145,8 @@ class Client:
|
|
154
145
|
*,
|
155
146
|
timeout: typing.Optional[float] = None,
|
156
147
|
metadata: typing.Union[
|
157
|
-
|
158
|
-
|
148
|
+
collections.abc.Mapping[str, typing.Union[str, bytes]],
|
149
|
+
collections.abc.Collection[tuple[str, typing.Union[str, bytes]]],
|
159
150
|
None,
|
160
151
|
] = None,
|
161
152
|
) -> typing.Any: ...
|
@@ -165,11 +156,11 @@ class Client:
|
|
165
156
|
request: typing.Any,
|
166
157
|
*,
|
167
158
|
metadata: typing.Union[
|
168
|
-
|
169
|
-
|
159
|
+
collections.abc.Mapping[str, typing.Union[str, bytes]],
|
160
|
+
collections.abc.Collection[tuple[str, typing.Union[str, bytes]]],
|
170
161
|
None,
|
171
162
|
],
|
172
|
-
) ->
|
163
|
+
) -> collections.abc.AsyncGenerator[typing.Any, None]: ...
|
173
164
|
|
174
165
|
class UnaryUnaryWrapper(typing.Generic[RequestType, ResponseType]):
|
175
166
|
wrapped_method: grpclib.client.UnaryUnaryMethod[RequestType, ResponseType]
|
@@ -184,8 +175,8 @@ class UnaryUnaryWrapper(typing.Generic[RequestType, ResponseType]):
|
|
184
175
|
*,
|
185
176
|
timeout: typing.Optional[float] = None,
|
186
177
|
metadata: typing.Union[
|
187
|
-
|
188
|
-
|
178
|
+
collections.abc.Mapping[str, typing.Union[str, bytes]],
|
179
|
+
collections.abc.Collection[tuple[str, typing.Union[str, bytes]]],
|
189
180
|
None,
|
190
181
|
] = None,
|
191
182
|
) -> ResponseType: ...
|
modal/cloud_bucket_mount.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1
1
|
# Copyright Modal Labs 2022
|
2
2
|
from dataclasses import dataclass
|
3
|
-
from typing import
|
3
|
+
from typing import Optional
|
4
4
|
from urllib.parse import urlparse
|
5
5
|
|
6
6
|
from modal_proto import api_pb2
|
@@ -116,9 +116,9 @@ class _CloudBucketMount:
|
|
116
116
|
requester_pays: bool = False
|
117
117
|
|
118
118
|
|
119
|
-
def cloud_bucket_mounts_to_proto(mounts:
|
119
|
+
def cloud_bucket_mounts_to_proto(mounts: list[tuple[str, _CloudBucketMount]]) -> list[api_pb2.CloudBucketMount]:
|
120
120
|
"""Helper function to convert `CloudBucketMount` to a list of protobufs that can be passed to the server."""
|
121
|
-
cloud_bucket_mounts:
|
121
|
+
cloud_bucket_mounts: list[api_pb2.CloudBucketMount] = []
|
122
122
|
|
123
123
|
for path, mount in mounts:
|
124
124
|
# crude mapping from mount arguments to type.
|
modal/cloud_bucket_mount.pyi
CHANGED
@@ -23,8 +23,8 @@ class _CloudBucketMount:
|
|
23
23
|
def __eq__(self, other): ...
|
24
24
|
|
25
25
|
def cloud_bucket_mounts_to_proto(
|
26
|
-
mounts:
|
27
|
-
) ->
|
26
|
+
mounts: list[tuple[str, _CloudBucketMount]],
|
27
|
+
) -> list[modal_proto.api_pb2.CloudBucketMount]: ...
|
28
28
|
|
29
29
|
class CloudBucketMount:
|
30
30
|
bucket_name: str
|