From 85b8c524819069568c83378b7cded1cef289fdf0 Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Wed, 4 Sep 2024 12:13:45 +0000
Subject: [PATCH] Fix typo, NodeExport was not using is_input and is_output
---
aidge_core/export_utils/node_export.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/aidge_core/export_utils/node_export.py b/aidge_core/export_utils/node_export.py
index 522daa716..eeba09c92 100644
--- a/aidge_core/export_utils/node_export.py
+++ b/aidge_core/export_utils/node_export.py
@@ -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 = []
--
GitLab