diff --git a/pytorch_lightning/overrides/data_parallel.py b/pytorch_lightning/overrides/data_parallel.py index 6e97abbd13..921c5ef082 100644 --- a/pytorch_lightning/overrides/data_parallel.py +++ b/pytorch_lightning/overrides/data_parallel.py @@ -14,7 +14,7 @@ import itertools import threading -from collections import Mapping, Iterable +from collections.abc import Mapping, Iterable from itertools import chain import torch