Skip to content
Snippets Groups Projects
Commit 85b8c524 authored by Cyril Moineau's avatar Cyril Moineau
Browse files

Fix typo, NodeExport was not using is_input and is_output

parent 092cb4bd
No related branches found
No related tags found
3 merge requests!279v0.4.0,!253v0.4.0,!163Export refactor
......@@ -143,8 +143,8 @@ class ExportNode(ABC):
# TODO : this check doesn't work if we export a subgraph !
# Maybe we need to add the graph we want to export as parameter !
# Actually may be mandatory for memory manager ...
self.attributes["is_input"] = len(self.node.get_parents()) == 0
self.attributes["is_output"] = len(self.node.get_children()) == 0
self.attributes["is_input"] = is_input
self.attributes["is_output"] = is_output
self.inputs = []
self.outputs = []
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment