From 72f5d52343d551b1134b85cd428b55f1ca1e4456 Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Wed, 24 Jul 2024 13:30:36 +0000
Subject: [PATCH] Make meminfo and is_io optional.
---
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 5eba1647c..581e9070b 100644
--- a/aidge_core/export_utils/node_export.py
+++ b/aidge_core/export_utils/node_export.py
@@ -113,7 +113,7 @@ class ExportNode(ABC):
"""
@abstractmethod
- def __init__(self, aidge_node: aidge_core.Node, mem_info: List[dict], is_input: bool, is_output: bool) -> None:
+ def __init__(self, aidge_node: aidge_core.Node, mem_info: List[dict]=None, is_input: bool=False, is_output: bool=False) -> None:
"""Create ExportNode and retieve attriubtes from ``aidge_node``:
"""
@@ -159,7 +159,7 @@ class ExportNode(ABC):
# Producer don't have meminfo
# TODO: document this attribute
# true if node have meminfo else false
- self.attributes["meminfo"] = self.node.type() != "Producer"
+ self.attributes["meminfo"] = mem_info or self.node.type() != "Producer"
if self.attributes["meminfo"]:
self.attributes["mem_info_size"] = [None] * self.attributes["nb_out"]
self.attributes["mem_info_offset"] = [None] * self.attributes["nb_out"]
--
GitLab