FIx mypy for init_ddp_connection (#9051)

This commit is contained in:
Kaushik B 2021-08-23 21:16:15 +05:30 committed by GitHub
parent 49c52b0d4b
commit f3c5889aa3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -353,7 +353,7 @@ def init_ddp_connection(
torch_distributed_backend: str,
global_rank: Optional[int] = None,
world_size: Optional[int] = None,
**kwargs,
**kwargs: Any,
) -> None:
"""
Utility function to initialize DDP connection by setting env variables