From 1f18fa2024665facc146c6254e93b02af19c4508 Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Thu, 26 Sep 2024 08:08:33 +0000
Subject: [PATCH] Export node pass Input information extraction for Optional
Data and Param.
---
aidge_core/export_utils/node_export.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/aidge_core/export_utils/node_export.py b/aidge_core/export_utils/node_export.py
index cafe10b55..15911ee58 100644
--- a/aidge_core/export_utils/node_export.py
+++ b/aidge_core/export_utils/node_export.py
@@ -126,7 +126,7 @@ class ExportNode(ABC):
@abstractmethod
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``:
+ """Create ExportNode and retrieve attributes from ``aidge_node``:
"""
super().__init__()
@@ -199,7 +199,11 @@ class ExportNode(ABC):
self.attributes["in_chan"][idx] = get_chan(tensor)
self.attributes["in_height"][idx] = get_height(tensor)
self.attributes["in_width"][idx] = get_width(tensor)
+ elif self.operator.input_category(idx) == aidge_core.InputCategory.OptionalParam or \
+ self.operator.input_category(idx) == aidge_core.InputCategory.OptionalData:
+ pass
else:
+ print(self.operator.input_category(idx))
raise RuntimeError(f"No input for {self.node.name()} at input {idx}")
for idx, list_child_node_in_id in enumerate(self.node.outputs()):
self.outputs += [node_in_id[0]
--
GitLab