From cff451cdacede0ee291ce4b39a69ce9615abfc30 Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Tue, 8 Oct 2024 12:50:55 +0000
Subject: [PATCH] [NodeExport] Update error message when no input dims.
---
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 70194a141..8059eeea1 100644
--- a/aidge_core/export_utils/node_export.py
+++ b/aidge_core/export_utils/node_export.py
@@ -203,7 +203,7 @@ class ExportNode(ABC):
self.operator.input_category(idx) == aidge_core.InputCategory.OptionalData:
pass
else:
- aidge_core.Log.notice(f"No input for {self.node.name()} at input {idx}")
+ raise RuntimeError(f"No input for {self.node.name()} at input {idx}, did you forget to forward dims?")
for idx, list_child_node_in_id in enumerate(self.node.outputs()):
self.outputs += [node_in_id[0]
for node_in_id in list_child_node_in_id]
@@ -256,7 +256,7 @@ class ExportNode(ABC):
else:
self.attributes["mem_info_wrap_size"][idx] = 0
else:
- aidge_core.Log.notice(f"No output for {self.node.name()}")
+ raise RuntimeError(f"No output for {self.node.name()}")
class ExportNodeCpp(ExportNode):
# Path to the template defining how to export the node definition
--
GitLab