提交 9b0bd695 编写于 作者: M Megvii Engine Team

fix(imperative/utils): fix name filter of network varnode

GitOrigin-RevId: 20ead353355d10bf122f3243bb9e6b520eb32c36
上级 af62cf6f
......@@ -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
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册