broke brains because of typesafety

dry-python · Aug 12, 2024 · 7a2a8fc · 7a2a8fc
1 parent afe493f
commit 7a2a8fc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/returns/methods/partition.py b/returns/methods/partition.py
@@ -1,7 +1,7 @@
 
-from typing import Iterable, List, TypeVar, Union
+from typing import Iterable, List, TypeVar
 
-from returns.interfaces.specific import ioresult, result
+from returns.interfaces.specific import result
 from returns.primitives.exceptions import UnwrapFailedError
 
 _ValueType = TypeVar('_ValueType', covariant=True)
@@ -11,10 +11,7 @@
 
 def partition(
  containers: Iterable[
- Union[
- result.ResultBasedN[_ValueType, _ErrorType, _AdditionalType],
- ioresult.IOResultBasedN[_ValueType, _ErrorType, _AdditionalType],
- ]
+ result.UnwrappableResult[_ValueType, _ErrorType, _AdditionalType, _ValueType, _ErrorType],
  ],
 ) -> tuple[List[_ValueType], List[_ErrorType]]:
  """

diff --git a/typesafety/test_methods/test_partition.yml b/typesafety/test_methods/test_partition.yml
@@ -17,5 +17,5 @@
  from returns.io import IO, IOResult, IOSuccess
 
  x: Tuple[IOResult[int, str], IOResult[int, str]]
- reveal_type(partition(x)) # N: Revealed type is "Tuple[builtins.list[IO[builtin.int]], builtins.list[IO[builtin.int]]"
+ reveal_type(partition(x)) # N: Revealed type is "Tuple[builtins.list[IO[builtin.int]], builtins.list[IO[builtin.str]]"