Skip to content

Commit 876703a

Browse files
committed
Change type hint for DDP model
1 parent 94a5ab7 commit 876703a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch_lightning/utilities/distributed.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -244,7 +244,7 @@ def all_gather_ddp_if_available(
244244

245245

246246
def register_ddp_comm_hook(
247-
model: Type[DistributedDataParallel],
247+
model: DistributedDataParallel,
248248
ddp_comm_state: Optional[object] = None,
249249
ddp_comm_hook: Optional[Callable] = None,
250250
ddp_comm_wrapper: Optional[Callable] = None,

0 commit comments

Comments
 (0)