diff --git a/aidge_core/export_utils/node_export.py b/aidge_core/export_utils/node_export.py index a6a370c6b384fd718a29316d7f9d61357c64828e..ce3d53501b0fdc505084863282ff3e8e8daba2f3 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]