fixes

returns/methods/partition.py

@@ -1,7 +1,7 @@
 
-from typing import Iterable, List, TypeVar
+from typing import Iterable, List, TypeVar, Union
 
-from returns.interfaces.specific import result
+from returns.interfaces.specific import ioresult, result
 from returns.primitives.exceptions import UnwrapFailedError
 
 _ValueType = TypeVar('_ValueType', covariant=True)
@@ -11,7 +11,10 @@
 
 def partition(
     containers: Iterable[
-        result.ResultBasedN[_ValueType, _ErrorType, _AdditionalType]
+        Union[
+        result.ResultBasedN[_ValueType, _ErrorType, _AdditionalType],
+        ioresult.IOResultBasedN[_ValueType, _ErrorType, _AdditionalType],
+        ]
     ],
 ) -> tuple[List[_ValueType], List[_ErrorType]]:
     """

typesafety/test_methods/test_partition.yml

@@ -1,19 +1,20 @@
 - case: partition_no_params
   disable_cache: false
   main: |
-    from returns.result import Success, Failure
+    from typing import List
+    from returns.result import Success, Failure, Result
     from returns.methods import partition
 
     x: List[Result[int, str]]
-    reveal_type(partition(x))  # N: Revealed type is "Tuple[builtins.list[builtin.int], builtins.list[builtin.str]]"
+    reveal_type(partition(x))  # N: Revealed type is "Tuple[builtins.list[builtins.int], builtins.list[builtins.str]]"
 
 - case: partition_io_results
   disable_cache: false
   main: |
     from typing import Tuple
     from returns.result import Success, Failure
     from returns.methods import partition
-    from returns.io import IO, IOResult
+    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]]"