[follow ups] add generator-based do notation

src/result/result.py

@@ -536,8 +536,8 @@ def __init__(self, value: Any) -> None:
                 # If the function type check passes,
                 # then `e.value` will be a `Result[T, E]` type.
                 # Therefore, this return is respecting the signature of `wrapper`.
-                # Unfortunately, mypy does not infer this, so we suppress it.
-                return e.value  # type: ignore
+                # Pyright infers this, but unfortunately, mypy does not.
+                return e.value
             except TypeError as te:
                 raise TypeError(
                     "The function passed to `do` must be a generator.\n"

tests/test_result.py

@@ -408,6 +408,17 @@ def no_yield() -> DoResult[int, str]:
         assert exp_msg in str(exc_info.value)
 
 
+def test_do_always_ok() -> None:
+    if run_do_test():
+        @do
+        def always_ok() -> DoResult[int, str]:
+            _ = yield Ok(1)
+            return Ok(42)
+
+        result = always_ok()
+        assert isinstance(result, Ok), "Expected Ok, got Err"
+
+
 def test_do_always_err() -> None:
     if run_do_test():
         @do