From 81af63eb48055a7c4b42b169aebd5046a2d85f68 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Thu, 24 Oct 2019 14:30:46 +0100
Subject: [PATCH] [Util] allow specifying name field for _by_name
---
src/finn/core/utils.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/src/finn/core/utils.py b/src/finn/core/utils.py
index bbb976b77..d10da1ed3 100644
--- a/src/finn/core/utils.py
+++ b/src/finn/core/utils.py
@@ -11,9 +11,9 @@ def valueinfo_to_tensor(vi):
)
-def get_by_name(container, name):
+def get_by_name(container, name, name_field="name"):
"""Return item from container by .name field if it exists, None otherwise"""
- names = [x.name for x in container]
+ names = [getattr(x, name_field) for x in container]
try:
ind = names.index(name)
return container[ind]
@@ -21,8 +21,8 @@ def get_by_name(container, name):
return None
-def remove_by_name(container, name):
+def remove_by_name(container, name, name_field="name"):
"""Remove item from container by .name field if it exists"""
- item = get_by_name(container, name)
+ item = get_by_name(container, name, name_field)
if item is not None:
container.remove(item)
--
GitLab