Fix covariance issue

vxgmichel · May 3, 2024 · 7471da1 · 7471da1
1 parent d9554ac
commit 7471da1
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/aiostream/aiter_utils.py b/aiostream/aiter_utils.py
@@ -105,7 +105,7 @@ def assert_async_iterator(obj: object) -> None:
 
 # Async iterator context
 
-T = TypeVar("T")
+T = TypeVar("T", covariant=True)
 Self = TypeVar("Self", bound="AsyncIteratorContext[Any]")
 
 

diff --git a/aiostream/core.py b/aiostream/core.py
@@ -38,7 +38,7 @@ class StreamEmpty(Exception):
 
 # Helpers
 
-T = TypeVar("T")
+T = TypeVar("T", covariant=True)
 X = TypeVar("X")
 A = TypeVar("A", contravariant=True)
 P = ParamSpec("P")

diff --git a/tests/test_advanced.py b/tests/test_advanced.py
@@ -13,10 +13,7 @@ def target2(x: int, *_) -> Stream[int]:
         return stream.range(x, x + 4, interval=1)
 
     def target3(x: int, *_) -> Stream[int]:
-        # TODO: fix covariance issue
-        return (
-            stream.range(0, 3, interval=1) if x else stream.throw(ZeroDivisionError)
-        )  # type: ignore
+        return stream.range(0, 3, interval=1) if x else stream.throw(ZeroDivisionError)
 
     # Concurrent run
     with assert_cleanup() as loop: