forked from goodboy/tractor
1
0
Fork 0

Fix typing

bistream_backup
Tyler Goodlet 2021-05-07 11:52:08 -04:00
parent c4d5f9d41e
commit 15b63b7190
2 changed files with 17 additions and 5 deletions

View File

@ -340,7 +340,7 @@ class Portal:
self,
func: Callable,
**kwargs,
) -> Context:
) -> AsyncGenerator[Tuple[Context, Any], None]:
"""Open an inter-actor task context.
This is a synchronous API which allows for deterministic

View File

@ -1,7 +1,15 @@
"""
Message stream types and APIs.
"""
import inspect
from contextlib import contextmanager, asynccontextmanager
from dataclasses import dataclass
from typing import Any, Iterator, Optional, Callable
from typing import (
Any, Iterator, Optional, Callable,
AsyncGenerator,
)
import warnings
import trio
@ -288,7 +296,7 @@ class Context:
async def open_stream(
self,
shield: bool = False,
) -> MsgStream:
) -> AsyncGenerator[MsgStream, None]:
# TODO
actor = current_actor()
@ -339,7 +347,9 @@ def stream(func: Callable) -> Callable:
"""
# annotate
func._tractor_stream_function = True
# TODO: apply whatever solution ``mypy`` ends up picking for this:
# https://github.com/python/mypy/issues/2087#issuecomment-769266912
func._tractor_stream_function = True # type: ignore
sig = inspect.signature(func)
params = sig.parameters
@ -369,7 +379,9 @@ def context(func: Callable) -> Callable:
"""
# annotate
func._tractor_context_function = True
# TODO: apply whatever solution ``mypy`` ends up picking for this:
# https://github.com/python/mypy/issues/2087#issuecomment-769266912
func._tractor_context_function = True # type: ignore
sig = inspect.signature(func)
params = sig.parameters