From c8efcdd0d3f4513c6c7537bfd3bd481b42ddc56d Mon Sep 17 00:00:00 2001 From: Tyler Goodlet Date: Sun, 8 Jan 2023 13:17:59 -0500 Subject: [PATCH] Drop `ReceiveMsgStream` from test suite --- tests/test_advanced_streaming.py | 2 +- tests/test_task_broadcasting.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_advanced_streaming.py b/tests/test_advanced_streaming.py index 99414a5..799a089 100644 --- a/tests/test_advanced_streaming.py +++ b/tests/test_advanced_streaming.py @@ -14,7 +14,7 @@ def is_win(): return platform.system() == 'Windows' -_registry: dict[str, set[tractor.ReceiveMsgStream]] = { +_registry: dict[str, set[tractor.MsgStream]] = { 'even': set(), 'odd': set(), } diff --git a/tests/test_task_broadcasting.py b/tests/test_task_broadcasting.py index 1e2f6b4..636f92b 100644 --- a/tests/test_task_broadcasting.py +++ b/tests/test_task_broadcasting.py @@ -37,7 +37,7 @@ async def echo_sequences( async def ensure_sequence( - stream: tractor.ReceiveMsgStream, + stream: tractor.MsgStream, sequence: list, delay: Optional[float] = None,