From 9b0bd695216327ac286f9d268c98be661185ffc9 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Fri, 9 Apr 2021 18:23:55 +0800 Subject: [PATCH] fix(imperative/utils): fix name filter of network varnode GitOrigin-RevId: 20ead353355d10bf122f3243bb9e6b520eb32c36 --- imperative/python/megengine/utils/network.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/imperative/python/megengine/utils/network.py b/imperative/python/megengine/utils/network.py index 8bcf4d89..8738a5b5 100644 --- a/imperative/python/megengine/utils/network.py +++ b/imperative/python/megengine/utils/network.py @@ -663,6 +663,7 @@ class NodeFilterName(NodeFilter): def __init__(self, node_iter, pattern, ignorecase): super().__init__(node_iter) + self.pattern = pattern self._re = self.make_re(pattern, ignorecase) @classmethod @@ -676,5 +677,5 @@ class NodeFilterName(NodeFilter): def __iter__(self): for i in self._iter: - if self._re.match(i.name): + if self.pattern == i.name or self._re.match(i.name): yield i -- GitLab