From d60bea14f65ce51babdb8def77551c6be3c657c2 Mon Sep 17 00:00:00 2001
From: Filip Figiel <peccator93@gmail.com>
Date: Wed, 26 Oct 2016 16:19:24 +0200
Subject: [PATCH] Improve concurrent.futures.wait function stub (#634)

---
 stdlib/3/concurrent/futures/_base.pyi | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/stdlib/3/concurrent/futures/_base.pyi b/stdlib/3/concurrent/futures/_base.pyi
index 819e79cf7..461c56a94 100644
--- a/stdlib/3/concurrent/futures/_base.pyi
+++ b/stdlib/3/concurrent/futures/_base.pyi
@@ -1,4 +1,4 @@
-from typing import TypeVar, Generic, Any, Iterable, Iterator, Callable, Tuple, Optional
+from typing import TypeVar, Generic, Any, Iterable, Iterator, Callable, Tuple, Optional, Set
 from collections import namedtuple
 
 FIRST_COMPLETED = ... # type: Any
@@ -41,4 +41,4 @@ class Executor:
 
 def as_completed(fs: Iterable[Future], timeout: Optional[float] = ...) -> Iterator[Future]: ...
 
-def wait(fs: Iterable[Future], timeout: Optional[float] = ..., return_when: str = ...) -> Tuple[Iterable[Future], Iterable[Future]]: ...
+def wait(fs: Iterable[Future], timeout: Optional[float] = ..., return_when: str = ...) -> Tuple[Set[Future], Set[Future]]: ...