diff --git a/ignite/distributed/comp_models/native.py b/ignite/distributed/comp_models/native.py index 2d19e54a0fb4..4fcde6626684 100644 --- a/ignite/distributed/comp_models/native.py +++ b/ignite/distributed/comp_models/native.py @@ -331,8 +331,8 @@ def spawn( # type: ignore[override] def _do_all_reduce(self, tensor: torch.Tensor, op: str = "SUM") -> torch.Tensor: if op not in self._reduce_op_map: raise ValueError("Unsupported reduction operation: '{}'".format(op)) - op = self._reduce_op_map[op] - dist.all_reduce(tensor, op) + reduce_op = self._reduce_op_map[op] + dist.all_reduce(tensor, reduce_op) return tensor def _do_all_gather(self, tensor: torch.Tensor) -> torch.Tensor: diff --git a/ignite/handlers/checkpoint.py b/ignite/handlers/checkpoint.py index b793f4fd659c..3090b8eac5f3 100644 --- a/ignite/handlers/checkpoint.py +++ b/ignite/handlers/checkpoint.py @@ -4,7 +4,7 @@ import tempfile import warnings from abc import ABCMeta, abstractmethod -from collections import OrderedDict, namedtuple +from collections import OrderedDict from tempfile import _TemporaryFileWrapper # type: ignore[attr-defined] from typing import Any, Callable, Dict, List, Mapping, NamedTuple, Optional, Tuple, Union