From eff7dc2f49eec310a97600da2b1d831e51b1a64a Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Tue, 23 Jul 2024 09:26:15 +0000
Subject: [PATCH] Raise error if no input set instead of just a warning.
---
aidge_core/export_utils/node_export.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/aidge_core/export_utils/node_export.py b/aidge_core/export_utils/node_export.py
index a6a370c6b..ce3d53501 100644
--- a/aidge_core/export_utils/node_export.py
+++ b/aidge_core/export_utils/node_export.py
@@ -187,7 +187,7 @@ class ExportNode(ABC):
self.attributes["in_height"][idx] = get_height(tensor)
self.attributes["in_width"][idx] = get_width(tensor)
else:
- print(f"No input for {self.node.name()}")
+ 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]
for node_in_id in list_child_node_in_id]
--
GitLab