From f3c5889aa3116bb8299a849f4f02dfb535b5624e Mon Sep 17 00:00:00 2001 From: Kaushik B <45285388+kaushikb11@users.noreply.github.com> Date: Mon, 23 Aug 2021 21:16:15 +0530 Subject: [PATCH] FIx mypy for init_ddp_connection (#9051) --- pytorch_lightning/utilities/distributed.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/utilities/distributed.py b/pytorch_lightning/utilities/distributed.py index 0973f11694..68868a0ff7 100644 --- a/pytorch_lightning/utilities/distributed.py +++ b/pytorch_lightning/utilities/distributed.py @@ -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