diff --git a/src/finn/core/modelwrapper.py b/src/finn/core/modelwrapper.py index d63986971b1eab7dbc7330cc78a0abc73fddbd21..dc5b36920a5639933463d682dc66fb8bc15b35f2 100644 --- a/src/finn/core/modelwrapper.py +++ b/src/finn/core/modelwrapper.py @@ -301,7 +301,7 @@ class ModelWrapper: else: return None - def find_successors(self, node): + def find_direct_successors(self, node): """Finds and returns a list of the nodes that are successors of given node.""" successors = [] @@ -315,7 +315,7 @@ class ModelWrapper: else: return None - def find_predecessors(self, node): + def find_direct_predecessors(self, node): """Finds and returns a list of the nodes that are predecessors of given node.""" predecessors = []