From 58e465aa4727998e474b3726600ca69af64c0e78 Mon Sep 17 00:00:00 2001 From: chentianyu03 Date: Fri, 25 Jun 2021 14:53:44 +0800 Subject: [PATCH] replace six lib in container file (#33775) --- python/paddle/nn/layer/container.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/python/paddle/nn/layer/container.py b/python/paddle/nn/layer/container.py index ad41535f44a..48697aa8f50 100644 --- a/python/paddle/nn/layer/container.py +++ b/python/paddle/nn/layer/container.py @@ -14,7 +14,7 @@ from collections import OrderedDict from ...fluid.dygraph.layers import Layer -from six.moves import collections_abc +from collections.abc import Iterable, Mapping __all__ = [] @@ -276,12 +276,11 @@ class LayerDict(Layer): """ assert isinstance( - sublayers, collections_abc.Iterable + sublayers, Iterable ), "The type of sublayers is not iterable of key/value pairs, the type of sublayers is " + type( sublayers).__name__ - if isinstance(sublayers, - (OrderedDict, LayerDict, collections_abc.Mapping)): + if isinstance(sublayers, (OrderedDict, LayerDict, Mapping)): for key, layer in sublayers.items(): self.add_sublayer(key, layer) else: -- GitLab