Fix stub error

kakaobrain · Jun 18, 2019 · bd13f16 · bd13f16
1 parent 1e0779a
commit bd13f16
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stubs/torch/nn/__init__.pyi b/stubs/torch/nn/__init__.pyi
@@ -1,5 +1,5 @@
 #MODIFIED BY TORCHGPIPE
-from typing import Any, Callable, Dict, Iterable, Iterator, Optional, Tuple, TypeVar, Union, overload
+from typing import Any, Callable, Dict, Iterable, Iterator, List, Optional, Tuple, TypeVar, Union, overload
 
 from torch import Tensor, device