diff --git a/ignite/engine/__init__.py b/ignite/engine/__init__.py index 59b46d19d67b..3fbc83861b33 100644 --- a/ignite/engine/__init__.py +++ b/ignite/engine/__init__.py @@ -1,4 +1,4 @@ -import collections +from collections.abc import Mapping from typing import Any, Callable, Dict, Optional, Sequence, Tuple, Union import torch @@ -28,7 +28,7 @@ def _prepare_batch( batch: Sequence[torch.Tensor], device: Optional[Union[str, torch.device]] = None, non_blocking: bool = False -) -> Tuple[Union[torch.Tensor, collections.Sequence, collections.Mapping, str, bytes], ...]: +) -> Tuple[Union[torch.Tensor, Sequence, Mapping, str, bytes], ...]: """Prepare batch for training: pass to a device with options. """