c941730b96c3bff1490086dc1989d6553b74b175,test/test_futures.py,,,#,268

Before Change

        with self.assertRaisesRegex(RuntimeError, "Expected error"):
            torch.futures.wait_all([fut3, fut2])

if __name__ == "__main__":
    run_tests()

After Change

from torch.testing._internal.common_utils import IS_WINDOWS, TestCase, TemporaryFileName, run_tests
from typing import TypeVar

T = TypeVar("T")


def add_one(fut):
    return fut.wait() + 1

Instances