diff --git a/mypy.ini b/mypy.ini index a7c82cb6935..50287b8ee99 100644 --- a/mypy.ini +++ b/mypy.ini @@ -186,8 +186,8 @@ ignore_errors = True [mypy-torch.cuda.amp.*] ignore_errors = True -[mypy-torch.cuda.comm] -ignore_errors = True +#[mypy-torch.cuda.comm] +#ignore_errors = True [mypy-torch.cuda.nccl] ignore_errors = True diff --git a/torch/cuda/comm.py b/torch/cuda/comm.py index f9856eda380..557ffb0c0de 100644 --- a/torch/cuda/comm.py +++ b/torch/cuda/comm.py @@ -2,4 +2,4 @@ from torch.nn.parallel.comm import broadcast, broadcast_coalesced, reduce_add, \ reduce_add_coalesced, scatter, gather -__all__ = [broadcast, broadcast_coalesced, reduce_add, reduce_add_coalesced, scatter, gather] +__all__ = ['broadcast', 'broadcast_coalesced', 'reduce_add', 'reduce_add_coalesced', 'scatter', 'gather']