Fix mypy error in torch.random

kakaobrain · Oct 21, 2019 · b68f1a9 · b68f1a9
1 parent c7391bf
commit b68f1a9
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/stubs/torch/random.pyi b/stubs/torch/random.pyi
@@ -1,13 +1,14 @@
 #MODIFIED BY TORCHGPIPE
 from contextlib import contextmanager
-from typing import Generator, Iterable, Union
-
+from typing import Any, Generator, Iterable, Union
 from torch import ByteTensor, device
 
-
 def set_rng_state(new_state: ByteTensor) -> None: ...
 def get_rng_state() -> ByteTensor: ...
 
+def manual_seed(seed: int) -> Any: ...
+def seed() -> int: ...
+def initial_seed() -> int: ...
 
 @contextmanager
 def fork_rng(devices: Iterable[Union[device, str, int]] = ..., enabled: bool = ...) -> Generator[None, None, None]: ...